Home > php教程 > php手册 > 用反射来生成SQL的CREATE语句

用反射来生成SQL的CREATE语句

WBOY
Release: 2016-06-13 09:38:56
Original
1266 people have browsed it

下面的程序使用Reflection来构造"CREATE TABLE"的sql语句。如果你不是很熟悉反射机制,可以从这个程序中看看反射的魅力与作用。

<?php
/**
 * Creates an SQL 'Create Table' based upon an entity
 *
 * @author Chris Tankersley <chris@ctankersley.com>
 * @copyright 2010 Chris Tankersley
 * @package PhpORM_Cli
 */
class PhpORM_Cli_GenerateSql
{
    /**
     * Use a MySQL database
     */
    const MYSQL = 'mysql';
    /**
     * Use a SQLite database
     */
    const SQLITE = 'sqlite';
    /**
     * Types that are allowed to have a length
     * @var array
     */
    protected $_hasLength = array('integer', 'varchar');
    /**
     * Regexes needed to pull out the different comments
     * @var array
     */
    protected $_regexes = array(
        'type' => '/ type=([a-z_]*) /',
        'length' => '/ length=([0-9]*) /',
        'default' => '/ default="(.*)" /',
        'null' => '/ null /',
    );
    /**
     * Types that we support
     * @var array
     */
    protected $_validTypes = array(
        'boolean' => 'BOOL',
        'date' => 'DATE',
        'integer' => 'INT',
        'primary_autoincrement' => 'INT AUTO_INCREMENT PRIMARY KEY',
        'text' => 'TEXT',
        'timestamp' => 'TIMESTAMP',
        'varchar' => 'VARCHAR',
    );
    /**
     * Name of the class we will interperet
     * @var string
     */
    protected $_className;
    /**
     * Name of the table we are generating
     * @var string
     */
    protected $_tableName;
    /**
     * The type of database we are generating
     * @var string
     */
    protected $_type;
    /**
     * Sets the name of the class we are working with
     * @param string $class
     * @param string $table_name
     * @param string $type
     */
    public function __construct($class, $table_name, $type = self::MYSQL)
    {
        $this->_className = $class;
        $this->_tableName = $table_name;
        $this->_type = $type;
    }
    /**
     * Builds an SQL Line for a property
     * @param ReflectionProperty $property
     * @return string
     */
    protected function _getDefinition($property)
    {
        $type = '';
        $length = '';
        $null = '';
        
        preg_match($this->_regexes['type'], $property->getDocComment(), $matches);
        if(count($matches) == 2) {
            if(array_key_exists($matches[1], $this->_validTypes)) {
                $type = $this->_validTypes[$matches[1]];
                if(in_array($matches[1], $this->_hasLength)) {
                    $length = $this->_getLength($property);
                }
                if($matches[1] != 'primary_autoincrement') {
                    $null = $this->_getNull($property);
                }
                $sql = '`'.$property->getName().'` '.$type.' '.$length.' '.$null;
                return $sql;
            } else {
                throw new Exception('Type "'.$matches[1].'" is not a supported SQL type');
            }
        } else {
            throw new Exception('Found '.count($matches).' when checking Type for property '.$property->getName());
        }
    }
    /**
     * Extracts the Length from a property
     * @param ReflectionProperty $property
     * @return string
     */
    protected function _getLength($property)
    {
        preg_match($this->_regexes['length'], $property->getDocComment(), $matches);
        if(count($matches) == 2) {
            return '('.$matches[1].')';
        } else {
            return '';
        }
    }
    /**
     * Determines if a Property is allowed to be null
     * @param ReflectionProperty $property
     * @return string
     */
    protected function _getNull($property)
    {
        preg_match($this->_regexes['null'], $property->getDocComment(), $matches);
        if(count($matches) == 1) {
            return 'NULL';
        } else {
            return 'NOT NULL';
        }
    }
    /**
     * Generates a block of SQL to create a table from an Entity
     * @return string
     */
    public function getSql()
    {
        $class = new ReflectionClass($this->_className);
        $definitions = array();
        foreach($class->getProperties() as $property) {
            if(strpos($property->getName(), '_') === false) {
                $definitions[] = $this->_getDefinition($property);
            }
        }
        $columns = implode(",n", $definitions);
        
        $sql = "CREATE TABLE ".$this->_tableName." (".$columns.")";
        if($this->_type == self::MYSQL) {
            $sql .= " ENGINE=MYISAM";
        }
        return $sql.";";
    }
}
Copy after login
Related labels:
source:php.cn
Statement of this Website
The content of this article is voluntarily contributed by netizens, and the copyright belongs to the original author. This site does not assume corresponding legal responsibility. If you find any content suspected of plagiarism or infringement, please contact admin@php.cn
Popular Recommendations
Popular Tutorials
More>
Latest Downloads
More>
Web Effects
Website Source Code
Website Materials
Front End Template