counter.js 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778
  1. /**
  2. * Created by Tony on 2017/3/21.
  3. */
  4. var mongoose = require('mongoose');
  5. var counterModel = mongoose.model("counters");
  6. // All counter Must predefine in DB
  7. const COUNTER_MODULE_NAME = {
  8. project: 'projects',
  9. user: 'users',
  10. bills: 'bills',
  11. rations: 'rations',
  12. GLJ: 'glj',
  13. rationMap: 'rationMaps',
  14. rationTree: 'rationChapterTrees',
  15. report: 'rptTemplates',
  16. fee: 'fees',
  17. unitPriceFile: 'unitPriceFile',
  18. unitPriceGLJ: 'unitPriceGLJ',
  19. template_bills: 'temp_bills',
  20. billsLib: 'billsLib',
  21. coeList: 'coeList',
  22. complementaryCoeList: 'complementary_coe_list',
  23. glj_list: 'glj_list',
  24. mix_ratio: 'mix_ratio',
  25. unit_price_file: 'unit_price_file',
  26. unit_price: 'unit_price'
  27. }
  28. /*const PROJECT_COUNTER = 'projects', USER_COUNTER = 'users', BILL_COUNTER = 'bills', RATION_COUNTER = 'rations',
  29. REPORT_COUNTER = 'rptTemplates', FEE_COUNTER = 'fees'*/
  30. var counterDAO = function(){};
  31. /*
  32. * callback = function (err, result) {
  33. * result.value.sequence_value Ϊ�޸ĺ��id
  34. * }
  35. */
  36. counterDAO.prototype.getIDAfterCount = function(moduleName, stepCount, callback) {
  37. var sc = stepCount;
  38. if (isNaN(stepCount) || (stepCount < 0)) {
  39. sc = 1;
  40. } else if (!(/^-?\d+$/.test(stepCount))) {
  41. sc = Math.round(stepCount + 0.5);
  42. }
  43. counterModel.findOneAndUpdate({_id: moduleName}, { $inc: { sequence_value: sc } }, {'new':true}, callback);//findOneAndUpdate
  44. }
  45. counterDAO.prototype.getIDAfterCountSync = async function (moduleName, stepCount) {
  46. let sc = stepCount;
  47. if (isNaN(stepCount) || (stepCount < 0)) {
  48. sc = 1;
  49. } else if (!(/^-?\d+$/.test(stepCount))) {
  50. sc = Math.round(stepCount + 0.5);
  51. }
  52. return await counterModel.findOneAndUpdate({_id: moduleName}, { $inc: { sequence_value: sc } }, {'new':true});//findOneAndUpdate
  53. };
  54. counterDAO.prototype.getCurrentID = function(moduleName, callback) {
  55. if (callback) {
  56. counterModel.findOne({_id: moduleName}).exec()
  57. .then(function(result, err) {
  58. if (callback) {
  59. callback(result, err);
  60. }
  61. }
  62. );
  63. return null;
  64. } else {
  65. var rst = counterModel.findOne({_id: moduleName}).exec() ;
  66. return rst;
  67. }
  68. }
  69. module.exports = {
  70. counterDAO: new counterDAO(),
  71. moduleName: COUNTER_MODULE_NAME
  72. };