zhamao-framework/src/ZM/DB/UpdateBody.php

<?php


namespace ZM\DB;


use ZM\Exception\DbException;

class UpdateBody
{
    use WhereBody;

    /**
     * @var Table
     */
    private $table;
    /**
     * @var array
     */
    private $set_value;
    /**
     * UpdateBody constructor.
     * @param Table $table
     * @param array $set_value
     */
    public function __construct(Table $table, array $set_value) {
        $this->table = $table;
        $this->set_value = $set_value;
    }

    /**
     * @throws DbException
     */
    public function save(){
        $arr = [];
        $msg = [];
        foreach($this->set_value as $k => $v) {
            $msg []= $k .' = ?';
            $arr[]=$v;
        }
        if(($msg ?? []) == []) throw new DbException('update value sets can not be empty!');
        $line = 'UPDATE '.$this->table->getTableName().' SET '.implode(', ', $msg);
        if($this->where_thing != []) {
            list($sql, $param) = $this->getWhereSQL();
            $arr = array_merge($arr, $param);
            $line .= ' WHERE '.$sql;
        }
        return DB::rawQuery($line, $arr);
    }
}