mirror of
https://e.coding.net/circlecloud/MinecraftAccount.git
synced 2025-11-26 21:46:09 +00:00
j502647092
595
ThinkPHP/Library/Think/Model/AdvModel.class.php
Normal file
595
ThinkPHP/Library/Think/Model/AdvModel.class.php
Normal file
@@ -0,0 +1,595 @@
|
||||
<?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\Model;
|
||||
use Think\Model;
|
||||
/**
|
||||
* 高级模型扩展
|
||||
*/
|
||||
class AdvModel extends Model {
|
||||
protected $optimLock = 'lock_version';
|
||||
protected $returnType = 'array';
|
||||
protected $blobFields = array();
|
||||
protected $blobValues = null;
|
||||
protected $serializeField = array();
|
||||
protected $readonlyField = array();
|
||||
protected $_filter = array();
|
||||
protected $partition = array();
|
||||
|
||||
public function __construct($name='',$tablePrefix='',$connection='') {
|
||||
if('' !== $name || is_subclass_of($this,'AdvModel') ){
|
||||
// 如果是AdvModel子类或者有传入模型名称则获取字段缓存
|
||||
}else{
|
||||
// 空的模型 关闭字段缓存
|
||||
$this->autoCheckFields = false;
|
||||
}
|
||||
parent::__construct($name,$tablePrefix,$connection);
|
||||
}
|
||||
|
||||
/**
|
||||
* 利用__call方法重载 实现一些特殊的Model方法 （魔术方法）
|
||||
* @access public
|
||||
* @param string $method 方法名称
|
||||
* @param mixed $args 调用参数
|
||||
* @return mixed
|
||||
*/
|
||||
public function __call($method,$args) {
|
||||
if(strtolower(substr($method,0,3))=='top'){
|
||||
// 获取前N条记录
|
||||
$count = substr($method,3);
|
||||
array_unshift($args,$count);
|
||||
return call_user_func_array(array(&$this, 'topN'), $args);
|
||||
}else{
|
||||
return parent::__call($method,$args);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 对保存到数据库的数据进行处理
|
||||
* @access protected
|
||||
* @param mixed $data 要操作的数据
|
||||
* @return boolean
|
||||
*/
|
||||
protected function _facade($data) {
|
||||
// 检查序列化字段
|
||||
$data = $this->serializeField($data);
|
||||
return parent::_facade($data);
|
||||
}
|
||||
|
||||
// 查询成功后的回调方法
|
||||
protected function _after_find(&$result,$options='') {
|
||||
// 检查序列化字段
|
||||
$this->checkSerializeField($result);
|
||||
// 获取文本字段
|
||||
$this->getBlobFields($result);
|
||||
// 检查字段过滤
|
||||
$result = $this->getFilterFields($result);
|
||||
// 缓存乐观锁
|
||||
$this->cacheLockVersion($result);
|
||||
}
|
||||
|
||||
// 查询数据集成功后的回调方法
|
||||
protected function _after_select(&$resultSet,$options='') {
|
||||
// 检查序列化字段
|
||||
$resultSet = $this->checkListSerializeField($resultSet);
|
||||
// 获取文本字段
|
||||
$resultSet = $this->getListBlobFields($resultSet);
|
||||
// 检查列表字段过滤
|
||||
$resultSet = $this->getFilterListFields($resultSet);
|
||||
}
|
||||
|
||||
// 写入前的回调方法
|
||||
protected function _before_insert(&$data,$options='') {
|
||||
// 记录乐观锁
|
||||
$data = $this->recordLockVersion($data);
|
||||
// 检查文本字段
|
||||
$data = $this->checkBlobFields($data);
|
||||
// 检查字段过滤
|
||||
$data = $this->setFilterFields($data);
|
||||
}
|
||||
|
||||
protected function _after_insert($data,$options) {
|
||||
// 保存文本字段
|
||||
$this->saveBlobFields($data);
|
||||
}
|
||||
|
||||
// 更新前的回调方法
|
||||
protected function _before_update(&$data,$options='') {
|
||||
// 检查乐观锁
|
||||
$pk = $this->getPK();
|
||||
if(isset($options['where'][$pk])){
|
||||
$id = $options['where'][$pk];
|
||||
if(!$this->checkLockVersion($id,$data)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
// 检查文本字段
|
||||
$data = $this->checkBlobFields($data);
|
||||
// 检查只读字段
|
||||
$data = $this->checkReadonlyField($data);
|
||||
// 检查字段过滤
|
||||
$data = $this->setFilterFields($data);
|
||||
}
|
||||
|
||||
protected function _after_update($data,$options) {
|
||||
// 保存文本字段
|
||||
$this->saveBlobFields($data);
|
||||
}
|
||||
|
||||
protected function _after_delete($data,$options) {
|
||||
// 删除Blob数据
|
||||
$this->delBlobFields($data);
|
||||
}
|
||||
|
||||
/**
|
||||
* 记录乐观锁
|
||||
* @access protected
|
||||
* @param array $data 数据对象
|
||||
* @return array
|
||||
*/
|
||||
protected function recordLockVersion($data) {
|
||||
// 记录乐观锁
|
||||
if($this->optimLock && !isset($data[$this->optimLock]) ) {
|
||||
if(in_array($this->optimLock,$this->fields,true)) {
|
||||
$data[$this->optimLock] = 0;
|
||||
}
|
||||
}
|
||||
return $data;
|
||||
}
|
||||
|
||||
/**
|
||||
* 缓存乐观锁
|
||||
* @access protected
|
||||
* @param array $data 数据对象
|
||||
* @return void
|
||||
*/
|
||||
protected function cacheLockVersion($data) {
|
||||
if($this->optimLock) {
|
||||
if(isset($data[$this->optimLock]) && isset($data[$this->getPk()])) {
|
||||
// 只有当存在乐观锁字段和主键有值的时候才记录乐观锁
|
||||
$_SESSION[$this->name.'_'.$data[$this->getPk()].'_lock_version'] = $data[$this->optimLock];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查乐观锁
|
||||
* @access protected
|
||||
* @param inteter $id 当前主键
|
||||
* @param array $data 当前数据
|
||||
* @return mixed
|
||||
*/
|
||||
protected function checkLockVersion($id,&$data) {
|
||||
// 检查乐观锁
|
||||
$identify = $this->name.'_'.$id.'_lock_version';
|
||||
if($this->optimLock && isset($_SESSION[$identify])) {
|
||||
$lock_version = $_SESSION[$identify];
|
||||
$vo = $this->field($this->optimLock)->find($id);
|
||||
$_SESSION[$identify] = $lock_version;
|
||||
$curr_version = $vo[$this->optimLock];
|
||||
if(isset($curr_version)) {
|
||||
if($curr_version>0 && $lock_version != $curr_version) {
|
||||
// 记录已经更新
|
||||
$this->error = L('_RECORD_HAS_UPDATE_');
|
||||
return false;
|
||||
}else{
|
||||
// 更新乐观锁
|
||||
$save_version = $data[$this->optimLock];
|
||||
if($save_version != $lock_version+1) {
|
||||
$data[$this->optimLock] = $lock_version+1;
|
||||
}
|
||||
$_SESSION[$identify] = $lock_version+1;
|
||||
}
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 查找前N个记录
|
||||
* @access public
|
||||
* @param integer $count 记录个数
|
||||
* @param array $options 查询表达式
|
||||
* @return array
|
||||
*/
|
||||
public function topN($count,$options=array()) {
|
||||
$options['limit'] = $count;
|
||||
return $this->select($options);
|
||||
}
|
||||
|
||||
/**
|
||||
* 查询符合条件的第N条记录
|
||||
* 0 表示第一条记录 -1 表示最后一条记录
|
||||
* @access public
|
||||
* @param integer $position 记录位置
|
||||
* @param array $options 查询表达式
|
||||
* @return mixed
|
||||
*/
|
||||
public function getN($position=0,$options=array()) {
|
||||
if($position>=0) { // 正向查找
|
||||
$options['limit'] = $position.',1';
|
||||
$list = $this->select($options);
|
||||
return $list?$list[0]:false;
|
||||
}else{ // 逆序查找
|
||||
$list = $this->select($options);
|
||||
return $list?$list[count($list)-abs($position)]:false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取满足条件的第一条记录
|
||||
* @access public
|
||||
* @param array $options 查询表达式
|
||||
* @return mixed
|
||||
*/
|
||||
public function first($options=array()) {
|
||||
return $this->getN(0,$options);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取满足条件的最后一条记录
|
||||
* @access public
|
||||
* @param array $options 查询表达式
|
||||
* @return mixed
|
||||
*/
|
||||
public function last($options=array()) {
|
||||
return $this->getN(-1,$options);
|
||||
}
|
||||
|
||||
/**
|
||||
* 返回数据
|
||||
* @access public
|
||||
* @param array $data 数据
|
||||
* @param string $type 返回类型 默认为数组
|
||||
* @return mixed
|
||||
*/
|
||||
public function returnResult($data,$type='') {
|
||||
if('' === $type)
|
||||
$type = $this->returnType;
|
||||
switch($type) {
|
||||
case 'array' : return $data;
|
||||
case 'object': return (object)$data;
|
||||
default:// 允许用户自定义返回类型
|
||||
if(class_exists($type))
|
||||
return new $type($data);
|
||||
else
|
||||
E(L('_CLASS_NOT_EXIST_').':'.$type);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取数据的时候过滤数据字段
|
||||
* @access protected
|
||||
* @param mixed $result 查询的数据
|
||||
* @return array
|
||||
*/
|
||||
protected function getFilterFields(&$result) {
|
||||
if(!empty($this->_filter)) {
|
||||
foreach ($this->_filter as $field=>$filter){
|
||||
if(isset($result[$field])) {
|
||||
$fun = $filter[1];
|
||||
if(!empty($fun)) {
|
||||
if(isset($filter[2]) && $filter[2]){
|
||||
// 传递整个数据对象作为参数
|
||||
$result[$field] = call_user_func($fun,$result);
|
||||
}else{
|
||||
// 传递字段的值作为参数
|
||||
$result[$field] = call_user_func($fun,$result[$field]);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
protected function getFilterListFields(&$resultSet) {
|
||||
if(!empty($this->_filter)) {
|
||||
foreach ($resultSet as $key=>$result)
|
||||
$resultSet[$key] = $this->getFilterFields($result);
|
||||
}
|
||||
return $resultSet;
|
||||
}
|
||||
|
||||
/**
|
||||
* 写入数据的时候过滤数据字段
|
||||
* @access protected
|
||||
* @param mixed $result 查询的数据
|
||||
* @return array
|
||||
*/
|
||||
protected function setFilterFields($data) {
|
||||
if(!empty($this->_filter)) {
|
||||
foreach ($this->_filter as $field=>$filter){
|
||||
if(isset($data[$field])) {
|
||||
$fun = $filter[0];
|
||||
if(!empty($fun)) {
|
||||
if(isset($filter[2]) && $filter[2]) {
|
||||
// 传递整个数据对象作为参数
|
||||
$data[$field] = call_user_func($fun,$data);
|
||||
}else{
|
||||
// 传递字段的值作为参数
|
||||
$data[$field] = call_user_func($fun,$data[$field]);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return $data;
|
||||
}
|
||||
|
||||
/**
|
||||
* 返回数据列表
|
||||
* @access protected
|
||||
* @param array $resultSet 数据
|
||||
* @param string $type 返回类型 默认为数组
|
||||
* @return void
|
||||
*/
|
||||
protected function returnResultSet(&$resultSet,$type='') {
|
||||
foreach ($resultSet as $key=>$data)
|
||||
$resultSet[$key] = $this->returnResult($data,$type);
|
||||
return $resultSet;
|
||||
}
|
||||
|
||||
protected function checkBlobFields(&$data) {
|
||||
// 检查Blob文件保存字段
|
||||
if(!empty($this->blobFields)) {
|
||||
foreach ($this->blobFields as $field){
|
||||
if(isset($data[$field])) {
|
||||
if(isset($data[$this->getPk()]))
|
||||
$this->blobValues[$this->name.'/'.$data[$this->getPk()].'_'.$field] = $data[$field];
|
||||
else
|
||||
$this->blobValues[$this->name.'/@?id@_'.$field] = $data[$field];
|
||||
unset($data[$field]);
|
||||
}
|
||||
}
|
||||
}
|
||||
return $data;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取数据集的文本字段
|
||||
* @access protected
|
||||
* @param mixed $resultSet 查询的数据
|
||||
* @param string $field 查询的字段
|
||||
* @return void
|
||||
*/
|
||||
protected function getListBlobFields(&$resultSet,$field='') {
|
||||
if(!empty($this->blobFields)) {
|
||||
foreach ($resultSet as $key=>$result){
|
||||
$result = $this->getBlobFields($result,$field);
|
||||
$resultSet[$key] = $result;
|
||||
}
|
||||
}
|
||||
return $resultSet;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取数据的文本字段
|
||||
* @access protected
|
||||
* @param mixed $data 查询的数据
|
||||
* @param string $field 查询的字段
|
||||
* @return void
|
||||
*/
|
||||
protected function getBlobFields(&$data,$field='') {
|
||||
if(!empty($this->blobFields)) {
|
||||
$pk = $this->getPk();
|
||||
$id = $data[$pk];
|
||||
if(empty($field)) {
|
||||
foreach ($this->blobFields as $field){
|
||||
$identify = $this->name.'/'.$id.'_'.$field;
|
||||
$data[$field] = F($identify);
|
||||
}
|
||||
return $data;
|
||||
}else{
|
||||
$identify = $this->name.'/'.$id.'_'.$field;
|
||||
return F($identify);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 保存File方式的字段
|
||||
* @access protected
|
||||
* @param mixed $data 保存的数据
|
||||
* @return void
|
||||
*/
|
||||
protected function saveBlobFields(&$data) {
|
||||
if(!empty($this->blobFields)) {
|
||||
foreach ($this->blobValues as $key=>$val){
|
||||
if(strpos($key,'@?id@'))
|
||||
$key = str_replace('@?id@',$data[$this->getPk()],$key);
|
||||
F($key,$val);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除File方式的字段
|
||||
* @access protected
|
||||
* @param mixed $data 保存的数据
|
||||
* @param string $field 查询的字段
|
||||
* @return void
|
||||
*/
|
||||
protected function delBlobFields(&$data,$field='') {
|
||||
if(!empty($this->blobFields)) {
|
||||
$pk = $this->getPk();
|
||||
$id = $data[$pk];
|
||||
if(empty($field)) {
|
||||
foreach ($this->blobFields as $field){
|
||||
$identify = $this->name.'/'.$id.'_'.$field;
|
||||
F($identify,null);
|
||||
}
|
||||
}else{
|
||||
$identify = $this->name.'/'.$id.'_'.$field;
|
||||
F($identify,null);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查序列化数据字段
|
||||
* @access protected
|
||||
* @param array $data 数据
|
||||
* @return array
|
||||
*/
|
||||
protected function serializeField(&$data) {
|
||||
// 检查序列化字段
|
||||
if(!empty($this->serializeField)) {
|
||||
// 定义方式 $this->serializeField = array('ser'=>array('name','email'));
|
||||
foreach ($this->serializeField as $key=>$val){
|
||||
if(empty($data[$key])) {
|
||||
$serialize = array();
|
||||
foreach ($val as $name){
|
||||
if(isset($data[$name])) {
|
||||
$serialize[$name] = $data[$name];
|
||||
unset($data[$name]);
|
||||
}
|
||||
}
|
||||
if(!empty($serialize)) {
|
||||
$data[$key] = serialize($serialize);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return $data;
|
||||
}
|
||||
|
||||
// 检查返回数据的序列化字段
|
||||
protected function checkSerializeField(&$result) {
|
||||
// 检查序列化字段
|
||||
if(!empty($this->serializeField)) {
|
||||
foreach ($this->serializeField as $key=>$val){
|
||||
if(isset($result[$key])) {
|
||||
$serialize = unserialize($result[$key]);
|
||||
foreach ($serialize as $name=>$value)
|
||||
$result[$name] = $value;
|
||||
unset($serialize,$result[$key]);
|
||||
}
|
||||
}
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
// 检查数据集的序列化字段
|
||||
protected function checkListSerializeField(&$resultSet) {
|
||||
// 检查序列化字段
|
||||
if(!empty($this->serializeField)) {
|
||||
foreach ($this->serializeField as $key=>$val){
|
||||
foreach ($resultSet as $k=>$result){
|
||||
if(isset($result[$key])) {
|
||||
$serialize = unserialize($result[$key]);
|
||||
foreach ($serialize as $name=>$value)
|
||||
$result[$name] = $value;
|
||||
unset($serialize,$result[$key]);
|
||||
$resultSet[$k] = $result;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return $resultSet;
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查只读字段
|
||||
* @access protected
|
||||
* @param array $data 数据
|
||||
* @return array
|
||||
*/
|
||||
protected function checkReadonlyField(&$data) {
|
||||
if(!empty($this->readonlyField)) {
|
||||
foreach ($this->readonlyField as $key=>$field){
|
||||
if(isset($data[$field]))
|
||||
unset($data[$field]);
|
||||
}
|
||||
}
|
||||
return $data;
|
||||
}
|
||||
|
||||
/**
|
||||
* 批处理执行SQL语句
|
||||
* 批处理的指令都认为是execute操作
|
||||
* @access public
|
||||
* @param array $sql SQL批处理指令
|
||||
* @return boolean
|
||||
*/
|
||||
public function patchQuery($sql=array()) {
|
||||
if(!is_array($sql)) return false;
|
||||
// 自动启动事务支持
|
||||
$this->startTrans();
|
||||
try{
|
||||
foreach ($sql as $_sql){
|
||||
$result = $this->execute($_sql);
|
||||
if(false === $result) {
|
||||
// 发生错误自动回滚事务
|
||||
$this->rollback();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
// 提交事务
|
||||
$this->commit();
|
||||
} catch (ThinkException $e) {
|
||||
$this->rollback();
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 得到分表的的数据表名
|
||||
* @access public
|
||||
* @param array $data 操作的数据
|
||||
* @return string
|
||||
*/
|
||||
public function getPartitionTableName($data=array()) {
|
||||
// 对数据表进行分区
|
||||
if(isset($data[$this->partition['field']])) {
|
||||
$field = $data[$this->partition['field']];
|
||||
switch($this->partition['type']) {
|
||||
case 'id':
|
||||
// 按照id范围分表
|
||||
$step = $this->partition['expr'];
|
||||
$seq = floor($field / $step)+1;
|
||||
break;
|
||||
case 'year':
|
||||
// 按照年份分表
|
||||
if(!is_numeric($field)) {
|
||||
$field = strtotime($field);
|
||||
}
|
||||
$seq = date('Y',$field)-$this->partition['expr']+1;
|
||||
break;
|
||||
case 'mod':
|
||||
// 按照id的模数分表
|
||||
$seq = ($field % $this->partition['num'])+1;
|
||||
break;
|
||||
case 'md5':
|
||||
// 按照md5的序列分表
|
||||
$seq = (ord(substr(md5($field),0,1)) % $this->partition['num'])+1;
|
||||
break;
|
||||
default :
|
||||
if(function_exists($this->partition['type'])) {
|
||||
// 支持指定函数哈希
|
||||
$fun = $this->partition['type'];
|
||||
$seq = (ord(substr($fun($field),0,1)) % $this->partition['num'])+1;
|
||||
}else{
|
||||
// 按照字段的首字母的值分表
|
||||
$seq = (ord($field{0}) % $this->partition['num'])+1;
|
||||
}
|
||||
}
|
||||
return $this->getTableName().'_'.$seq;
|
||||
}else{
|
||||
// 当设置的分表字段不在查询条件或者数据中
|
||||
// 进行联合查询，必须设定 partition['num']
|
||||
$tableName = array();
|
||||
for($i=0;$i<$this->partition['num'];$i++)
|
||||
$tableName[] = 'SELECT * FROM '.$this->getTableName().'_'.($i+1);
|
||||
$tableName = '( '.implode(" UNION ",$tableName).') AS '.$this->name;
|
||||
return $tableName;
|
||||
}
|
||||
}
|
||||
}
|
||||
403
ThinkPHP/Library/Think/Model/MergeModel.class.php
Normal file
403
ThinkPHP/Library/Think/Model/MergeModel.class.php
Normal file
@@ -0,0 +1,403 @@
|
||||
<?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\Model;
|
||||
use Think\Model;
|
||||
/**
|
||||
* ThinkPHP 聚合模型扩展
|
||||
*/
|
||||
class MergeModel extends Model {
|
||||
|
||||
protected $modelList = array(); // 包含的模型列表 第一个必须是主表模型
|
||||
protected $masterModel = ''; // 主模型
|
||||
protected $joinType = 'INNER'; // 聚合模型的查询JOIN类型
|
||||
protected $fk = ''; // 外键名 默认为主表名_id
|
||||
protected $mapFields = array(); // 需要处理的模型映射字段，避免混淆 array( id => 'user.id' )
|
||||
|
||||
/**
|
||||
* 架构函数
|
||||
* 取得DB类的实例对象 字段检查
|
||||
* @access public
|
||||
* @param string $name 模型名称
|
||||
* @param string $tablePrefix 表前缀
|
||||
* @param mixed $connection 数据库连接信息
|
||||
*/
|
||||
public function __construct($name='',$tablePrefix='',$connection=''){
|
||||
parent::__construct($name,$tablePrefix,$connection);
|
||||
// 聚合模型的字段信息
|
||||
if(empty($this->fields) && !empty($this->modelList)){
|
||||
$fields = array();
|
||||
foreach($this->modelList as $model){
|
||||
// 获取模型的字段信息
|
||||
$result = $this->db->getFields(M($model)->getTableName());
|
||||
$_fields = array_keys($result);
|
||||
// $this->mapFields = array_intersect($fields,$_fields);
|
||||
$fields = array_merge($fields,$_fields);
|
||||
}
|
||||
$this->fields = $fields;
|
||||
}
|
||||
|
||||
// 设置第一个模型为主表模型
|
||||
if(empty($this->masterModel) && !empty($this->modelList)){
|
||||
$this->masterModel = $this->modelList[0];
|
||||
}
|
||||
// 主表的主键名
|
||||
$this->pk = M($this->masterModel)->getPk();
|
||||
|
||||
// 设置默认外键名 仅支持单一外键
|
||||
if(empty($this->fk)){
|
||||
$this->fk = strtolower($this->masterModel).'_id';
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 得到完整的数据表名
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function getTableName() {
|
||||
if(empty($this->trueTableName)) {
|
||||
$tableName = array();
|
||||
$models = $this->modelList;
|
||||
foreach($models as $model){
|
||||
$tableName[] = M($model)->getTableName().' '.$model;
|
||||
}
|
||||
$this->trueTableName = implode(',',$tableName);
|
||||
}
|
||||
return $this->trueTableName;
|
||||
}
|
||||
|
||||
/**
|
||||
* 自动检测数据表信息
|
||||
* @access protected
|
||||
* @return void
|
||||
*/
|
||||
protected function _checkTableInfo() {}
|
||||
|
||||
/**
|
||||
* 新增聚合数据
|
||||
* @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;
|
||||
}
|
||||
}
|
||||
// 启动事务
|
||||
$this->startTrans();
|
||||
// 写入主表数据
|
||||
$result = M($this->masterModel)->strict(false)->add($data);
|
||||
if($result){
|
||||
// 写入外键数据
|
||||
$data[$this->fk] = $result;
|
||||
$models = $this->modelList;
|
||||
array_shift($models);
|
||||
// 写入附表数据
|
||||
foreach($models as $model){
|
||||
$res = M($model)->strict(false)->add($data);
|
||||
if(!$res){
|
||||
$this->rollback();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
// 提交事务
|
||||
$this->commit();
|
||||
}else{
|
||||
$this->rollback();
|
||||
return false;
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 对保存到数据库的数据进行处理
|
||||
* @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)){
|
||||
unset($data[$key]);
|
||||
}elseif(array_key_exists($key,$this->mapFields)){
|
||||
// 需要处理映射字段
|
||||
$data[$this->mapFields[$key]] = $val;
|
||||
unset($data[$key]);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 安全过滤
|
||||
if(!empty($this->options['filter'])) {
|
||||
$data = array_map($this->options['filter'],$data);
|
||||
unset($this->options['filter']);
|
||||
}
|
||||
$this->_before_write($data);
|
||||
return $data;
|
||||
}
|
||||
|
||||
/**
|
||||
* 保存聚合模型数据
|
||||
* @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;
|
||||
}
|
||||
}
|
||||
if(empty($data)){
|
||||
// 没有数据则不执行
|
||||
$this->error = L('_DATA_TYPE_INVALID_');
|
||||
return false;
|
||||
}
|
||||
// 如果存在主键数据 则自动作为更新条件
|
||||
$pk = $this->pk;
|
||||
if(isset($data[$pk])) {
|
||||
$where[$pk] = $data[$pk];
|
||||
$options['where'] = $where;
|
||||
unset($data[$pk]);
|
||||
}
|
||||
$options['join'] = '';
|
||||
$options = $this->_parseOptions($options);
|
||||
// 更新操作不使用JOIN
|
||||
$options['table'] = $this->getTableName();
|
||||
|
||||
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) {
|
||||
if(isset($pkValue)) $data[$pk] = $pkValue;
|
||||
$this->_after_update($data,$options);
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除聚合模型数据
|
||||
* @access public
|
||||
* @param mixed $options 表达式
|
||||
* @return mixed
|
||||
*/
|
||||
public function delete($options=array()){
|
||||
$pk = $this->pk;
|
||||
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;
|
||||
}
|
||||
// 分析表达式
|
||||
$options['join'] = '';
|
||||
$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];
|
||||
}
|
||||
|
||||
$options['table'] = implode(',',$this->modelList);
|
||||
$options['using'] = $this->getTableName();
|
||||
if(false === $this->_before_delete($options)) {
|
||||
return false;
|
||||
}
|
||||
$result = $this->db->delete($options);
|
||||
if(false !== $result) {
|
||||
$data = array();
|
||||
if(isset($pkValue)) $data[$pk] = $pkValue;
|
||||
$this->_after_delete($data,$options);
|
||||
}
|
||||
// 返回删除记录个数
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 表达式过滤方法
|
||||
* @access protected
|
||||
* @param string $options 表达式
|
||||
* @return void
|
||||
*/
|
||||
protected function _options_filter(&$options) {
|
||||
if(!isset($options['join'])){
|
||||
$models = $this->modelList;
|
||||
array_shift($models);
|
||||
foreach($models as $model){
|
||||
$options['join'][] = $this->joinType.' JOIN '.M($model)->getTableName().' '.$model.' ON '.$this->masterModel.'.'.$this->pk.' = '.$model.'.'.$this->fk;
|
||||
}
|
||||
}
|
||||
$options['table'] = M($this->masterModel)->getTableName().' '.$this->masterModel;
|
||||
$options['field'] = $this->checkFields(isset($options['field'])?$options['field']:'');
|
||||
if(isset($options['group']))
|
||||
$options['group'] = $this->checkGroup($options['group']);
|
||||
if(isset($options['where']))
|
||||
$options['where'] = $this->checkCondition($options['where']);
|
||||
if(isset($options['order']))
|
||||
$options['order'] = $this->checkOrder($options['order']);
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查条件中的聚合字段
|
||||
* @access protected
|
||||
* @param mixed $data 条件表达式
|
||||
* @return array
|
||||
*/
|
||||
protected function checkCondition($where) {
|
||||
if(is_array($where)) {
|
||||
$view = array();
|
||||
foreach($where as $name=>$value){
|
||||
if(array_key_exists($name,$this->mapFields)){
|
||||
// 需要处理映射字段
|
||||
$view[$this->mapFields[$name]] = $value;
|
||||
unset($where[$name]);
|
||||
}
|
||||
}
|
||||
$where = array_merge($where,$view);
|
||||
}
|
||||
return $where;
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查Order表达式中的聚合字段
|
||||
* @access protected
|
||||
* @param string $order 字段
|
||||
* @return string
|
||||
*/
|
||||
protected function checkOrder($order='') {
|
||||
if(is_string($order) && !empty($order)) {
|
||||
$orders = explode(',',$order);
|
||||
$_order = array();
|
||||
foreach ($orders as $order){
|
||||
$array = explode(' ',trim($order));
|
||||
$field = $array[0];
|
||||
$sort = isset($array[1])?$array[1]:'ASC';
|
||||
if(array_key_exists($field,$this->mapFields)){
|
||||
// 需要处理映射字段
|
||||
$field = $this->mapFields[$field];
|
||||
}
|
||||
$_order[] = $field.' '.$sort;
|
||||
}
|
||||
$order = implode(',',$_order);
|
||||
}
|
||||
return $order;
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查Group表达式中的聚合字段
|
||||
* @access protected
|
||||
* @param string $group 字段
|
||||
* @return string
|
||||
*/
|
||||
protected function checkGroup($group='') {
|
||||
if(!empty($group)) {
|
||||
$groups = explode(',',$group);
|
||||
$_group = array();
|
||||
foreach ($groups as $field){
|
||||
// 解析成聚合字段
|
||||
if(array_key_exists($field,$this->mapFields)){
|
||||
// 需要处理映射字段
|
||||
$field = $this->mapFields[$field];
|
||||
}
|
||||
$_group[] = $field;
|
||||
}
|
||||
$group = implode(',',$_group);
|
||||
}
|
||||
return $group;
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查fields表达式中的聚合字段
|
||||
* @access protected
|
||||
* @param string $fields 字段
|
||||
* @return string
|
||||
*/
|
||||
protected function checkFields($fields='') {
|
||||
if(empty($fields) || '*'==$fields ) {
|
||||
// 获取全部聚合字段
|
||||
$fields = $this->fields;
|
||||
}
|
||||
if(!is_array($fields))
|
||||
$fields = explode(',',$fields);
|
||||
|
||||
// 解析成聚合字段
|
||||
$array = array();
|
||||
foreach ($fields as $field){
|
||||
if(array_key_exists($field,$this->mapFields)){
|
||||
// 需要处理映射字段
|
||||
$array[] = $this->mapFields[$field].' AS '.$field;
|
||||
}else{
|
||||
$array[] = $field;
|
||||
}
|
||||
}
|
||||
$fields = implode(',',$array);
|
||||
return $fields;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取数据表字段信息
|
||||
* @access public
|
||||
* @return array
|
||||
*/
|
||||
public function getDbFields(){
|
||||
return $this->fields;
|
||||
}
|
||||
|
||||
}
|
||||
422
ThinkPHP/Library/Think/Model/MongoModel.class.php
Normal file
422
ThinkPHP/Library/Think/Model/MongoModel.class.php
Normal file
@@ -0,0 +1,422 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | ThinkPHP [ WE CAN DO IT JUST THINK ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2010 http://topthink.com All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
namespace Think\Model;
|
||||
use Think\Model;
|
||||
/**
|
||||
* MongoModel模型类
|
||||
* 实现了ODM和ActiveRecords模式
|
||||
*/
|
||||
class MongoModel extends Model{
|
||||
// 主键类型
|
||||
const TYPE_OBJECT = 1;
|
||||
const TYPE_INT = 2;
|
||||
const TYPE_STRING = 3;
|
||||
|
||||
// 主键名称
|
||||
protected $pk = '_id';
|
||||
// _id 类型 1 Object 采用MongoId对象 2 Int 整形 支持自动增长 3 String 字符串Hash
|
||||
protected $_idType = self::TYPE_OBJECT;
|
||||
// 主键是否自增
|
||||
protected $_autoinc = true;
|
||||
// Mongo默认关闭字段检测 可以动态追加字段
|
||||
protected $autoCheckFields = false;
|
||||
// 链操作方法列表
|
||||
protected $methods = array('table','order','auto','filter','validate');
|
||||
|
||||
/**
|
||||
* 利用__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(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]);
|
||||
}else{
|
||||
E(__CLASS__.':'.$method.L('_METHOD_NOT_EXIST_'));
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取字段信息并缓存 主键和自增信息直接配置
|
||||
* @access public
|
||||
* @return void
|
||||
*/
|
||||
public function flush() {
|
||||
// 缓存不存在则查询数据表信息
|
||||
$fields = $this->db->getFields();
|
||||
if(!$fields) { // 暂时没有数据无法获取字段信息 下次查询
|
||||
return false;
|
||||
}
|
||||
$this->fields = array_keys($fields);
|
||||
foreach ($fields as $key=>$val){
|
||||
// 记录字段类型
|
||||
$type[$key] = $val['type'];
|
||||
}
|
||||
// 记录字段类型信息
|
||||
if(C('DB_FIELDTYPE_CHECK')) $this->fields['_type'] = $type;
|
||||
|
||||
// 2008-3-7 增加缓存开关控制
|
||||
if(C('DB_FIELDS_CACHE')){
|
||||
// 永久缓存数据表信息
|
||||
$db = $this->dbName?$this->dbName:C('DB_NAME');
|
||||
F('_fields/'.$db.'.'.$this->name,$this->fields);
|
||||
}
|
||||
}
|
||||
|
||||
// 写入数据前的回调方法 包括新增和更新
|
||||
protected function _before_write(&$data) {
|
||||
$pk = $this->getPk();
|
||||
// 根据主键类型处理主键数据
|
||||
if(isset($data[$pk]) && $this->_idType == self::TYPE_OBJECT) {
|
||||
$data[$pk] = new \MongoId($data[$pk]);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* count统计 配合where连贯操作
|
||||
* @access public
|
||||
* @return integer
|
||||
*/
|
||||
public function count(){
|
||||
// 分析表达式
|
||||
$options = $this->_parseOptions();
|
||||
return $this->db->count($options);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取唯一值
|
||||
* @access public
|
||||
* @return array | false
|
||||
*/
|
||||
public function distinct($field, $where=array() ){
|
||||
// 分析表达式
|
||||
$this->options = $this->_parseOptions();
|
||||
$this->options['where'] = array_merge((array)$this->options['where'], $where);
|
||||
|
||||
$command = array(
|
||||
"distinct" => $this->options['table'],
|
||||
"key" => $field,
|
||||
"query" => $this->options['where']
|
||||
);
|
||||
|
||||
$result = $this->command($command);
|
||||
return isset($result['values']) ? $result['values'] : false;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取下一ID 用于自动增长型
|
||||
* @access public
|
||||
* @param string $pk 字段名 默认为主键
|
||||
* @return mixed
|
||||
*/
|
||||
public function getMongoNextId($pk=''){
|
||||
if(empty($pk)) {
|
||||
$pk = $this->getPk();
|
||||
}
|
||||
return $this->db->getMongoNextId($pk);
|
||||
}
|
||||
|
||||
/**
|
||||
* 新增数据
|
||||
* @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;
|
||||
}
|
||||
}
|
||||
// 分析表达式
|
||||
$options = $this->_parseOptions($options);
|
||||
// 数据处理
|
||||
$data = $this->_facade($data);
|
||||
if(false === $this->_before_insert($data,$options)) {
|
||||
return false;
|
||||
}
|
||||
// 写入数据到数据库
|
||||
$result = $this->db->insert($data,$options,$replace);
|
||||
if(false !== $result ) {
|
||||
$this->_after_insert($data,$options);
|
||||
if(isset($data[$this->getPk()])){
|
||||
return $data[$this->getPk()];
|
||||
}
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
// 插入数据前的回调方法
|
||||
protected function _before_insert(&$data,$options) {
|
||||
// 写入数据到数据库
|
||||
if($this->_autoinc && $this->_idType== self::TYPE_INT) { // 主键自动增长
|
||||
$pk = $this->getPk();
|
||||
if(!isset($data[$pk])) {
|
||||
$data[$pk] = $this->db->getMongoNextId($pk);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public function clear(){
|
||||
return $this->db->clear();
|
||||
}
|
||||
|
||||
// 查询成功后的回调方法
|
||||
protected function _after_select(&$resultSet,$options) {
|
||||
array_walk($resultSet,array($this,'checkMongoId'));
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取MongoId
|
||||
* @access protected
|
||||
* @param array $result 返回数据
|
||||
* @return array
|
||||
*/
|
||||
protected function checkMongoId(&$result){
|
||||
if(is_object($result['_id'])) {
|
||||
$result['_id'] = $result['_id']->__toString();
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
// 表达式过滤回调方法
|
||||
protected function _options_filter(&$options) {
|
||||
$id = $this->getPk();
|
||||
if(isset($options['where'][$id]) && is_scalar($options['where'][$id]) && $this->_idType== self::TYPE_OBJECT) {
|
||||
$options['where'][$id] = new \MongoId($options['where'][$id]);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 查询数据
|
||||
* @access public
|
||||
* @param mixed $options 表达式参数
|
||||
* @return mixed
|
||||
*/
|
||||
public function find($options=array()) {
|
||||
if( is_numeric($options) || is_string($options)) {
|
||||
$id = $this->getPk();
|
||||
$where[$id] = $options;
|
||||
$options = array();
|
||||
$options['where'] = $where;
|
||||
}
|
||||
// 分析表达式
|
||||
$options = $this->_parseOptions($options);
|
||||
$result = $this->db->find($options);
|
||||
if(false === $result) {
|
||||
return false;
|
||||
}
|
||||
if(empty($result)) {// 查询结果为空
|
||||
return null;
|
||||
}else{
|
||||
$this->checkMongoId($result);
|
||||
}
|
||||
$this->data = $result;
|
||||
$this->_after_find($this->data,$options);
|
||||
return $this->data;
|
||||
}
|
||||
|
||||
/**
|
||||
* 字段值增长
|
||||
* @access public
|
||||
* @param string $field 字段名
|
||||
* @param integer $step 增长值
|
||||
* @return boolean
|
||||
*/
|
||||
public function setInc($field,$step=1) {
|
||||
return $this->setField($field,array('inc',$step));
|
||||
}
|
||||
|
||||
/**
|
||||
* 字段值减少
|
||||
* @access public
|
||||
* @param string $field 字段名
|
||||
* @param integer $step 减少值
|
||||
* @return boolean
|
||||
*/
|
||||
public function setDec($field,$step=1) {
|
||||
return $this->setField($field,array('inc','-'.$step));
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取一条记录的某个字段值
|
||||
* @access public
|
||||
* @param string $field 字段名
|
||||
* @param string $spea 字段数据间隔符号
|
||||
* @return mixed
|
||||
*/
|
||||
public function getField($field,$sepa=null) {
|
||||
$options['field'] = $field;
|
||||
$options = $this->_parseOptions($options);
|
||||
if(strpos($field,',')) { // 多字段
|
||||
if(is_numeric($sepa)) {// 限定数量
|
||||
$options['limit'] = $sepa;
|
||||
$sepa = null;// 重置为null 返回数组
|
||||
}
|
||||
$resultSet = $this->db->select($options);
|
||||
if(!empty($resultSet)) {
|
||||
$_field = explode(',', $field);
|
||||
$field = array_keys($resultSet[0]);
|
||||
$key = array_shift($field);
|
||||
$key2 = array_shift($field);
|
||||
$cols = array();
|
||||
$count = count($_field);
|
||||
foreach ($resultSet as $result){
|
||||
$name = $result[$key];
|
||||
if(2==$count) {
|
||||
$cols[$name] = $result[$key2];
|
||||
}else{
|
||||
$cols[$name] = is_null($sepa)?$result:implode($sepa,$result);
|
||||
}
|
||||
}
|
||||
return $cols;
|
||||
}
|
||||
}else{
|
||||
// 返回数据个数
|
||||
if(true !== $sepa) {// 当sepa指定为true的时候 返回所有数据
|
||||
$options['limit'] = is_numeric($sepa)?$sepa:1;
|
||||
} // 查找符合的记录
|
||||
$result = $this->db->select($options);
|
||||
if(!empty($result)) {
|
||||
if(1==$options['limit']) {
|
||||
$result = reset($result);
|
||||
return $result[$field];
|
||||
}
|
||||
foreach ($result as $val){
|
||||
$array[] = $val[$field];
|
||||
}
|
||||
return $array;
|
||||
}
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行Mongo指令
|
||||
* @access public
|
||||
* @param array $command 指令
|
||||
* @return mixed
|
||||
*/
|
||||
public function command($command, $options=array()) {
|
||||
$options = $this->_parseOptions($options);
|
||||
return $this->db->command($command, $options);
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行MongoCode
|
||||
* @access public
|
||||
* @param string $code MongoCode
|
||||
* @param array $args 参数
|
||||
* @return mixed
|
||||
*/
|
||||
public function mongoCode($code,$args=array()) {
|
||||
return $this->db->execute($code,$args);
|
||||
}
|
||||
|
||||
// 数据库切换后回调方法
|
||||
protected function _after_db() {
|
||||
// 切换Collection
|
||||
$this->db->switchCollection($this->getTableName(),$this->dbName?$this->dbName:C('db_name'));
|
||||
}
|
||||
|
||||
/**
|
||||
* 得到完整的数据表名 Mongo表名不带dbName
|
||||
* @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 $this->trueTableName;
|
||||
}
|
||||
|
||||
/**
|
||||
* 分组查询
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function group($key, $init, $reduce, $option=array()) {
|
||||
$option = $this->_parseOptions($option);
|
||||
|
||||
//合并查询条件
|
||||
if(isset($option['where']))
|
||||
$option['condition'] = array_merge((array)$option['condition'], $option['where']);
|
||||
|
||||
return $this->db->group($key, $init, $reduce, $option);
|
||||
}
|
||||
|
||||
/**
|
||||
* 返回Mongo运行错误信息
|
||||
* @access public
|
||||
* @return json
|
||||
*/
|
||||
public function getLastError(){
|
||||
return $this->db->command(array('getLastError'=>1));
|
||||
}
|
||||
|
||||
/**
|
||||
* 返回指定集合的统计信息，包括数据大小、已分配的存储空间和索引的大小
|
||||
* @access public
|
||||
* @return json
|
||||
*/
|
||||
public function status(){
|
||||
$option = $this->_parseOptions();
|
||||
return $this->db->command(array('collStats'=>$option['table']));
|
||||
}
|
||||
|
||||
/**
|
||||
* 取得当前数据库的对象
|
||||
* @access public
|
||||
* @return object
|
||||
*/
|
||||
public function getDB(){
|
||||
return $this->db->getDB();
|
||||
}
|
||||
|
||||
/**
|
||||
* 取得集合对象，可以进行创建索引等查询
|
||||
* @access public
|
||||
* @return object
|
||||
*/
|
||||
public function getCollection(){
|
||||
return $this->db->getCollection();
|
||||
}
|
||||
}
|
||||
412
ThinkPHP/Library/Think/Model/RelationModel.class.php
Normal file
412
ThinkPHP/Library/Think/Model/RelationModel.class.php
Normal file
@@ -0,0 +1,412 @@
|
||||
<?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\Model;
|
||||
use Think\Model;
|
||||
/**
|
||||
* ThinkPHP关联模型扩展
|
||||
*/
|
||||
class RelationModel extends Model {
|
||||
|
||||
const HAS_ONE = 1;
|
||||
const BELONGS_TO = 2;
|
||||
const HAS_MANY = 3;
|
||||
const MANY_TO_MANY= 4;
|
||||
|
||||
// 关联定义
|
||||
protected $_link = array();
|
||||
|
||||
/**
|
||||
* 动态方法实现
|
||||
* @access public
|
||||
* @param string $method 方法名称
|
||||
* @param array $args 调用参数
|
||||
* @return mixed
|
||||
*/
|
||||
public function __call($method,$args) {
|
||||
if(strtolower(substr($method,0,8))=='relation'){
|
||||
$type = strtoupper(substr($method,8));
|
||||
if(in_array($type,array('ADD','SAVE','DEL'),true)) {
|
||||
array_unshift($args,$type);
|
||||
return call_user_func_array(array(&$this, 'opRelation'), $args);
|
||||
}
|
||||
}else{
|
||||
return parent::__call($method,$args);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 得到关联的数据表名
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function getRelationTableName($relation) {
|
||||
$relationTable = !empty($this->tablePrefix) ? $this->tablePrefix : '';
|
||||
$relationTable .= $this->tableName?$this->tableName:$this->name;
|
||||
$relationTable .= '_'.$relation->getModelName();
|
||||
return strtolower($relationTable);
|
||||
}
|
||||
|
||||
// 查询成功后的回调方法
|
||||
protected function _after_find(&$result,$options) {
|
||||
// 获取关联数据 并附加到结果中
|
||||
if(!empty($options['link']))
|
||||
$this->getRelation($result,$options['link']);
|
||||
}
|
||||
|
||||
// 查询数据集成功后的回调方法
|
||||
protected function _after_select(&$result,$options) {
|
||||
// 获取关联数据 并附加到结果中
|
||||
if(!empty($options['link']))
|
||||
$this->getRelations($result,$options['link']);
|
||||
}
|
||||
|
||||
// 写入成功后的回调方法
|
||||
protected function _after_insert($data,$options) {
|
||||
// 关联写入
|
||||
if(!empty($options['link']))
|
||||
$this->opRelation('ADD',$data,$options['link']);
|
||||
}
|
||||
|
||||
// 更新成功后的回调方法
|
||||
protected function _after_update($data,$options) {
|
||||
// 关联更新
|
||||
if(!empty($options['link']))
|
||||
$this->opRelation('SAVE',$data,$options['link']);
|
||||
}
|
||||
|
||||
// 删除成功后的回调方法
|
||||
protected function _after_delete($data,$options) {
|
||||
// 关联删除
|
||||
if(!empty($options['link']))
|
||||
$this->opRelation('DEL',$data,$options['link']);
|
||||
}
|
||||
|
||||
/**
|
||||
* 对保存到数据库的数据进行处理
|
||||
* @access protected
|
||||
* @param mixed $data 要操作的数据
|
||||
* @return boolean
|
||||
*/
|
||||
protected function _facade($data) {
|
||||
$this->_before_write($data);
|
||||
return $data;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取返回数据集的关联记录
|
||||
* @access protected
|
||||
* @param array $resultSet 返回数据
|
||||
* @param string|array $name 关联名称
|
||||
* @return array
|
||||
*/
|
||||
protected function getRelations(&$resultSet,$name='') {
|
||||
// 获取记录集的主键列表
|
||||
foreach($resultSet as $key=>$val) {
|
||||
$val = $this->getRelation($val,$name);
|
||||
$resultSet[$key] = $val;
|
||||
}
|
||||
return $resultSet;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取返回数据的关联记录
|
||||
* @access protected
|
||||
* @param mixed $result 返回数据
|
||||
* @param string|array $name 关联名称
|
||||
* @param boolean $return 是否返回关联数据本身
|
||||
* @return array
|
||||
*/
|
||||
protected function getRelation(&$result,$name='',$return=false) {
|
||||
if(!empty($this->_link)) {
|
||||
foreach($this->_link as $key=>$val) {
|
||||
$mappingName = !empty($val['mapping_name'])?$val['mapping_name']:$key; // 映射名称
|
||||
if(empty($name) || true === $name || $mappingName == $name || (is_array($name) && in_array($mappingName,$name))) {
|
||||
$mappingType = !empty($val['mapping_type'])?$val['mapping_type']:$val; // 关联类型
|
||||
$mappingClass = !empty($val['class_name'])?$val['class_name']:$key; // 关联类名
|
||||
$mappingFields = !empty($val['mapping_fields'])?$val['mapping_fields']:'*'; // 映射字段
|
||||
$mappingCondition = !empty($val['condition'])?$val['condition']:'1=1'; // 关联条件
|
||||
$mappingKey =!empty($val['mapping_key'])? $val['mapping_key'] : $this->getPk(); // 关联键名
|
||||
if(strtoupper($mappingClass)==strtoupper($this->name)) {
|
||||
// 自引用关联 获取父键名
|
||||
$mappingFk = !empty($val['parent_key'])? $val['parent_key'] : 'parent_id';
|
||||
}else{
|
||||
$mappingFk = !empty($val['foreign_key'])?$val['foreign_key']:strtolower($this->name).'_id'; // 关联外键
|
||||
}
|
||||
// 获取关联模型对象
|
||||
$model = D($mappingClass);
|
||||
switch($mappingType) {
|
||||
case self::HAS_ONE:
|
||||
$pk = $result[$mappingKey];
|
||||
$mappingCondition .= " AND {$mappingFk}='{$pk}'";
|
||||
$relationData = $model->where($mappingCondition)->field($mappingFields)->find();
|
||||
if (!empty($val['relation_deep'])){
|
||||
$model->getRelation($relationData,$val['relation_deep']);
|
||||
}
|
||||
break;
|
||||
case self::BELONGS_TO:
|
||||
if(strtoupper($mappingClass)==strtoupper($this->name)) {
|
||||
// 自引用关联 获取父键名
|
||||
$mappingFk = !empty($val['parent_key'])? $val['parent_key'] : 'parent_id';
|
||||
}else{
|
||||
$mappingFk = !empty($val['foreign_key'])?$val['foreign_key']:strtolower($model->getModelName()).'_id'; // 关联外键
|
||||
}
|
||||
$fk = $result[$mappingFk];
|
||||
$mappingCondition .= " AND {$model->getPk()}='{$fk}'";
|
||||
$relationData = $model->where($mappingCondition)->field($mappingFields)->find();
|
||||
if (!empty($val['relation_deep'])){
|
||||
$model->getRelation($relationData,$val['relation_deep']);
|
||||
}
|
||||
break;
|
||||
case self::HAS_MANY:
|
||||
$pk = $result[$mappingKey];
|
||||
$mappingCondition .= " AND {$mappingFk}='{$pk}'";
|
||||
$mappingOrder = !empty($val['mapping_order'])?$val['mapping_order']:'';
|
||||
$mappingLimit = !empty($val['mapping_limit'])?$val['mapping_limit']:'';
|
||||
// 延时获取关联记录
|
||||
$relationData = $model->where($mappingCondition)->field($mappingFields)->order($mappingOrder)->limit($mappingLimit)->select();
|
||||
if (!empty($val['relation_deep'])){
|
||||
foreach($relationData as $key=>$data){
|
||||
$model->getRelation($data,$val['relation_deep']);
|
||||
$relationData[$key] = $data;
|
||||
}
|
||||
}
|
||||
break;
|
||||
case self::MANY_TO_MANY:
|
||||
$pk = $result[$mappingKey];
|
||||
$prefix = $this->tablePrefix;
|
||||
$mappingCondition = " {$mappingFk}='{$pk}'";
|
||||
$mappingOrder = $val['mapping_order'];
|
||||
$mappingLimit = $val['mapping_limit'];
|
||||
$mappingRelationFk = $val['relation_foreign_key']?$val['relation_foreign_key']:$model->getModelName().'_id';
|
||||
if(isset($val['relation_table'])){
|
||||
$mappingRelationTable = preg_replace_callback("/__([A-Z_-]+)__/sU", function($match) use($prefix){ return $prefix.strtolower($match[1]);}, $val['relation_table']);
|
||||
}else{
|
||||
$mappingRelationTable = $this->getRelationTableName($model);
|
||||
}
|
||||
$sql = "SELECT b.{$mappingFields} FROM {$mappingRelationTable} AS a, ".$model->getTableName()." AS b WHERE a.{$mappingRelationFk} = b.{$model->getPk()} AND a.{$mappingCondition}";
|
||||
if(!empty($val['condition'])) {
|
||||
$sql .= ' AND '.$val['condition'];
|
||||
}
|
||||
if(!empty($mappingOrder)) {
|
||||
$sql .= ' ORDER BY '.$mappingOrder;
|
||||
}
|
||||
if(!empty($mappingLimit)) {
|
||||
$sql .= ' LIMIT '.$mappingLimit;
|
||||
}
|
||||
$relationData = $this->query($sql);
|
||||
if (!empty($val['relation_deep'])){
|
||||
foreach($relationData as $key=>$data){
|
||||
$model->getRelation($data,$val['relation_deep']);
|
||||
$relationData[$key] = $data;
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
if(!$return){
|
||||
if(isset($val['as_fields']) && in_array($mappingType,array(self::HAS_ONE,self::BELONGS_TO)) ) {
|
||||
// 支持直接把关联的字段值映射成数据对象中的某个字段
|
||||
// 仅仅支持HAS_ONE BELONGS_TO
|
||||
$fields = explode(',',$val['as_fields']);
|
||||
foreach ($fields as $field){
|
||||
if(strpos($field,':')) {
|
||||
list($relationName,$nick) = explode(':',$field);
|
||||
$result[$nick] = $relationData[$relationName];
|
||||
}else{
|
||||
$result[$field] = $relationData[$field];
|
||||
}
|
||||
}
|
||||
}else{
|
||||
$result[$mappingName] = $relationData;
|
||||
}
|
||||
unset($relationData);
|
||||
}else{
|
||||
return $relationData;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 操作关联数据
|
||||
* @access protected
|
||||
* @param string $opType 操作方式 ADD SAVE DEL
|
||||
* @param mixed $data 数据对象
|
||||
* @param string $name 关联名称
|
||||
* @return mixed
|
||||
*/
|
||||
protected function opRelation($opType,$data='',$name='') {
|
||||
$result = false;
|
||||
if(empty($data) && !empty($this->data)){
|
||||
$data = $this->data;
|
||||
}elseif(!is_array($data)){
|
||||
// 数据无效返回
|
||||
return false;
|
||||
}
|
||||
if(!empty($this->_link)) {
|
||||
// 遍历关联定义
|
||||
foreach($this->_link as $key=>$val) {
|
||||
// 操作制定关联类型
|
||||
$mappingName = $val['mapping_name']?$val['mapping_name']:$key; // 映射名称
|
||||
if(empty($name) || true === $name || $mappingName == $name || (is_array($name) && in_array($mappingName,$name)) ) {
|
||||
// 操作制定的关联
|
||||
$mappingType = !empty($val['mapping_type'])?$val['mapping_type']:$val; // 关联类型
|
||||
$mappingClass = !empty($val['class_name'])?$val['class_name']:$key; // 关联类名
|
||||
$mappingKey =!empty($val['mapping_key'])? $val['mapping_key'] : $this->getPk(); // 关联键名
|
||||
// 当前数据对象主键值
|
||||
$pk = $data[$mappingKey];
|
||||
if(strtoupper($mappingClass)==strtoupper($this->name)) {
|
||||
// 自引用关联 获取父键名
|
||||
$mappingFk = !empty($val['parent_key'])? $val['parent_key'] : 'parent_id';
|
||||
}else{
|
||||
$mappingFk = !empty($val['foreign_key'])?$val['foreign_key']:strtolower($this->name).'_id'; // 关联外键
|
||||
}
|
||||
if(!empty($val['condition'])) {
|
||||
$mappingCondition = $val['condition'];
|
||||
}else{
|
||||
$mappingCondition = array();
|
||||
$mappingCondition[$mappingFk] = $pk;
|
||||
}
|
||||
// 获取关联model对象
|
||||
$model = D($mappingClass);
|
||||
$mappingData = isset($data[$mappingName])?$data[$mappingName]:false;
|
||||
if(!empty($mappingData) || $opType == 'DEL') {
|
||||
switch($mappingType) {
|
||||
case self::HAS_ONE:
|
||||
switch (strtoupper($opType)){
|
||||
case 'ADD': // 增加关联数据
|
||||
$mappingData[$mappingFk] = $pk;
|
||||
$result = $model->add($mappingData);
|
||||
break;
|
||||
case 'SAVE': // 更新关联数据
|
||||
$result = $model->where($mappingCondition)->save($mappingData);
|
||||
break;
|
||||
case 'DEL': // 根据外键删除关联数据
|
||||
$result = $model->where($mappingCondition)->delete();
|
||||
break;
|
||||
}
|
||||
break;
|
||||
case self::BELONGS_TO:
|
||||
break;
|
||||
case self::HAS_MANY:
|
||||
switch (strtoupper($opType)){
|
||||
case 'ADD' : // 增加关联数据
|
||||
$model->startTrans();
|
||||
foreach ($mappingData as $val){
|
||||
$val[$mappingFk] = $pk;
|
||||
$result = $model->add($val);
|
||||
}
|
||||
$model->commit();
|
||||
break;
|
||||
case 'SAVE' : // 更新关联数据
|
||||
$model->startTrans();
|
||||
$pk = $model->getPk();
|
||||
foreach ($mappingData as $vo){
|
||||
if(isset($vo[$pk])) {// 更新数据
|
||||
$mappingCondition = "$pk ={$vo[$pk]}";
|
||||
$result = $model->where($mappingCondition)->save($vo);
|
||||
}else{ // 新增数据
|
||||
$vo[$mappingFk] = $data[$mappingKey];
|
||||
$result = $model->add($vo);
|
||||
}
|
||||
}
|
||||
$model->commit();
|
||||
break;
|
||||
case 'DEL' : // 删除关联数据
|
||||
$result = $model->where($mappingCondition)->delete();
|
||||
break;
|
||||
}
|
||||
break;
|
||||
case self::MANY_TO_MANY:
|
||||
$mappingRelationFk = $val['relation_foreign_key']?$val['relation_foreign_key']:$model->getModelName().'_id';// 关联
|
||||
$prefix = $this->tablePrefix;
|
||||
if(isset($val['relation_table'])){
|
||||
$mappingRelationTable = preg_replace_callback("/__([A-Z_-]+)__/sU", function($match) use($prefix){ return $prefix.strtolower($match[1]);}, $val['relation_table']);
|
||||
}else{
|
||||
$mappingRelationTable = $this->getRelationTableName($model);
|
||||
}
|
||||
if(is_array($mappingData)) {
|
||||
$ids = array();
|
||||
foreach ($mappingData as $vo)
|
||||
$ids[] = $vo[$mappingKey];
|
||||
$relationId = implode(',',$ids);
|
||||
}
|
||||
switch (strtoupper($opType)){
|
||||
case 'ADD': // 增加关联数据
|
||||
if(isset($relationId)) {
|
||||
$this->startTrans();
|
||||
// 插入关联表数据
|
||||
$sql = 'INSERT INTO '.$mappingRelationTable.' ('.$mappingFk.','.$mappingRelationFk.') SELECT a.'.$this->getPk().',b.'.$model->getPk().' FROM '.$this->getTableName().' AS a ,'.$model->getTableName()." AS b where a.".$this->getPk().' ='. $pk.' AND b.'.$model->getPk().' IN ('.$relationId.") ";
|
||||
$result = $model->execute($sql);
|
||||
if(false !== $result)
|
||||
// 提交事务
|
||||
$this->commit();
|
||||
else
|
||||
// 事务回滚
|
||||
$this->rollback();
|
||||
}
|
||||
break;
|
||||
case 'SAVE': // 更新关联数据
|
||||
if(isset($relationId)) {
|
||||
$this->startTrans();
|
||||
// 删除关联表数据
|
||||
$this->table($mappingRelationTable)->where($mappingCondition)->delete();
|
||||
// 插入关联表数据
|
||||
$sql = 'INSERT INTO '.$mappingRelationTable.' ('.$mappingFk.','.$mappingRelationFk.') SELECT a.'.$this->getPk().',b.'.$model->getPk().' FROM '.$this->getTableName().' AS a ,'.$model->getTableName()." AS b where a.".$this->getPk().' ='. $pk.' AND b.'.$model->getPk().' IN ('.$relationId.") ";
|
||||
$result = $model->execute($sql);
|
||||
if(false !== $result)
|
||||
// 提交事务
|
||||
$this->commit();
|
||||
else
|
||||
// 事务回滚
|
||||
$this->rollback();
|
||||
}
|
||||
break;
|
||||
case 'DEL': // 根据外键删除中间表关联数据
|
||||
$result = $this->table($mappingRelationTable)->where($mappingCondition)->delete();
|
||||
break;
|
||||
}
|
||||
break;
|
||||
}
|
||||
if (!empty($val['relation_deep'])){
|
||||
$model->opRelation($opType,$mappingData,$val['relation_deep']);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 进行关联查询
|
||||
* @access public
|
||||
* @param mixed $name 关联名称
|
||||
* @return Model
|
||||
*/
|
||||
public function relation($name) {
|
||||
$this->options['link'] = $name;
|
||||
return $this;
|
||||
}
|
||||
|
||||
/**
|
||||
* 关联数据获取 仅用于查询后
|
||||
* @access public
|
||||
* @param string $name 关联名称
|
||||
* @return array
|
||||
*/
|
||||
public function relationGet($name) {
|
||||
if(empty($this->data))
|
||||
return false;
|
||||
return $this->getRelation($this->data,$name,true);
|
||||
}
|
||||
}
|
||||
243
ThinkPHP/Library/Think/Model/ViewModel.class.php
Normal file
243
ThinkPHP/Library/Think/Model/ViewModel.class.php
Normal file
@@ -0,0 +1,243 @@
|
||||
<?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\Model;
|
||||
use Think\Model;
|
||||
/**
|
||||
* ThinkPHP视图模型扩展
|
||||
*/
|
||||
class ViewModel extends Model {
|
||||
|
||||
protected $viewFields = array();
|
||||
|
||||
/**
|
||||
* 自动检测数据表信息
|
||||
* @access protected
|
||||
* @return void
|
||||
*/
|
||||
protected function _checkTableInfo() {}
|
||||
|
||||
/**
|
||||
* 得到完整的数据表名
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function getTableName() {
|
||||
if(empty($this->trueTableName)) {
|
||||
$tableName = '';
|
||||
foreach ($this->viewFields as $key=>$view){
|
||||
// 获取数据表名称
|
||||
if(isset($view['_table'])) { // 2011/10/17 添加实际表名定义支持 可以实现同一个表的视图
|
||||
$tableName .= $view['_table'];
|
||||
$prefix = $this->tablePrefix;
|
||||
$tableName = preg_replace_callback("/__([A-Z_-]+)__/sU", function($match) use($prefix){ return $prefix.strtolower($match[1]);}, $tableName);
|
||||
}else{
|
||||
$class = $key.'Model';
|
||||
$Model = class_exists($class)?new $class():M($key);
|
||||
$tableName .= $Model->getTableName();
|
||||
}
|
||||
// 表别名定义
|
||||
$tableName .= !empty($view['_as'])?' '.$view['_as']:' '.$key;
|
||||
// 支持ON 条件定义
|
||||
$tableName .= !empty($view['_on'])?' ON '.$view['_on']:'';
|
||||
// 指定JOIN类型 例如 RIGHT INNER LEFT 下一个表有效
|
||||
$type = !empty($view['_type'])?$view['_type']:'';
|
||||
$tableName .= ' '.strtoupper($type).' JOIN ';
|
||||
$len = strlen($type.'_JOIN ');
|
||||
}
|
||||
$tableName = substr($tableName,0,-$len);
|
||||
$this->trueTableName = $tableName;
|
||||
}
|
||||
return $this->trueTableName;
|
||||
}
|
||||
|
||||
/**
|
||||
* 表达式过滤方法
|
||||
* @access protected
|
||||
* @param string $options 表达式
|
||||
* @return void
|
||||
*/
|
||||
protected function _options_filter(&$options) {
|
||||
if(isset($options['field']))
|
||||
$options['field'] = $this->checkFields($options['field']);
|
||||
else
|
||||
$options['field'] = $this->checkFields();
|
||||
if(isset($options['group']))
|
||||
$options['group'] = $this->checkGroup($options['group']);
|
||||
if(isset($options['where']))
|
||||
$options['where'] = $this->checkCondition($options['where']);
|
||||
if(isset($options['order']))
|
||||
$options['order'] = $this->checkOrder($options['order']);
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查是否定义了所有字段
|
||||
* @access protected
|
||||
* @param string $name 模型名称
|
||||
* @param array $fields 字段数组
|
||||
* @return array
|
||||
*/
|
||||
private function _checkFields($name,$fields) {
|
||||
if(false !== $pos = array_search('*',$fields)) {// 定义所有字段
|
||||
$fields = array_merge($fields,M($name)->getDbFields());
|
||||
unset($fields[$pos]);
|
||||
}
|
||||
return $fields;
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查条件中的视图字段
|
||||
* @access protected
|
||||
* @param mixed $data 条件表达式
|
||||
* @return array
|
||||
*/
|
||||
protected function checkCondition($where) {
|
||||
if(is_array($where)) {
|
||||
$view = array();
|
||||
// 检查视图字段
|
||||
foreach ($this->viewFields as $key=>$val){
|
||||
$k = isset($val['_as'])?$val['_as']:$key;
|
||||
$val = $this->_checkFields($key,$val);
|
||||
foreach ($where as $name=>$value){
|
||||
if(false !== $field = array_search($name,$val,true)) {
|
||||
// 存在视图字段
|
||||
$_key = is_numeric($field)? $k.'.'.$name : $k.'.'.$field;
|
||||
$view[$_key] = $value;
|
||||
unset($where[$name]);
|
||||
}
|
||||
}
|
||||
}
|
||||
$where = array_merge($where,$view);
|
||||
}
|
||||
return $where;
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查Order表达式中的视图字段
|
||||
* @access protected
|
||||
* @param string $order 字段
|
||||
* @return string
|
||||
*/
|
||||
protected function checkOrder($order='') {
|
||||
if(is_string($order) && !empty($order)) {
|
||||
$orders = explode(',',$order);
|
||||
$_order = array();
|
||||
foreach ($orders as $order){
|
||||
$array = explode(' ',trim($order));
|
||||
$field = $array[0];
|
||||
$sort = isset($array[1])?$array[1]:'ASC';
|
||||
// 解析成视图字段
|
||||
foreach ($this->viewFields as $name=>$val){
|
||||
$k = isset($val['_as'])?$val['_as']:$name;
|
||||
$val = $this->_checkFields($name,$val);
|
||||
if(false !== $_field = array_search($field,$val,true)) {
|
||||
// 存在视图字段
|
||||
$field = is_numeric($_field)?$k.'.'.$field:$k.'.'.$_field;
|
||||
break;
|
||||
}
|
||||
}
|
||||
$_order[] = $field.' '.$sort;
|
||||
}
|
||||
$order = implode(',',$_order);
|
||||
}
|
||||
return $order;
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查Group表达式中的视图字段
|
||||
* @access protected
|
||||
* @param string $group 字段
|
||||
* @return string
|
||||
*/
|
||||
protected function checkGroup($group='') {
|
||||
if(!empty($group)) {
|
||||
$groups = explode(',',$group);
|
||||
$_group = array();
|
||||
foreach ($groups as $field){
|
||||
// 解析成视图字段
|
||||
foreach ($this->viewFields as $name=>$val){
|
||||
$k = isset($val['_as'])?$val['_as']:$name;
|
||||
$val = $this->_checkFields($name,$val);
|
||||
if(false !== $_field = array_search($field,$val,true)) {
|
||||
// 存在视图字段
|
||||
$field = is_numeric($_field)?$k.'.'.$field:$k.'.'.$_field;
|
||||
break;
|
||||
}
|
||||
}
|
||||
$_group[] = $field;
|
||||
}
|
||||
$group = implode(',',$_group);
|
||||
}
|
||||
return $group;
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查fields表达式中的视图字段
|
||||
* @access protected
|
||||
* @param string $fields 字段
|
||||
* @return string
|
||||
*/
|
||||
protected function checkFields($fields='') {
|
||||
if(empty($fields) || '*'==$fields ) {
|
||||
// 获取全部视图字段
|
||||
$fields = array();
|
||||
foreach ($this->viewFields as $name=>$val){
|
||||
$k = isset($val['_as'])?$val['_as']:$name;
|
||||
$val = $this->_checkFields($name,$val);
|
||||
foreach ($val as $key=>$field){
|
||||
if(is_numeric($key)) {
|
||||
$fields[] = $k.'.'.$field.' AS '.$field;
|
||||
}elseif('_' != substr($key,0,1)) {
|
||||
// 以_开头的为特殊定义
|
||||
if( false !== strpos($key,'*') || false !== strpos($key,'(') || false !== strpos($key,'.')) {
|
||||
//如果包含* 或者 使用了sql方法 则不再添加前面的表名
|
||||
$fields[] = $key.' AS '.$field;
|
||||
}else{
|
||||
$fields[] = $k.'.'.$key.' AS '.$field;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
$fields = implode(',',$fields);
|
||||
}else{
|
||||
if(!is_array($fields))
|
||||
$fields = explode(',',$fields);
|
||||
// 解析成视图字段
|
||||
$array = array();
|
||||
foreach ($fields as $key=>$field){
|
||||
if(strpos($field,'(') || strpos(strtolower($field),' as ')){
|
||||
// 使用了函数或者别名
|
||||
$array[] = $field;
|
||||
unset($fields[$key]);
|
||||
}
|
||||
}
|
||||
foreach ($this->viewFields as $name=>$val){
|
||||
$k = isset($val['_as'])?$val['_as']:$name;
|
||||
$val = $this->_checkFields($name,$val);
|
||||
foreach ($fields as $key=>$field){
|
||||
if(false !== $_field = array_search($field,$val,true)) {
|
||||
// 存在视图字段
|
||||
if(is_numeric($_field)) {
|
||||
$array[] = $k.'.'.$field.' AS '.$field;
|
||||
}elseif('_' != substr($_field,0,1)){
|
||||
if( false !== strpos($_field,'*') || false !== strpos($_field,'(') || false !== strpos($_field,'.'))
|
||||
//如果包含* 或者 使用了sql方法 则不再添加前面的表名
|
||||
$array[] = $_field.' AS '.$field;
|
||||
else
|
||||
$array[] = $k.'.'.$_field.' AS '.$field;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
$fields = implode(',',$array);
|
||||
}
|
||||
return $fields;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user