MysqlManage.class.php 5.75 KB
<?php
/*
 * mysql表结构处理类
 * 创建数据表,增加,编辑,删除表中字段
 *
 */
class MysqlManage{
    /**
     * 创建数据库,并且主键是aid
     * @param table 要查询的表名
     * @param chart InnoDB MyISAM
     */
    function createTable($table,$chart='MyISAM'){
        $sql="CREATE TABLE IF NOT EXISTS `$table` (`aid` INT NOT NULL primary key)ENGINE = $chart;";
        M()->execute($sql);
        $this->checkTable($table);
    }
    /**
     * 检测表是否存在,也可以获取表中所有字段的信息
     * @param table 要查询的表名
     * return 表里所有字段的信息
     */
    function checkTable($table){
        $sql="desc `$table`";
        $info=M()->execute($sql);
        return $info;
    }
    /**
     * 检测字段是否存在,也可以获取字段信息(只能是一个字段)
     * @param table 表名
     * @param field 字段名
     */
    function checkField($table,$field){
        $sql="desc `$table` $field";
        $info=M()->execute($sql);
        return $info;
    }
    /**
     * 添加字段
     * @param table 表名
     * @param info  字段信息数组 array
     * info[name]   字段名称
     * info[type]   字段类型
     * info[length]  字段长度
     * info[isNull]  是否为空
     * info['default']   字段默认值
     * info['comment']   字段备注
     */
    function addField($table,$info){
        $sql="alter table `$table` add column ";
        $sql.=$this->filterFieldInfo($info);
        M()->execute($sql);
        $info=$this->checkField($table,$info['name']);
        return $info;
    }
    /**
     * 修改字段
     * 不能修改字段名称,只能修改
     * @param table 表名
     * @param [array] info
     * info[name]   字段名称
     * info[type]   字段类型
     * info[length]  字段长度
     * info[isNull]  是否为空
     * info['default']   字段默认值
     * info['comment']   字段备注
     */
    function editField($table,$info){
        $sql="alter table `$table` modify ";
        $sql.=$this->filterFieldInfo($info);
        M()->execute($sql);
        $this->checkField($table,$info['name']);
    }
    /**
     * 字段信息数组处理,供添加更新字段时候使用
     * @param info[name]   字段名称
     * @param info[type]   字段类型
     * @param info[length]  字段长度
     * @param info[isNull]  是否为空  1为空 2不能为空
     * @param info[unsigned] unsigned 存在则unsigned
     * @param info['default']   字段默认值
     * @param info['comment']   字段备注
     */
    private function filterFieldInfo($info){
        if(!is_array($info))
            return;
        $newInfo=array();
        $newInfo['name']=$info['name'];
        $newInfo['type']=$info['type'];
        switch($info['type']){
            case 'varchar':
            case 'char':
                $newInfo['length']=empty($info['length'])?100:$info['length'];
                $newInfo['isNull']=$info['isNull']==1?'NULL':'NOT NULL';
                $newInfo['default']=empty($info['default'])?'':'DEFAULT "'.trim($info['default'],'"').'"';
                $newInfo['comment']=empty($info['comment'])?'':'COMMENT "'.$info['comment'].'"';
                break;
            case 'int':
                $newInfo['length']=empty($info['length'])?7:$info['length'];
                $newInfo['isNull']=$info['isNull']==1?'NULL':'NOT NULL';
                $newInfo['unsigned']=empty($info['unsigned'])?'':'unsigned';
                $newInfo['default']=(empty($info['default']) && $info['default']!=0)?'':'DEFAULT "'.$info['default'].'"';
                $newInfo['comment']=empty($info['comment'])?'':'COMMENT "'.$info['comment'].'"';
             	break;
            case 'tinyint':
                $newInfo['length']=empty($info['length'])?4:$info['length'];
                $newInfo['isNull']=$info['isNull']==1?'NULL':'NOT NULL';
                $newInfo['unsigned']=empty($info['unsigned'])?'':'unsigned';
                $newInfo['default']=(empty($info['default']) && $info['default']!=0)?'':'DEFAULT "'.$info['default'].'"';
                $newInfo['comment']=empty($info['comment'])?'':'COMMENT "'.$info['comment'].'"';
                break;
            case 'text':
                $newInfo['length']='';
                $newInfo['isNull']=$info['isNull']==1?'NULL':'NOT NULL';
                $newInfo['default']='';
                $newInfo['comment']=empty($info['comment'])?'':'COMMENT "'.$info['comment'].'"';
                                break;
        }
        $sql=$newInfo['name']." ".$newInfo['type'];
        $sql.=(!empty($newInfo['length']))?'('.$newInfo['length'].')' .' ':' ';
        $sql.=$newInfo['isNull'].' ';
        if($newInfo['unsigned']) $sql.=$newInfo['unsigned'].' ';
        $sql.=$newInfo['default'].' ';
        $sql.=$newInfo['comment'];
        return $sql;
    }
    /**
     * 删除字段
     * 如果返回了字段信息则说明删除失败,返回false,则为删除成功
     * @param table 表名
     * @param field 字段(单个字段)
     */
    function dropField($table,$field){
        $sql="alter table `$table` drop column $field";
        M()->execute($sql);
        $this->checkField($table,$filed);
    }
    /**
     * 获取指定表中指定字段的信息(多字段)
     * @param table 表名
     * @param field 字段(单个字段或者数组) 
     */
    function getFieldInfo($table,$field){
        $info=array();
        if(is_string($field)){
            $this->checkField($table,$field);
        }else{
            foreach($field as $v){
                $info[$v]=$this->checkField($table,$v);
            }
        }
        return $info;
    }
}