123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485 |
- <?php
- // +----------------------------------------------------------------------
- // | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
- // +----------------------------------------------------------------------
- // | Copyright (c) 2006-2014 http://thinkphp.cn All rights reserved.
- // +----------------------------------------------------------------------
- // | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
- // +----------------------------------------------------------------------
- // | Author: liu21st <liu21st@gmail.com>
- // +----------------------------------------------------------------------
- namespace Think;
- /**
- * ThinkPHP Model模型类
- * 实现了ORM和ActiveRecords模式
- */
- class Model {
- // 当前数据库操作对象
- protected $db = null;
- // 数据库对象池
- private $_db = array();
- // 主键名称
- protected $pk = 'id';
- // 主键是否自动增长
- protected $autoinc = false;
- // 数据表前缀
- protected $tablePrefix = null;
- // 模型名称
- protected $name = '';
- // 数据库名称
- protected $dbName = '';
- //数据库配置
- protected $connection = '';
- // 数据表名(不包含表前缀)
- protected $tableName = '';
- // 实际数据表名(包含表前缀)
- protected $trueTableName = '';
- // 最近错误信息
- protected $error = '';
- // 字段信息
- protected $fields = array();
- // 数据信息
- protected $data = array();
- // 查询表达式参数
- protected $options = array();
- protected $_validate = array(); // 自动验证定义
- protected $_auto = array(); // 自动完成定义
- protected $_map = array(); // 字段映射定义
- protected $_scope = array(); // 命名范围定义
- // 是否自动检测数据表字段信息
- protected $autoCheckFields = true;
- // 是否批处理验证
- protected $patchValidate = false;
- // 链操作方法列表
- protected $methods = array('strict','order','alias','having','group','lock','distinct','auto','filter','validate','result','token','index','force');
- /**
- * 架构函数
- * 取得DB类的实例对象 字段检查
- * @access public
- * @param string $name 模型名称
- * @param string $tablePrefix 表前缀
- * @param mixed $connection 数据库连接信息
- */
- public function __construct($name='',$tablePrefix='',$connection='') {
- // 模型初始化
- $this->_initialize();
- // 获取模型名称
- if(!empty($name)) {
- if(strpos($name,'.')) { // 支持 数据库名.模型名的 定义
- list($this->dbName,$this->name) = explode('.',$name);
- }else{
- $this->name = $name;
- }
- }elseif(empty($this->name)){
- $this->name = $this->getModelName();
- }
- // 设置表前缀
- if(is_null($tablePrefix)) {// 前缀为Null表示没有前缀
- $this->tablePrefix = '';
- }elseif('' != $tablePrefix) {
- $this->tablePrefix = $tablePrefix;
- }elseif(!isset($this->tablePrefix)){
- $this->tablePrefix = C('DB_PREFIX');
- }
- // 数据库初始化操作
- // 获取数据库操作对象
- // 当前模型有独立的数据库连接信息
- $this->db(0,empty($this->connection)?$connection:$this->connection,true);
- }
- /**
- * 自动检测数据表信息
- * @access protected
- * @return void
- */
- protected function _checkTableInfo() {
- // 如果不是Model类 自动记录数据表信息
- // 只在第一次执行记录
- if(empty($this->fields)) {
- // 如果数据表字段没有定义则自动获取
- if(C('DB_FIELDS_CACHE')) {
- $db = $this->dbName?:C('DB_NAME');
- $fields = F('_fields/'.strtolower($db.'.'.$this->tablePrefix.$this->name));
- if($fields) {
- $this->fields = $fields;
- if(!empty($fields['_pk'])){
- $this->pk = $fields['_pk'];
- }
- return ;
- }
- }
- // 每次都会读取数据表信息
- $this->flush();
- }
- }
- /**
- * 获取字段信息并缓存
- * @access public
- * @return void
- */
- public function flush() {
- // 缓存不存在则查询数据表信息
- $this->db->setModel($this->name);
- $fields = $this->db->getFields($this->getTableName());
- if(!$fields) { // 无法获取字段信息
- return false;
- }
- $this->fields = array_keys($fields);
- unset($this->fields['_pk']);
- foreach ($fields as $key=>$val){
- // 记录字段类型
- $type[$key] = $val['type'];
- if($val['primary']) {
- // 增加复合主键支持
- if (isset($this->fields['_pk']) && $this->fields['_pk'] != null) {
- if (is_string($this->fields['_pk'])) {
- $this->pk = array($this->fields['_pk']);
- $this->fields['_pk'] = $this->pk;
- }
- $this->pk[] = $key;
- $this->fields['_pk'][] = $key;
- } else {
- $this->pk = $key;
- $this->fields['_pk'] = $key;
- }
- if($val['autoinc']) $this->autoinc = true;
- }
- }
- // 记录字段类型信息
- $this->fields['_type'] = $type;
- // 2008-3-7 增加缓存开关控制
- if(C('DB_FIELDS_CACHE')){
- // 永久缓存数据表信息
- $db = $this->dbName?:C('DB_NAME');
- F('_fields/'.strtolower($db.'.'.$this->tablePrefix.$this->name),$this->fields);
- }
- }
- /**
- * 设置数据对象的值
- * @access public
- * @param string $name 名称
- * @param mixed $value 值
- * @return void
- */
- public function __set($name,$value) {
- // 设置数据对象属性
- $this->data[$name] = $value;
- }
- /**
- * 获取数据对象的值
- * @access public
- * @param string $name 名称
- * @return mixed
- */
- public function __get($name) {
- return isset($this->data[$name])?$this->data[$name]:null;
- }
- /**
- * 检测数据对象的值
- * @access public
- * @param string $name 名称
- * @return boolean
- */
- public function __isset($name) {
- return isset($this->data[$name]);
- }
- /**
- * 销毁数据对象的值
- * @access public
- * @param string $name 名称
- * @return void
- */
- public function __unset($name) {
- unset($this->data[$name]);
- }
- /**
- * 利用__call方法实现一些特殊的Model方法
- * @access public
- * @param string $method 方法名称
- * @param array $args 调用参数
- * @return mixed
- */
- public function __call($method,$args) {
- if(in_array(strtolower($method),$this->methods,true)) {
- // 连贯操作的实现
- $this->options[strtolower($method)] = $args[0];
- return $this;
- }elseif(in_array(strtolower($method),array('count','sum','min','max','avg'),true)){
- // 统计查询的实现
- $field = isset($args[0])?$args[0]:'*';
- return $this->getField(strtoupper($method).'('.$field.') AS tp_'.$method);
- }elseif(strtolower(substr($method,0,5))=='getby') {
- // 根据某个字段获取记录
- $field = parse_name(substr($method,5));
- $where[$field] = $args[0];
- return $this->where($where)->find();
- }elseif(strtolower(substr($method,0,10))=='getfieldby') {
- // 根据某个字段获取记录的某个值
- $name = parse_name(substr($method,10));
- $where[$name] =$args[0];
- return $this->where($where)->getField($args[1]);
- }elseif(isset($this->_scope[$method])){// 命名范围的单独调用支持
- return $this->scope($method,$args[0]);
- }else{
- E(__CLASS__.':'.$method.L('_METHOD_NOT_EXIST_'));
- return;
- }
- }
- // 回调方法 初始化模型
- protected function _initialize() {}
- /**
- * 对保存到数据库的数据进行处理
- * @access protected
- * @param mixed $data 要操作的数据
- * @return boolean
- */
- protected function _facade($data) {
- // 检查数据字段合法性
- if(!empty($this->fields)) {
- if(!empty($this->options['field'])) {
- $fields = $this->options['field'];
- unset($this->options['field']);
- if(is_string($fields)) {
- $fields = explode(',',$fields);
- }
- }else{
- $fields = $this->fields;
- }
- foreach ($data as $key=>$val){
- if(!in_array($key,$fields,true)){
- if(!empty($this->options['strict'])){
- E(L('_DATA_TYPE_INVALID_').':['.$key.'=>'.$val.']');
- }
- unset($data[$key]);
- }elseif(is_scalar($val)) {
- // 字段类型检查 和 强制转换
- $this->_parseType($data,$key);
- }
- }
- }
-
- // 安全过滤
- if(!empty($this->options['filter'])) {
- $data = array_map($this->options['filter'],$data);
- unset($this->options['filter']);
- }
- $this->_before_write($data);
- return $data;
- }
- // 写入数据前的回调方法 包括新增和更新
- protected function _before_write(&$data) {}
- /**
- * 新增数据
- * @access public
- * @param mixed $data 数据
- * @param array $options 表达式
- * @param boolean $replace 是否replace
- * @return mixed
- */
- public function add($data='',$options=array(),$replace=false) {
- if(empty($data)) {
- // 没有传递数据,获取当前数据对象的值
- if(!empty($this->data)) {
- $data = $this->data;
- // 重置数据
- $this->data = array();
- }else{
- $this->error = L('_DATA_TYPE_INVALID_');
- return false;
- }
- }
- // 数据处理
- $data = $this->_facade($data);
- // 分析表达式
- $options = $this->_parseOptions($options);
- if(false === $this->_before_insert($data,$options)) {
- return false;
- }
- // 写入数据到数据库
- $result = $this->db->insert($data,$options,$replace);
- if(false !== $result && is_numeric($result)) {
- $pk = $this->getPk();
- // 增加复合主键支持
- if (is_array($pk)) return $result;
- $insertId = $this->getLastInsID();
- if($insertId) {
- // 自增主键返回插入ID
- $data[$pk] = $insertId;
- if(false === $this->_after_insert($data,$options)) {
- return false;
- }
- return $insertId;
- }
- if(false === $this->_after_insert($data,$options)) {
- return false;
- }
- }
- return $result;
- }
- // 插入数据前的回调方法
- protected function _before_insert(&$data,$options) {}
- // 插入成功后的回调方法
- protected function _after_insert($data,$options) {}
- public function addAll($dataList,$options=array(),$replace=false){
- if(empty($dataList)) {
- $this->error = L('_DATA_TYPE_INVALID_');
- return false;
- }
- // 数据处理
- foreach ($dataList as $key=>$data){
- $dataList[$key] = $this->_facade($data);
- }
- // 分析表达式
- $options = $this->_parseOptions($options);
- // 写入数据到数据库
- $result = $this->db->insertAll($dataList,$options,$replace);
- if(false !== $result ) {
- $insertId = $this->getLastInsID();
- if($insertId) {
- return $insertId;
- }
- }
- return $result;
- }
- /**
- * 保存数据
- * @access public
- * @param mixed $data 数据
- * @param array $options 表达式
- * @return boolean
- */
- public function save($data='',$options=array()) {
- if(empty($data)) {
- // 没有传递数据,获取当前数据对象的值
- if(!empty($this->data)) {
- $data = $this->data;
- // 重置数据
- $this->data = array();
- }else{
- $this->error = L('_DATA_TYPE_INVALID_');
- return false;
- }
- }
- // 数据处理
- $data = $this->_facade($data);
- if(empty($data)){
- // 没有数据则不执行
- $this->error = L('_DATA_TYPE_INVALID_');
- return false;
- }
- // 分析表达式
- $options = $this->_parseOptions($options);
- $pk = $this->getPk();
- if(!isset($options['where']) ) {
- // 如果存在主键数据 则自动作为更新条件
- if (is_string($pk) && isset($data[$pk])) {
- $where[$pk] = $data[$pk];
- unset($data[$pk]);
- } elseif (is_array($pk)) {
- // 增加复合主键支持
- foreach ($pk as $field) {
- if(isset($data[$field])) {
- $where[$field] = $data[$field];
- } else {
- // 如果缺少复合主键数据则不执行
- $this->error = L('_OPERATION_WRONG_');
- return false;
- }
- unset($data[$field]);
- }
- }
- if(!isset($where)){
- // 如果没有任何更新条件则不执行
- $this->error = L('_OPERATION_WRONG_');
- return false;
- }else{
- $options['where'] = $where;
- }
- }
- if(is_array($options['where']) && isset($options['where'][$pk])){
- $pkValue = $options['where'][$pk];
- }
- if(false === $this->_before_update($data,$options)) {
- return false;
- }
- $result = $this->db->update($data,$options);
- if(false !== $result && is_numeric($result)) {
- if(isset($pkValue)) $data[$pk] = $pkValue;
- $this->_after_update($data,$options);
- }
- return $result;
- }
- // 更新数据前的回调方法
- protected function _before_update(&$data,$options) {}
- // 更新成功后的回调方法
- protected function _after_update($data,$options) {}
- /**
- * 删除数据
- * @access public
- * @param mixed $options 表达式
- * @return mixed
- */
- public function delete($options=array()) {
- $pk = $this->getPk();
- if(empty($options) && empty($this->options['where'])) {
- // 如果删除条件为空 则删除当前数据对象所对应的记录
- if(!empty($this->data) && isset($this->data[$pk]))
- return $this->delete($this->data[$pk]);
- else
- return false;
- }
- if(is_numeric($options) || is_string($options)) {
- // 根据主键删除记录
- if(strpos($options,',')) {
- $where[$pk] = array('IN', $options);
- }else{
- $where[$pk] = $options;
- }
- $options = array();
- $options['where'] = $where;
- }
- // 根据复合主键删除记录
- if (is_array($options) && (count($options) > 0) && is_array($pk)) {
- $count = 0;
- foreach (array_keys($options) as $key) {
- if (is_int($key)) $count++;
- }
- if ($count == count($pk)) {
- $i = 0;
- foreach ($pk as $field) {
- $where[$field] = $options[$i];
- unset($options[$i++]);
- }
- $options['where'] = $where;
- } else {
- return false;
- }
- }
- // 分析表达式
- $options = $this->_parseOptions($options);
- if(empty($options['where'])){
- // 如果条件为空 不进行删除操作 除非设置 1=1
- return false;
- }
- if(is_array($options['where']) && isset($options['where'][$pk])){
- $pkValue = $options['where'][$pk];
- }
- if(false === $this->_before_delete($options)) {
- return false;
- }
- $result = $this->db->delete($options);
- if(false !== $result && is_numeric($result)) {
- $data = array();
- if(isset($pkValue)) $data[$pk] = $pkValue;
- $this->_after_delete($data,$options);
- }
- // 返回删除记录个数
- return $result;
- }
- // 删除数据前的回调方法
- protected function _before_delete($options) {}
- // 删除成功后的回调方法
- protected function _after_delete($data,$options) {}
- /**
- * 查询数据集
- * @access public
- * @param array $options 表达式参数
- * @return mixed
- */
- public function select($options=array()) {
- $pk = $this->getPk();
- if(is_string($options) || is_numeric($options)) {
- // 根据主键查询
- if(strpos($options,',')) {
- $where[$pk] = array('IN',$options);
- }else{
- $where[$pk] = $options;
- }
- $options = array();
- $options['where'] = $where;
- }elseif (is_array($options) && (count($options) > 0) && is_array($pk)) {
- // 根据复合主键查询
- $count = 0;
- foreach (array_keys($options) as $key) {
- if (is_int($key)) $count++;
- }
- if ($count == count($pk)) {
- $i = 0;
- foreach ($pk as $field) {
- $where[$field] = $options[$i];
- unset($options[$i++]);
- }
- $options['where'] = $where;
- } else {
- return false;
- }
- } elseif(false === $options){ // 用于子查询 不查询只返回SQL
- $options = array();
- // 分析表达式
- $options = $this->_parseOptions($options);
- return '( '.$this->fetchSql(true)->select($options).' )';
- }
- // 分析表达式
- $options = $this->_parseOptions($options);
- // 判断查询缓存
- if(isset($options['cache'])){
- $cache = $options['cache'];
- $key = is_string($cache['key'])?$cache['key']:md5(serialize($options));
- $data = S($key,'',$cache);
- if(false !== $data){
- return $data;
- }
- }
- $resultSet = $this->db->select($options);
- if(false === $resultSet) {
- return false;
- }
- if(empty($resultSet)) { // 查询结果为空
- return null;
- }
- if(is_string($resultSet)){
- return $resultSet;
- }
- $resultSet = array_map(array($this,'_read_data'),$resultSet);
- $this->_after_select($resultSet,$options);
- if(isset($options['index'])){ // 对数据集进行索引
- $index = explode(',',$options['index']);
- foreach ($resultSet as $result){
- $_key = $result[$index[0]];
- if(isset($index[1]) && isset($result[$index[1]])){
- $cols[$_key] = $result[$index[1]];
- }else{
- $cols[$_key] = $result;
- }
- }
- $resultSet = $cols;
- }
- if(isset($cache)){
- S($key,$resultSet,$cache);
- }
- return $resultSet;
- }
- // 查询成功后的回调方法
- protected function _after_select(&$resultSet,$options) {}
- /**
- * 分析表达式
- * @access protected
- * @param array $options 表达式参数
- * @return array
- */
- protected function _parseOptions($options=array()) {
- if(is_array($options))
- $options = array_merge($this->options,$options);
- if(!isset($options['table'])){
- // 自动获取表名
- $options['table'] = $this->getTableName();
- $fields = $this->fields;
- }else{
- // 指定数据表 则重新获取字段列表 但不支持类型检测
- $fields = $this->getDbFields();
- }
- // 数据表别名
- if(!empty($options['alias'])) {
- $options['table'] .= ' '.$options['alias'];
- }
- // 记录操作的模型名称
- $options['model'] = $this->name;
- // 字段类型验证
- if(isset($options['where']) && is_array($options['where']) && !empty($fields) && !isset($options['join'])) {
- // 对数组查询条件进行字段类型检查
- foreach ($options['where'] as $key=>$val){
- $key = trim($key);
- if(in_array($key,$fields,true)){
- if(is_scalar($val)) {
- $this->_parseType($options['where'],$key);
- }
- }elseif(!is_numeric($key) && '_' != substr($key,0,1) && false === strpos($key,'.') && false === strpos($key,'(') && false === strpos($key,'|') && false === strpos($key,'&')){
- if(!empty($this->options['strict'])){
- E(L('_ERROR_QUERY_EXPRESS_').':['.$key.'=>'.$val.']');
- }
- unset($options['where'][$key]);
- }
- }
- }
- // 查询过后清空sql表达式组装 避免影响下次查询
- $this->options = array();
- // 表达式过滤
- $this->_options_filter($options);
- return $options;
- }
- // 表达式过滤回调方法
- protected function _options_filter(&$options) {}
- /**
- * 数据类型检测
- * @access protected
- * @param mixed $data 数据
- * @param string $key 字段名
- * @return void
- */
- protected function _parseType(&$data,$key) {
- if(!isset($this->options['bind'][':'.$key]) && isset($this->fields['_type'][$key])){
- $fieldType = strtolower($this->fields['_type'][$key]);
- if(false !== strpos($fieldType,'enum')){
- // 支持ENUM类型优先检测
- }elseif(false === strpos($fieldType,'bigint') && false !== strpos($fieldType,'int')) {
- $data[$key] = intval($data[$key]);
- }elseif(false !== strpos($fieldType,'float') || false !== strpos($fieldType,'double')){
- $data[$key] = floatval($data[$key]);
- }elseif(false !== strpos($fieldType,'bool')){
- $data[$key] = (bool)$data[$key];
- }
- }
- }
- /**
- * 数据读取后的处理
- * @access protected
- * @param array $data 当前数据
- * @return array
- */
- protected function _read_data($data) {
- // 检查字段映射
- if(!empty($this->_map) && C('READ_DATA_MAP')) {
- foreach ($this->_map as $key=>$val){
- if(isset($data[$val])) {
- $data[$key] = $data[$val];
- unset($data[$val]);
- }
- }
- }
- return $data;
- }
- /**
- * 查询数据
- * @access public
- * @param mixed $options 表达式参数
- * @return mixed
- */
- public function find($options=array()) {
- if(is_numeric($options) || is_string($options)) {
- $where[$this->getPk()] = $options;
- $options = array();
- $options['where'] = $where;
- }
- // 根据复合主键查找记录
- $pk = $this->getPk();
- if (is_array($options) && (count($options) > 0) && is_array($pk)) {
- // 根据复合主键查询
- $count = 0;
- foreach (array_keys($options) as $key) {
- if (is_int($key)) $count++;
- }
- if ($count == count($pk)) {
- $i = 0;
- foreach ($pk as $field) {
- $where[$field] = $options[$i];
- unset($options[$i++]);
- }
- $options['where'] = $where;
- } else {
- return false;
- }
- }
- // 总是查找一条记录
- $options['limit'] = 1;
- // 分析表达式
- $options = $this->_parseOptions($options);
- // 判断查询缓存
- if(isset($options['cache'])){
- $cache = $options['cache'];
- $key = is_string($cache['key'])?$cache['key']:md5(serialize($options));
- $data = S($key,'',$cache);
- if(false !== $data){
- $this->data = $data;
- return $data;
- }
- }
- $resultSet = $this->db->select($options);
- if(false === $resultSet) {
- return false;
- }
- if(empty($resultSet)) {// 查询结果为空
- return null;
- }
- if(is_string($resultSet)){
- return $resultSet;
- }
- // 读取数据后的处理
- $data = $this->_read_data($resultSet[0]);
- $this->_after_find($data,$options);
- $this->data = $data;
- if(isset($cache)){
- S($key,$data,$cache);
- }
- return $this->data;
- }
- // 查询成功的回调方法
- protected function _after_find(&$result,$options) {}
- /**
- * 设置记录的某个字段值
- * 支持使用数据库字段和方法
- * @access public
- * @param string|array $field 字段名
- * @param string $value 字段值
- * @return boolean
- */
- public function setField($field,$value='') {
- if(is_array($field)) {
- $data = $field;
- }else{
- $data[$field] = $value;
- }
- return $this->save($data);
- }
- /**
- * 字段值增长
- * @access public
- * @param string $field 字段名
- * @param integer $step 增长值
- * @param integer $lazyTime 延时时间(s)
- * @return boolean
- */
- public function setInc($field,$step=1) {
- return $this->setField($field,array('exp',$field.'+'.$step));
- }
- /**
- * 字段值减少
- * @access public
- * @param string $field 字段名
- * @param integer $step 减少值
- * @param integer $lazyTime 延时时间(s)
- * @return boolean
- */
- public function setDec($field,$step=1) {
- return $this->setField($field,array('exp',$field.'-'.$step));
- }
- /**
- * 获取一条记录的某个字段值
- * @access public
- * @param string $field 字段名
- * @param string $spea 字段数据间隔符号 NULL返回数组
- * @return mixed
- */
- public function getField($field,$sepa=null) {
- $options['field'] = $field;
- $options = $this->_parseOptions($options);
- // 判断查询缓存
- if(isset($options['cache'])){
- $cache = $options['cache'];
- $key = is_string($cache['key'])?$cache['key']:md5($sepa.serialize($options));
- $data = S($key,'',$cache);
- if(false !== $data){
- return $data;
- }
- }
- $field = trim($field);
- if(strpos($field,',') && false !== $sepa) { // 多字段
- if(!isset($options['limit'])){
- $options['limit'] = is_numeric($sepa)?$sepa:'';
- }
- $resultSet = $this->db->select($options);
- if(!empty($resultSet)) {
- $_field = explode(',', $field);
- $field = array_keys($resultSet[0]);
- $key1 = array_shift($field);
- $key2 = array_shift($field);
- $cols = array();
- $count = count($_field);
- foreach ($resultSet as $result){
- $name = $result[$key1];
- if(2==$count) {
- $cols[$name] = $result[$key2];
- }else{
- $cols[$name] = is_string($sepa)?implode($sepa,array_slice($result,1)):$result;
- }
- }
- if(isset($cache)){
- S($key,$cols,$cache);
- }
- return $cols;
- }
- }else{ // 查找一条记录
- // 返回数据个数
- if(true !== $sepa) {// 当sepa指定为true的时候 返回所有数据
- $options['limit'] = is_numeric($sepa)?$sepa:1;
- }
- $result = $this->db->select($options);
- if(!empty($result)) {
- if(true !== $sepa && 1==$options['limit']) {
- $data = reset($result[0]);
- if(isset($cache)){
- S($key,$data,$cache);
- }
- return $data;
- }
- foreach ($result as $val){
- $array[] = $val[$field];
- }
- if(isset($cache)){
- S($key,$array,$cache);
- }
- return $array;
- }
- }
- return null;
- }
- /**
- * 创建数据对象 但不保存到数据库
- * @access public
- * @param mixed $data 创建数据
- * @return mixed
- */
- public function create($data='') {
- // 如果没有传值默认取POST数据
- if(empty($data)) {
- $data = I('post.');
- }elseif(is_object($data)){
- $data = get_object_vars($data);
- }
- // 验证数据
- if(empty($data) || !is_array($data)) {
- $this->error = L('_DATA_TYPE_INVALID_');
- return false;
- }
- // 检测提交字段的合法性
- if(isset($this->options['field'])) { // $this->field('field1,field2...')->create()
- $fields = $this->options['field'];
- unset($this->options['field']);
- }
- if(isset($fields)) {
- if(is_string($fields)) {
- $fields = explode(',',$fields);
- }
- }
- // 验证完成生成数据对象
- if($this->autoCheckFields) { // 开启字段检测 则过滤非法字段数据
- $fields = $this->getDbFields();
- foreach ($data as $key=>$val){
- if(!in_array($key,$fields)) {
- unset($data[$key]);
- }elseif(MAGIC_QUOTES_GPC && is_string($val)){
- $data[$key] = stripslashes($val);
- }
- }
- }
- // 赋值当前数据对象
- $this->data = $data;
- // 返回创建的数据以供其他调用
- return $data;
- }
- /**
- * 使用正则验证数据
- * @access public
- * @param string $value 要验证的数据
- * @param string $rule 验证规则
- * @return boolean
- */
- public function regex($value,$rule) {
- $validate = array(
- 'require' => '/\S+/',
- 'email' => '/^\w+([-+.]\w+)*@\w+([-.]\w+)*\.\w+([-.]\w+)*$/',
- 'url' => '/^http(s?):\/\/(?:[A-za-z0-9-]+\.)+[A-za-z]{2,4}(:\d+)?(?:[\/\?#][\/=\?%\-&~`@[\]\':+!\.#\w]*)?$/',
- 'currency' => '/^\d+(\.\d+)?$/',
- 'number' => '/^\d+$/',
- 'zip' => '/^\d{6}$/',
- 'integer' => '/^[-\+]?\d+$/',
- 'double' => '/^[-\+]?\d+(\.\d+)?$/',
- 'english' => '/^[A-Za-z]+$/',
- );
- // 检查是否有内置的正则表达式
- if(isset($validate[strtolower($rule)]))
- $rule = $validate[strtolower($rule)];
- return preg_match($rule,$value)===1;
- }
- /**
- * 验证数据 支持 in between equal length regex expire ip_allow ip_deny
- * @access public
- * @param string $value 验证数据
- * @param mixed $rule 验证表达式
- * @param string $type 验证方式 默认为正则验证
- * @return boolean
- */
- public function check($value,$rule,$type='regex'){
- $type = strtolower(trim($type));
- switch($type) {
- case 'in': // 验证是否在某个指定范围之内 逗号分隔字符串或者数组
- case 'notin':
- $range = is_array($rule)? $rule : explode(',',$rule);
- return $type == 'in' ? in_array($value ,$range) : !in_array($value ,$range);
- case 'between': // 验证是否在某个范围
- case 'notbetween': // 验证是否不在某个范围
- if (is_array($rule)){
- $min = $rule[0];
- $max = $rule[1];
- }else{
- list($min,$max) = explode(',',$rule);
- }
- return $type == 'between' ? $value>=$min && $value<=$max : $value<$min || $value>$max;
- case 'equal': // 验证是否等于某个值
- case 'notequal': // 验证是否等于某个值
- return $type == 'equal' ? $value == $rule : $value != $rule;
- case 'length': // 验证长度
- $length = mb_strlen($value,'utf-8'); // 当前数据长度
- if(strpos($rule,',')) { // 长度区间
- list($min,$max) = explode(',',$rule);
- return $length >= $min && $length <= $max;
- }else{// 指定长度
- return $length == $rule;
- }
- case 'expire':
- list($start,$end) = explode(',',$rule);
- if(!is_numeric($start)) $start = strtotime($start);
- if(!is_numeric($end)) $end = strtotime($end);
- return NOW_TIME >= $start && NOW_TIME <= $end;
- case 'ip_allow': // IP 操作许可验证
- return in_array(get_client_ip(),explode(',',$rule));
- case 'ip_deny': // IP 操作禁止验证
- return !in_array(get_client_ip(),explode(',',$rule));
- case 'regex':
- default: // 默认使用正则验证 可以使用验证类中定义的验证名称
- // 检查附加规则
- return $this->regex($value,$rule);
- }
- }
- /**
- * SQL查询
- * @access public
- * @param string $sql SQL指令
- * @return mixed
- */
- public function query($sql) {
- return $this->db->query($sql);
- }
- /**
- * 执行SQL语句
- * @access public
- * @param string $sql SQL指令
- * @return false | integer
- */
- public function execute($sql) {
- return $this->db->execute($sql);
- }
- /**
- * 切换当前的数据库连接
- * @access public
- * @param integer $linkNum 连接序号
- * @param mixed $config 数据库连接信息
- * @param boolean $force 强制重新连接
- * @return Model
- */
- public function db($linkNum='',$config='',$force=false) {
- if('' === $linkNum && $this->db) {
- return $this->db;
- }
- if(!isset($this->_db[$linkNum]) || $force ) {
- // 创建一个新的实例
- if(!empty($config) && is_string($config) && false === strpos($config,'/')) { // 支持读取配置参数
- $config = C($config);
- }
- $this->_db[$linkNum] = Db::getInstance($config);
- }elseif(NULL === $config){
- $this->_db[$linkNum]->close(); // 关闭数据库连接
- unset($this->_db[$linkNum]);
- return ;
- }
- // 切换数据库连接
- $this->db = $this->_db[$linkNum];
- $this->_after_db();
- // 字段检测
- if(!empty($this->name) && $this->autoCheckFields) $this->_checkTableInfo();
- return $this;
- }
- // 数据库切换后回调方法
- protected function _after_db() {}
- /**
- * 得到当前的数据对象名称
- * @access public
- * @return string
- */
- public function getModelName() {
- if(empty($this->name)){
- $name = substr(get_class($this),0,-strlen(C('DEFAULT_M_LAYER')));
- if ( $pos = strrpos($name,'\\') ) {//有命名空间
- $this->name = substr($name,$pos+1);
- }else{
- $this->name = $name;
- }
- }
- return $this->name;
- }
- /**
- * 得到完整的数据表名
- * @access public
- * @return string
- */
- public function getTableName() {
- if(empty($this->trueTableName)) {
- $tableName = !empty($this->tablePrefix) ? $this->tablePrefix : '';
- if(!empty($this->tableName)) {
- $tableName .= $this->tableName;
- }else{
- $tableName .= parse_name($this->name);
- }
- $this->trueTableName = strtolower($tableName);
- }
- return (!empty($this->dbName)?$this->dbName.'.':'').$this->trueTableName;
- }
- /**
- * 启动事务
- * @access public
- * @return void
- */
- public function startTrans() {
- $this->commit();
- $this->db->startTrans();
- return ;
- }
- /**
- * 提交事务
- * @access public
- * @return boolean
- */
- public function commit() {
- return $this->db->commit();
- }
- /**
- * 事务回滚
- * @access public
- * @return boolean
- */
- public function rollback() {
- return $this->db->rollback();
- }
- /**
- * 返回模型的错误信息
- * @access public
- * @return string
- */
- public function getError(){
- return $this->error;
- }
- /**
- * 返回数据库的错误信息
- * @access public
- * @return string
- */
- public function getDbError() {
- return $this->db->getError();
- }
- /**
- * 返回最后插入的ID
- * @access public
- * @return string
- */
- public function getLastInsID() {
- return $this->db->getLastInsID();
- }
- /**
- * 返回最后执行的sql语句
- * @access public
- * @return string
- */
- public function getLastSql() {
- return $this->db->getLastSql($this->name);
- }
- // 鉴于getLastSql比较常用 增加_sql 别名
- public function _sql(){
- return $this->getLastSql();
- }
- /**
- * 获取主键名称
- * @access public
- * @return string
- */
- public function getPk() {
- return $this->pk;
- }
- /**
- * 获取数据表字段信息
- * @access public
- * @return array
- */
- public function getDbFields(){
- if(isset($this->options['table'])) {// 动态指定表名
- if(is_array($this->options['table'])){
- $table = key($this->options['table']);
- }else{
- $table = $this->options['table'];
- }
- $fields = $this->db->getFields($table);
- return $fields ? array_keys($fields) : false;
- }
- if($this->fields) {
- $fields = $this->fields;
- unset($fields['_type'],$fields['_pk']);
- return $fields;
- }
- return false;
- }
- /**
- * 设置数据对象值
- * @access public
- * @param mixed $data 数据
- * @return Model
- */
- public function data($data=''){
- if('' === $data && !empty($this->data)) {
- return $this->data;
- }
- if(is_object($data)){
- $data = get_object_vars($data);
- }elseif(is_string($data)){
- parse_str($data,$data);
- }elseif(!is_array($data)){
- E(L('_DATA_TYPE_INVALID_'));
- }
- $this->data = $data;
- return $this;
- }
- /**
- * 指定当前的数据表
- * @access public
- * @param mixed $table
- * @return Model
- */
- public function table($table) {
- $prefix = $this->tablePrefix;
- if(is_array($table)) {
- $this->options['table'] = $table;
- }elseif(!empty($table)) {
- //将__TABLE_NAME__替换成带前缀的表名
- $table = preg_replace_callback("/__([A-Z0-9_-]+)__/sU", function($match) use($prefix){ return $prefix.strtolower($match[1]);}, $table);
- $this->options['table'] = $table;
- }
- return $this;
- }
- /**
- * USING支持 用于多表删除
- * @access public
- * @param mixed $using
- * @return Model
- */
- public function using($using){
- $prefix = $this->tablePrefix;
- if(is_array($using)) {
- $this->options['using'] = $using;
- }elseif(!empty($using)) {
- //将__TABLE_NAME__替换成带前缀的表名
- $using = preg_replace_callback("/__([A-Z0-9_-]+)__/sU", function($match) use($prefix){ return $prefix.strtolower($match[1]);}, $using);
- $this->options['using'] = $using;
- }
- return $this;
- }
- /**
- * 查询SQL组装 join
- * @access public
- * @param mixed $join
- * @param string $type JOIN类型
- * @return Model
- */
- public function join($join,$type='INNER') {
- $prefix = $this->tablePrefix;
- if(is_array($join)) {
- foreach ($join as $key=>&$_join){
- $_join = preg_replace_callback("/__([A-Z0-9_-]+)__/sU", function($match) use($prefix){ return $prefix.strtolower($match[1]);}, $_join);
- $_join = false !== stripos($_join,'JOIN')? $_join : $type.' JOIN ' .$_join;
- }
- $this->options['join'] = $join;
- }elseif(!empty($join)) {
- //将__TABLE_NAME__字符串替换成带前缀的表名
- $join = preg_replace_callback("/__([A-Z0-9_-]+)__/sU", function($match) use($prefix){ return $prefix.strtolower($match[1]);}, $join);
- $this->options['join'][] = false !== stripos($join,'JOIN')? $join : $type.' JOIN '.$join;
- }
- return $this;
- }
- /**
- * 查询SQL组装 union
- * @access public
- * @param mixed $union
- * @param boolean $all
- * @return Model
- */
- public function union($union,$all=false) {
- if(empty($union)) return $this;
- if($all) {
- $this->options['union']['_all'] = true;
- }
- if(is_object($union)) {
- $union = get_object_vars($union);
- }
- // 转换union表达式
- if(is_string($union) ) {
- $prefix = $this->tablePrefix;
- //将__TABLE_NAME__字符串替换成带前缀的表名
- $options = preg_replace_callback("/__([A-Z0-9_-]+)__/sU", function($match) use($prefix){ return $prefix.strtolower($match[1]);}, $union);
- }elseif(is_array($union)){
- if(isset($union[0])) {
- $this->options['union'] = array_merge($this->options['union'],$union);
- return $this;
- }else{
- $options = $union;
- }
- }else{
- E(L('_DATA_TYPE_INVALID_'));
- }
- $this->options['union'][] = $options;
- return $this;
- }
- /**
- * 查询缓存
- * @access public
- * @param mixed $key
- * @param integer $expire
- * @param string $type
- * @return Model
- */
- public function cache($key=true,$expire=null,$type=''){
- // 增加快捷调用方式 cache(10) 等同于 cache(true, 10)
- if(is_numeric($key) && is_null($expire)){
- $expire = $key;
- $key = true;
- }
- if(false !== $key)
- $this->options['cache'] = array('key'=>$key,'expire'=>$expire,'type'=>$type);
- return $this;
- }
- /**
- * 指定查询字段 支持字段排除
- * @access public
- * @param mixed $field
- * @param boolean $except 是否排除
- * @return Model
- */
- public function field($field,$except=false){
- if(true === $field) {// 获取全部字段
- $fields = $this->getDbFields();
- $field = $fields?:'*';
- }elseif($except) {// 字段排除
- if(is_string($field)) {
- $field = explode(',',$field);
- }
- $fields = $this->getDbFields();
- $field = $fields?array_diff($fields,$field):$field;
- }
- $this->options['field'] = $field;
- return $this;
- }
- /**
- * 调用命名范围
- * @access public
- * @param mixed $scope 命名范围名称 支持多个 和直接定义
- * @param array $args 参数
- * @return Model
- */
- public function scope($scope='',$args=NULL){
- if('' === $scope) {
- if(isset($this->_scope['default'])) {
- // 默认的命名范围
- $options = $this->_scope['default'];
- }else{
- return $this;
- }
- }elseif(is_string($scope)){ // 支持多个命名范围调用 用逗号分割
- $scopes = explode(',',$scope);
- $options = array();
- foreach ($scopes as $name){
- if(!isset($this->_scope[$name])) continue;
- $options = array_merge($options,$this->_scope[$name]);
- }
- if(!empty($args) && is_array($args)) {
- $options = array_merge($options,$args);
- }
- }elseif(is_array($scope)){ // 直接传入命名范围定义
- $options = $scope;
- }
-
- if(is_array($options) && !empty($options)){
- $this->options = array_merge($this->options,array_change_key_case($options));
- }
- return $this;
- }
- /**
- * 指定查询条件 支持安全过滤
- * @access public
- * @param mixed $where 条件表达式
- * @param mixed $parse 预处理参数
- * @return Model
- */
- public function where($where,$parse=null){
- if(!is_null($parse) && is_string($where)) {
- if(!is_array($parse)) {
- $parse = func_get_args();
- array_shift($parse);
- }
- $parse = array_map(array($this->db,'escapeString'),$parse);
- $where = vsprintf($where,$parse);
- }elseif(is_object($where)){
- $where = get_object_vars($where);
- }
- if(is_string($where) && '' != $where){
- $map = array();
- $map['_string'] = $where;
- $where = $map;
- }
- if(isset($this->options['where'])){
- $this->options['where'] = array_merge($this->options['where'],$where);
- }else{
- $this->options['where'] = $where;
- }
-
- return $this;
- }
- /**
- * 指定查询数量
- * @access public
- * @param mixed $offset 起始位置
- * @param mixed $length 查询数量
- * @return Model
- */
- public function limit($offset,$length=null){
- if(is_null($length) && strpos($offset,',')){
- list($offset,$length) = explode(',',$offset);
- }
- $this->options['limit'] = intval($offset).( $length? ','.intval($length) : '' );
- return $this;
- }
- /**
- * 指定分页
- * @access public
- * @param mixed $page 页数
- * @param mixed $listRows 每页数量
- * @return Model
- */
- public function page($page,$listRows=null){
- if(is_null($listRows) && strpos($page,',')){
- list($page,$listRows) = explode(',',$page);
- }
- $this->options['page'] = array(intval($page),intval($listRows));
- return $this;
- }
- /**
- * 查询注释
- * @access public
- * @param string $comment 注释
- * @return Model
- */
- public function comment($comment){
- $this->options['comment'] = $comment;
- return $this;
- }
- /**
- * 获取执行的SQL语句
- * @access public
- * @param boolean $fetch 是否返回sql
- * @return Model
- */
- public function fetchSql($fetch){
- $this->options['fetch_sql'] = $fetch;
- return $this;
- }
- /**
- * 参数绑定
- * @access public
- * @param string $key 参数名
- * @param mixed $value 绑定的变量及绑定参数
- * @return Model
- */
- public function bind($key,$value=false) {
- if(is_array($key)){
- $this->options['bind'] = $key;
- }else{
- $num = func_num_args();
- if($num>2){
- $params = func_get_args();
- array_shift($params);
- $this->options['bind'][$key] = $params;
- }else{
- $this->options['bind'][$key] = $value;
- }
- }
- return $this;
- }
- /**
- * 设置模型的属性值
- * @access public
- * @param string $name 名称
- * @param mixed $value 值
- * @return Model
- */
- public function setProperty($name,$value) {
- if(property_exists($this,$name))
- $this->$name = $value;
- return $this;
- }
- }
|