/**
* 上锁
*
* @param string $key 要上的锁的key
* @param bool $wouldblock 是否堵塞
*
* @return mixed
*/
public function lock($key, $wouldblock = false)
{
if (empty($key)) {
return false;
}
$key = $this->getKey($key);
if (isset(self::$lockCache[$key]) && self::$lockCache[$key] == Model::getInstance()->cache($this->userCache)->getInstance()->get($key)) {
return true;
}
$driverType = Model::getInstance()->cache($this->userCache)->getDriverType();
if ($driverType === 1) {
//memcached
$isLock = Model::getInstance()->cache($this->userCache)->getInstance()->add($key, (string) Alpha::$nowMicroTime, $this->expire);
} else {
//memcache
$isLock = Model::getInstance()->cache($this->userCache)->getInstance()->add($key, (string) Alpha::$nowMicroTime, 0, $this->expire);
}
if ($isLock) {
self::$lockCache[$key] = (string) Alpha::$nowMicroTime;
return true;
}
//非堵塞模式
if (!$wouldblock) {
return false;
}
//堵塞模式
do {
usleep(200);
if ($driverType === 1) {
//memcached
$isLock = Model::getInstance()->cache($this->userCache)->getInstance()->add($key, (string) Alpha::$nowMicroTime, $this->expire);
} else {
//memcache
$isLock = Model::getInstance()->cache($this->userCache)->getInstance()->add($key, (string) Alpha::$nowMicroTime, 0, $this->expire);
}
} while (!$isLock);
self::$lockCache[$key] = (string) Alpha::$nowMicroTime;
return true;
}
/**
* 上锁
*
* @param string $key 要上的锁的key
* @param bool $wouldblock 是否堵塞
*
* @return mixed
*/
public function lock($key, $wouldblock = false)
{
if (empty($key)) {
return false;
}
$key = $this->getKey($key);
if (isset(self::$lockCache[$key]) && self::$lockCache[$key] == Model::getInstance()->cache($this->userCache)->getInstance()->get($key)) {
return true;
}
if (Model::getInstance()->cache($this->userCache)->getInstance()->set($key, Alpha::$nowMicroTime, array('nx', 'ex' => $this->expire))) {
self::$lockCache[$key] = (string) Alpha::$nowMicroTime;
return true;
}
//非堵塞模式
if (!$wouldblock) {
return false;
}
//堵塞模式
do {
usleep(200);
} while (!Model::getInstance()->cache($this->userCache)->getInstance()->set($key, Alpha::$nowMicroTime, array('nx', 'ex' => $this->expire)));
self::$lockCache[$key] = (string) Alpha::$nowMicroTime;
return true;
}
/**
* 设置cache版本号
*
* @param $table
*/
public function setCacheVer($table)
{
Model::getInstance()->cache()->set('db_cache_version_' . $table, microtime(true), $this->conf['cache_expire']);
}
/**
* 获取有权限的菜单列表
*
* @return array
*/
public static function getMenus()
{
$res = array();
$authInfo = self::getLoginInfo();
if (!$authInfo) {
return $res;
}
//登录超时
Model::getInstance()->db()->table(array('menus' => 'm'))->columns(array('m.id', 'm.pid', 'm.title', 'm.url'));
//当前登录用户是否为超级管理员
if (Config::get('administratorid') !== intval($authInfo['id'])) {
Model::getInstance()->db()->join(array('access' => 'a'), 'a.menuid=m.id')->lBrackets()->whereIn('a.groupid', $authInfo['groupid'])->_or()->where('a.userid', $authInfo['id'])->rBrackets()->_and();
}
$result = Model::getInstance()->db()->where('m.isshow', 1)->orderBy('m.sort', 'DESC')->orderBy('m.id', 'ASC')->limit(0, 2000)->select();
$res = Tree::getTreeNoFormat($result, 0);
return $res;
}
/**
* 返回驱动
*
* @return \Redis
*/
private function getDriver()
{
return Model::getInstance()->cache(Config::get('queue_use_cache'))->getInstance();
}
/**
* 任意等级的日志记录
*
* @param mixed $level
* @param string $message
* @param array $context
*
* @return null
*/
public function log($level, $message, array $context = array())
{
return Model::getInstance()->cache(Config::get('log_use_cache'))->getInstance()->lPush(Config::get('log_prefix') . '_' . $level, $this->format($message, $context));
}
/**
* 获取模型方法
*
* @return \Foundation\Model
*/
public function model()
{
return Model::getInstance();
}
/**
* 定义析构函数 自动释放获得的锁
*
*/
public function __destruct()
{
foreach (self::$lockCache as $key => $isMyLock) {
if ($isMyLock == Model::getInstance()->cache($this->userCache)->getInstance()->get($key)) {
Model::getInstance()->cache($this->userCache)->getInstance()->delete($key);
}
self::$lockCache[$key] = null;
//防止gc延迟,判断有误
unset(self::$lockCache[$key]);
}
}
/**
* 获取多条数据
*
* @param int $offset 偏移量
* @param int $limit 返回的条数
*
* @return array
*/
public function select($offset = null, $limit = null)
{
is_null($offset) || $this->limit($offset, $limit);
$this->sql['columns'] == '' && ($this->sql['columns'] = '*');
$columns = $this->sql['columns'] == '*' ? Config::get('db_fields_cache') ? $this->getDbFields($this->getRealTableName(key($this->table)), null, 1) : '*' : $this->sql['columns'];
$table = $operator = $cacheKey = '';
foreach ($this->table as $key => $val) {
$realTable = $this->getRealTableName($key);
$cacheKey .= $this->getCacheVer($realTable);
$on = null;
if (isset($this->join[$key])) {
$operator = ' INNER JOIN';
$on = $this->join[$key];
} elseif (isset($this->leftJoin[$key])) {
$operator = ' LEFT JOIN';
$on = $this->leftJoin[$key];
} elseif (isset($this->rightJoin[$key])) {
$operator = ' RIGHT JOIN';
$on = $this->rightJoin[$key];
} else {
empty($table) || ($operator = ' ,');
}
if (is_null($val)) {
$table .= "{$operator} `{$realTable}`";
} else {
$table .= "{$operator} `{$realTable}` AS `{$val}`";
}
is_null($on) || ($table .= " ON {$on}");
}
empty($table) && \Foundation\throwException(Lang::get('_PARSE_SQL_ERROR_NO_TABLE_', 'select'));
empty($this->sql['limit']) && ($this->sql['limit'] = "LIMIT 0, 100");
$sql = "SELECT {$columns} FROM {$table} " . $this->sql['where'] . $this->sql['groupBy'] . $this->sql['having'] . $this->sql['orderBy'] . $this->sql['limit'] . $this->union;
$cacheKey = md5($sql . json_encode($this->bindParams)) . $cacheKey;
$return = Model::getInstance()->cache()->get($cacheKey);
if ($return === false) {
$stmt = $this->prepare($sql, $this->rlink);
$this->execute($stmt);
$return = $stmt->fetchAll(\PDO::FETCH_ASSOC);
Model::getInstance()->cache()->set($cacheKey, $return, $this->conf['cache_expire']);
} else {
$this->reset();
$this->bindParams = array();
}
return $return;
}
