caoym / phpboot

src/DB/DB.php

<?php
namespace PhpBoot\DB;

use PhpBoot\Application;
use PhpBoot\DB\rules\select\SelectRule;
use PhpBoot\DB\rules\insert\InsertRule;
use PhpBoot\DB\rules\update\UpdateRule;
use PhpBoot\DB\rules\delete\DeleteRule;
use PhpBoot\DB\rules\replace\ReplaceIntoRule;
use PhpBoot\Utils\Logger;

/**
 * 
 * How-to-use:
 * 
 * $db = new DB(...);
 * // 1. select
 * $res = $db->select('a, b')
 *      ->from('table')
 *      ->leftJoin('table1')->on('table.id=table1.id')
 *      ->where('a=?',1)
 *      ->groupBy('b')->having('sum(b)=?', 2)
 *      ->orderBy('c', Sql::ORDER_BY_ASC)
 *      ->limit(0,1)
 *      ->forUpdate()->of('d')
 *      ->get();
 * 
 * // 2. update
 * $rows = $db->update('table')
 *      ->set('a', 1)
 *      ->where('b=?', 2)
 *      ->orderBy('c', Sql::ORDER_BY_ASC)
 *      ->limit(1)
 *      ->exec($db)
 *      ->rows
 *      
 * // 3. insert
 * $newId = $db->insertInto('table')
 *      ->values(['a'=>1])
 *      ->exec($db)
 *      ->lastInsertId()
 *      
 * //4. delete
 * $rows = $db->deleteFrom('table')
 *      ->where('b=?', 2)
 *      ->orderBy('c', Sql::ORDER_BY_ASC)
 *      ->limit(1)
 *      ->exec($db)
 *      ->rows
 *      
 * @author caoym <caoyangmin@gmail.com>
 */
class DB{

    /**
     * DB constructor.
     * @param Application $app
     * @param string $dsn @see \PDO
     * @param string $username @see \PDO
     * @param string $password @see \PDO
     * @param array $options @see \PDO
     */

    static public function connect(Application $app,
                                   $dsn,
                                  $username,
                                  $password,
                                  $options = [])
    {
        $options += [
            \PDO::ATTR_ERRMODE =>\PDO::ERRMODE_EXCEPTION,
            \PDO::MYSQL_ATTR_INIT_COMMAND => "SET NAMES'utf8';",
            \PDO::MYSQL_ATTR_FOUND_ROWS => true
        ];

        $connection = new \PDO($dsn, $username, $password, $options);
        return new DB($app, $connection);
    }

    public function __construct(Application $app, $connection)
    {
        $this->app = $app;
        $this->connection = $connection;
    }

    /**
     * select('column0', 'column1') => "SELECT column0,column1"
     * select(['column0', 'column1']) => "SELECT column0,column1"
     *
     * @param string $column0
     * @return \PhpBoot\DB\rules\select\FromRule
     */
    function select($column0=null, $_=null){

The parameter $_ is not used and could be removed.

It is generally recommended to explicitly declare the visibility for methods.

        $obj = new SelectRule(new Context($this->connection));
        if($column0 == null){

It seems like you are loosely comparing $column0 of type string|null against null; this is ambiguous if the string can be empty. Consider using a strict comparison === instead.

            $args = ['*'];
        }elseif(is_array($column0)){
            $args = $column0;
        }else{
            $args = func_get_args();
        }
        foreach ($args as &$arg){
            $arg = DB::wrap($arg);
            if($arg == '*'){
                continue;
            }
        }
        return $obj->select(implode(',', $args));
    }
    /** 
     * insertInto('table') => "INSERT INTO table"
     * 
     * @param string $table
     * @return \PhpBoot\DB\rules\insert\ValuesRule
     */
    public function insertInto($table) {
        $obj = new InsertRule(new Context($this->connection));
        return $obj->insertInto($table);
    }
    /**
     * update('table') => "UPDATE table"
     * @param string $table
     * @return \PhpBoot\DB\rules\update\UpdateSetRule
     */
    public function update($table) {
        $obj = new UpdateRule(new Context($this->connection));
        return $obj->update($table);
    }
    
    /**
     * deleteFrom('table') => "DELETE FROM table"
     * @param string $table
     * @return \PhpBoot\DB\rules\basic\WhereRule
     */
    public function deleteFrom($table){
        $obj  =  new DeleteRule(new Context($this->connection));
        return $obj->deleteFrom($table);
    }
    /**
     * replaceInto('table') => "REPLACE INTO table"
     * @param string $table
     * @return \PhpBoot\DB\rules\replace\ValuesRule
     */
    public function replaceInto($table){
        $obj  =  new ReplaceIntoRule(new Context($this->connection));
        return $obj->replaceInto($table);
    }

    /**
     * @param callable $callback
     * @return mixed return
     * @throws \Exception
     */
    public function transaction(callable $callback)
    {
        if($this->inTransaction){
            return $callback($this);
        }
        $this->getConnection()->beginTransaction() or \PhpBoot\abort('beginTransaction failed');
        $this->inTransaction = true;
        try{
            $res = $callback($this);
            $this->getConnection()->commit() or \PhpBoot\abort('commit failed');
            return $res;
        }catch (\Exception $e){
            $this->getConnection()->rollBack();
            Logger::warning('commit failed with '.get_class($e).' '.$e->getMessage());
            throw $e;
        }
    }
    /**
     * @return \PDO
     */
    public function getConnection()
    {
        return $this->connection;
    }
    /**
	 * Splice sql use raw string(without escaping)
     * for example:
     * where('time>?', 'now()') => " WHERE time > 'now()' "
     * where('time>?', Sql::raw('now()')) => " WHERE time > now() "
     * @param string $str
     * @return Raw
     */
    static public function raw($str){
        return new Raw($str);
    }
    static public function wrap($value)
    {
        if($value instanceof Raw){
            return $value->get();
        }
        $value = trim($value);
        if ($value === '*') {
            return $value;
        }

        if(strpos($value, '.') !== false && !preg_match('/\\s+/', $value)){
            return $value;
        }
        return '`'.str_replace('`', '``', $value).'`';
    }

    /**
     * @return Application
     */
    public function getApp()
    {
        return $this->app;
    }
    const ORDER_BY_ASC ='ASC';
    const ORDER_BY_DESC ='DESC';

    /**
     * @var \PDO
     */
    protected $connection;

    /**
     * @var Application
     */
    protected $app;

    protected $inTransaction = false;
}