counter.js 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. /**
  2. * Created by Tony on 2017/3/21.
  3. */
  4. var mongoose = require('mongoose');
  5. var dbm = require("../../config/db/db_manager");
  6. var projectdb = dbm.getCfgConnection("scConstruct");
  7. var Schema = mongoose.Schema;
  8. var counterSchema = new Schema({
  9. _id: String,
  10. sequence_value: Number
  11. });
  12. counterSchema.statics.findAndModify = function (query, sort, doc, options, callback) {
  13. return this.collection.findAndModify(query, sort, doc, options, callback);
  14. };
  15. var counterModel = projectdb.model("counters", counterSchema);
  16. // All counter Must predefine in DB
  17. const COUNTER_MODULE_NAME = {
  18. project: 'projects',
  19. user: 'users',
  20. bills: 'bills',
  21. ration: 'rations',
  22. rationMap: 'rationMaps',
  23. rationTree: 'rationChapterTrees',
  24. report: 'rptTemplates',
  25. fee: 'fees',
  26. template_bills: 'temp_bills'
  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.findAndModify({_id: moduleName}, [], { $inc: { sequence_value: sc } }, {'new':true}, callback);
  44. }
  45. counterDAO.prototype.getCurrentID = function(moduleName, callback) {
  46. if (callback) {
  47. counterModel.findOne({_id: moduleName}).exec()
  48. .then(function(result, err) {
  49. if (callback) {
  50. callback(result, err);
  51. }
  52. }
  53. );
  54. return null;
  55. } else {
  56. var rst = counterModel.findOne({_id: moduleName}).exec() ;
  57. return rst;
  58. }
  59. }
  60. module.exports = {
  61. counterDAO: new counterDAO(),
  62. moduleName: COUNTER_MODULE_NAME
  63. };