counter.js 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  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. }
  24. /*const PROJECT_COUNTER = 'projects', USER_COUNTER = 'users', BILL_COUNTER = 'bills', RATION_COUNTER = 'rations',
  25. REPORT_COUNTER = 'rptTemplates', FEE_COUNTER = 'fees'*/
  26. var counterDAO = function(){};
  27. /*
  28. * callback = function (err, result) {
  29. * result.value.sequence_value Ϊ�޸ĺ��id
  30. * }
  31. */
  32. counterDAO.prototype.getIDAfterCount = function(moduleName, stepCount, callback) {
  33. var sc = stepCount;
  34. if (isNaN(stepCount) || (stepCount < 0)) {
  35. sc = 1;
  36. } else if (!(/^-?\d+$/.test(stepCount))) {
  37. sc = Math.round(stepCount + 0.5);
  38. }
  39. counterModel.findOneAndUpdate({_id: moduleName}, { $inc: { sequence_value: sc } }, {'new':true}, callback);//findOneAndUpdate
  40. }
  41. counterDAO.prototype.getIDAfterCountSync = async function (moduleName, stepCount) {
  42. let sc = stepCount;
  43. if (isNaN(stepCount) || (stepCount < 0)) {
  44. sc = 1;
  45. } else if (!(/^-?\d+$/.test(stepCount))) {
  46. sc = Math.round(stepCount + 0.5);
  47. }
  48. return await counterModel.findOneAndUpdate({_id: moduleName}, { $inc: { sequence_value: sc } }, {'new':true});//findOneAndUpdate
  49. };
  50. counterDAO.prototype.getCurrentID = function(moduleName, callback) {
  51. if (callback) {
  52. counterModel.findOne({_id: moduleName}).exec()
  53. .then(function(result, err) {
  54. if (callback) {
  55. callback(result, err);
  56. }
  57. }
  58. );
  59. return null;
  60. } else {
  61. var rst = counterModel.findOne({_id: moduleName}).exec() ;
  62. return rst;
  63. }
  64. }
  65. module.exports = {
  66. counterDAO: new counterDAO(),
  67. moduleName: COUNTER_MODULE_NAME
  68. };