+Promise
This commit is contained in:
eshengsky
@@ -9,66 +9,54 @@ const i18n = require('../models/i18n');
|
||||
|
||||
/**
|
||||
* 获取分类数据
|
||||
* @param [cached] 是否读取缓存
|
||||
* @param callback 回调函数
|
||||
* @param [cached=true] 是否读取缓存
|
||||
*/
|
||||
exports.getAll = function (cached, callback) {
|
||||
if (typeof cached === 'function') {
|
||||
callback = cached;
|
||||
cached = true;
|
||||
}
|
||||
|
||||
exports.getAll = (cached = true) => {
|
||||
// 缓存的key名称
|
||||
const cache_key = 'categories';
|
||||
if (cached) {
|
||||
// 尝试读取缓存
|
||||
redisClient.getItem(cache_key, (err, categories) => {
|
||||
// 读取缓存出错
|
||||
if (err) {
|
||||
return callback(err);
|
||||
}
|
||||
|
||||
// 缓存中有数据
|
||||
if (categories) {
|
||||
return callback(null, categories);
|
||||
}
|
||||
|
||||
// 缓存中没有数据，则从数据库中读取
|
||||
categoryModel.find((err, categories) => {
|
||||
// 读取数据库出错
|
||||
if (err) {
|
||||
return callback(err);
|
||||
}
|
||||
|
||||
// 从数据库中读到数据
|
||||
return new Promise((resolve, reject) => {
|
||||
if (cached) {
|
||||
// 尝试读取缓存
|
||||
redisClient.getItem(cache_key).then(categories => {
|
||||
// 缓存中有数据
|
||||
if (categories) {
|
||||
// 将数据塞入缓存
|
||||
redisClient.setItem(cache_key, categories, redisClient.defaultExpired, err => {
|
||||
if (err) {
|
||||
return callback(err);
|
||||
}
|
||||
});
|
||||
return resolve(categories);
|
||||
}
|
||||
return callback(null, categories);
|
||||
|
||||
// 缓存中没有数据，则从数据库中读取
|
||||
categoryModel.find((err, categories) => {
|
||||
// 读取数据库出错
|
||||
if (err) {
|
||||
return reject(err);
|
||||
}
|
||||
|
||||
// 从数据库中读到数据
|
||||
if (categories) {
|
||||
// 将数据塞入缓存
|
||||
redisClient.setItem(cache_key, categories, redisClient.defaultExpired);
|
||||
}
|
||||
return resolve(categories);
|
||||
});
|
||||
}, err => {
|
||||
// 读取缓存出错
|
||||
reject(err);
|
||||
});
|
||||
});
|
||||
} else {
|
||||
categoryModel.find((err, categories) => {
|
||||
if (err) {
|
||||
return callback(err);
|
||||
}
|
||||
return callback(null, categories);
|
||||
});
|
||||
}
|
||||
} else {
|
||||
categoryModel.find((err, categories) => {
|
||||
if (err) {
|
||||
return reject(err);
|
||||
}
|
||||
return resolve(categories);
|
||||
});
|
||||
}
|
||||
});
|
||||
};
|
||||
|
||||
/**
|
||||
* 根据分类alias获取分类
|
||||
* @param alias 分类alias
|
||||
* @param callback 回调函数
|
||||
* @returns {*}
|
||||
*/
|
||||
exports.getByAlias = function (alias, callback) {
|
||||
exports.getByAlias = alias => {
|
||||
// 全部分类
|
||||
const cateAll = {
|
||||
_id: '',
|
||||
@@ -86,46 +74,41 @@ exports.getByAlias = function (alias, callback) {
|
||||
};
|
||||
|
||||
const cache_key = `category_${alias}`;
|
||||
if (alias) {
|
||||
if (alias === 'other') {
|
||||
return callback(null, cateOther);
|
||||
}
|
||||
redisClient.getItem(cache_key, (err, category) => {
|
||||
if (err) {
|
||||
return callback(err);
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
if (alias) {
|
||||
if (alias === 'other') {
|
||||
return resolve(cateOther);
|
||||
}
|
||||
if (category) {
|
||||
return callback(null, category);
|
||||
}
|
||||
categoryModel.findOne({ Alias: alias }, (err, category) => {
|
||||
if (err) {
|
||||
return callback(err);
|
||||
}
|
||||
redisClient.getItem(cache_key).then(category => {
|
||||
if (category) {
|
||||
redisClient.setItem(cache_key, category, redisClient.defaultExpired, err => {
|
||||
if (err) {
|
||||
return callback(err);
|
||||
}
|
||||
});
|
||||
return resolve(category);
|
||||
}
|
||||
return callback(null, category);
|
||||
categoryModel.findOne({ Alias: alias }, (err, category) => {
|
||||
if (err) {
|
||||
return reject(err);
|
||||
}
|
||||
if (category) {
|
||||
redisClient.setItem(cache_key, category, redisClient.defaultExpired);
|
||||
}
|
||||
return resolve(category);
|
||||
});
|
||||
}, err => {
|
||||
reject(err);
|
||||
});
|
||||
});
|
||||
} else {
|
||||
return callback(null, cateAll);
|
||||
}
|
||||
} else {
|
||||
resolve(cateAll);
|
||||
}
|
||||
});
|
||||
};
|
||||
|
||||
/**
|
||||
* 保存分类数据
|
||||
* @param array 分类集合
|
||||
* @param callback 回调函数
|
||||
*/
|
||||
exports.save = function (array, callback) {
|
||||
let jsonArray = [],
|
||||
toUpdate = [],
|
||||
updateQuery = [],
|
||||
cateNew;
|
||||
exports.save = array => {
|
||||
const jsonArray = [];
|
||||
const toUpdate = [];
|
||||
if (array.length > 0) {
|
||||
array.forEach(item => {
|
||||
jsonArray.push({
|
||||
@@ -139,58 +122,63 @@ exports.save = function (array, callback) {
|
||||
});
|
||||
});
|
||||
}
|
||||
categoryModel.find((err, categories) => {
|
||||
if (err) {
|
||||
return callback(err);
|
||||
}
|
||||
categories.forEach(old => {
|
||||
cateNew = tool.jsonQuery(jsonArray, { _id: old._id });
|
||||
if (!cateNew) {
|
||||
// 该分类将被删除
|
||||
toUpdate.push(old._id);
|
||||
} else {
|
||||
// 该分类依然存在，则创建时间沿用原创建时间
|
||||
cateNew.CreateTime = old.CreateTime;
|
||||
|
||||
// 若该分类未做任何修改，则修改时间沿用原修改时间
|
||||
if (cateNew.CateName.toString() === old.CateName.toString() && cateNew.Alias.toString() === old.Alias.toString() && cateNew.Img === old.Img && cateNew.Link === old.Link) {
|
||||
cateNew.ModifyTime = old.ModifyTime;
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
// 将已被删除分类的文章设为"未分类"
|
||||
if (toUpdate.length > 0) {
|
||||
toUpdate.forEach(cateId => {
|
||||
updateQuery.push({
|
||||
CategoryId: cateId
|
||||
});
|
||||
});
|
||||
post.update({ $or: updateQuery }, { CategoryId: 'other' }, { multi: true }, err => {
|
||||
if (err) {
|
||||
return callback(err);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
// 将分类全部删除
|
||||
categoryModel.remove(err => {
|
||||
return new Promise((resolve, reject) => {
|
||||
categoryModel.find((err, categories) => {
|
||||
if (err) {
|
||||
return callback(err);
|
||||
return reject(err);
|
||||
}
|
||||
if (jsonArray.length > 0) {
|
||||
// 插入全部分类
|
||||
// categoryModel.create(jsonArray, function (err) {}); //不用这个，因为这个内部实现依然是循环插入，不是真正的批量插入
|
||||
// 这里采用mongodb原生的insert来批量插入多个文档
|
||||
categoryModel.collection.insert(jsonArray, err => {
|
||||
if (err) {
|
||||
return callback(err);
|
||||
|
||||
categories.forEach(old => {
|
||||
const cateNew = tool.jsonQuery(jsonArray, { _id: old._id });
|
||||
if (!cateNew) {
|
||||
// 该分类将被删除
|
||||
toUpdate.push(old._id);
|
||||
} else {
|
||||
// 该分类依然存在，则创建时间沿用原创建时间
|
||||
cateNew.CreateTime = old.CreateTime;
|
||||
|
||||
// 若该分类未做任何修改，则修改时间沿用原修改时间
|
||||
if (cateNew.CateName.toString() === old.CateName.toString() && cateNew.Alias.toString() === old.Alias.toString() && cateNew.Img === old.Img && cateNew.Link === old.Link) {
|
||||
cateNew.ModifyTime = old.ModifyTime;
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
// 将已被删除分类的文章设为"未分类"
|
||||
if (toUpdate.length > 0) {
|
||||
const updateQuery = [];
|
||||
toUpdate.forEach(cateId => {
|
||||
updateQuery.push({
|
||||
CategoryId: cateId
|
||||
});
|
||||
});
|
||||
post.update({ $or: updateQuery }, { CategoryId: 'other' }, { multi: true }, err => {
|
||||
if (err) {
|
||||
return reject(err);
|
||||
}
|
||||
return callback(null);
|
||||
});
|
||||
} else {
|
||||
return callback(null);
|
||||
}
|
||||
|
||||
// 将分类全部删除
|
||||
categoryModel.remove(err => {
|
||||
if (err) {
|
||||
return reject(err);
|
||||
}
|
||||
if (jsonArray.length > 0) {
|
||||
// 插入全部分类
|
||||
// categoryModel.create(jsonArray, function (err) {}); //不用这个，因为这个内部实现依然是循环插入，不是真正的批量插入
|
||||
// 这里采用mongodb原生的insert来批量插入多个文档
|
||||
categoryModel.collection.insert(jsonArray, err => {
|
||||
if (err) {
|
||||
return reject(err);
|
||||
}
|
||||
return resolve();
|
||||
});
|
||||
} else {
|
||||
return resolve();
|
||||
}
|
||||
});
|
||||
});
|
||||
});
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user