blackteay/ThinkAdmin
1
0
Fork 0
mirror of https://gitee.com/zoujingli/ThinkAdmin.git synced 2025-04-06 03:58:04 +08:00
Code Issues Projects Releases Wiki Activity

[更新]优化数据服务组件

Browse Source
This commit is contained in:
Anyon 2017-06-20 12:13:55 +08:00
parent 0c012e2056
commit dd05c100bb
1 changed files with 26 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

71
extend/service/DataService.php
View File

@ -15,6 +15,7 @@
namespace service;
use think\Db;
use think\Request;
/**
* 基础数据服务
@ -45,14 +46,16 @@ class DataService {
public static function createSequence($length = 10, $type = 'SYSTEM') {
$times = 0;
while ($times++ < 10) {
$sequence = '';
$i = 0;
$sequence = '';
while ($i++ < $length) {
$sequence .= ($i <= 1 ? rand(1, 9) : rand(0, 9));
}
$data = ['sequence' => $sequence, 'type' => strtoupper($type)];
if (Db::name('SystemSequence')->where($data)->count() < 1 && Db::name('SystemSequence')->insert($data)) {
return $sequence;
if (Db::name('SystemSequence')->where($data)->count() < 1) {
if (Db::name('SystemSequence')->insert($data) !== false) {
return $sequence;
}
}
}
return null;
@ -62,40 +65,17 @@ class DataService {
* 数据增量保存
* @param \think\db\Query|string $dbQuery 数据查询对象
* @param array $data 需要保存或更新的数据
* @param string $upkey 条件主键限制
* @param string $key 条件主键限制
* @param array $where 其它的where条件
* @return bool
*/
public static function save($dbQuery, $data, $upkey = 'id', $where = []) {
public static function save($dbQuery, $data, $key = 'id', $where = []) {
$db = is_string($dbQuery) ? Db::name($dbQuery) : $dbQuery;
$fields = $db->getTableFields(['table' => $db->getTable()]);
$_data = [];
foreach ($data as $k => $v) {
in_array($k, $fields) && ($_data[$k] = $v);
$where[$key] = isset($data[$key]) ? $data[$key] : '';
if ($db->where($where)->count() > 0) {
return $db->where($where)->update($data) !== false;
}
if (self::_apply_save_where($db, $data, $upkey, $where)->count() > 0) {
return self::_apply_save_where($db, $data, $upkey, $where)->update($_data) !== FALSE;
}
return self::_apply_save_where($db, $data, $upkey, $where)->insert($_data) !== FALSE;
}
/**
* 应用 where 条件
* @param \think\db\Query|string $db 数据查询对象
* @param array $data 需要保存或更新的数据
* @param string $upkey 条件主键限制
* @param array $where 其它的where条件
* @return \think\db\Query
*/
protected static function _apply_save_where(&$db, $data, $upkey, $where) {
foreach (is_string($upkey) ? explode(',', $upkey) : $upkey as $v) {
if (is_string($v) && array_key_exists($v, $data)) {
$db->where($v, $data[$v]);
} elseif (is_string($v)) {
$db->where("{$v} IS NULL");
}
}
return $db->where($where);
return $db->insert($data) !== false;
}
/**
@ -105,23 +85,24 @@ class DataService {
* @return bool|null
*/
public static function update(&$dbQuery, $where = []) {
$request = Request::instance();
$db = is_string($dbQuery) ? Db::name($dbQuery) : $dbQuery;
$ids = explode(',', input("post.id", ''));
$field = input('post.field', '');
$value = input('post.value', '');
$ids = explode(',', $request->post('id', ''));
$field = $request->post('field', '');
$value = $request->post('value', '');
$pk = $db->getPk(['table' => $db->getTable()]);
$db->where(empty($pk) ? 'id' : $pk, 'in', $ids);
!empty($where) && $db->where($where);
// 删除模式
$where[empty($pk) ? 'id' : $pk] = ['in', $ids];
// 删除模式，如果存在 is_deleted 字段使用软删除
if ($field === 'delete') {
$fields = $db->getTableFields(['table' => $db->getTable()]);
if (in_array('is_deleted', $fields)) {
return false !== $db->update(['is_deleted' => 1]);
if (method_exists($db, 'getTableFields')) {
if (in_array('is_deleted', $db->getTableFields(['table' => $db->getTable()]))) {
return false !== $db->where($where)->update(['is_deleted' => 1]);
}
}
return false !== $db->delete();
return false !== $db->where($where)->delete();
}
// 更新模式
return false !== $db->update([$field => $value]);
// 更新模式，更新指定字段内容
return false !== $db->where($where)->update([$field => $value]);
}
}
}