sectionTreeModel.js 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146
  1. /**
  2. * Created by Zhong on 2017/12/21.
  3. */
  4. const mongoose = require('mongoose');
  5. const compleRationSectionTreeModel = mongoose.model('complementary_ration_section_tree');
  6. const compleRationModel = mongoose.model('complementary_ration_items');
  7. let counter = require('../../../public/counter/counter');
  8. let stdSectionTreeModel = require ('../../ration_repository/models/ration_section_tree').Model;
  9. const uuidV1 = require('uuid/v1');
  10. const sectionTemplateModel = mongoose.model('complementary_ration_section_templates');
  11. class SectionTreeDao {
  12. //从补充定额章节树模板中拷贝数据到用户的补充定额章节树中
  13. async copyDataFromTemplate(userId, compilationId){
  14. let templateData = await sectionTemplateModel.find({compilationId: compilationId});
  15. if (templateData.length > 0) {
  16. let insertDatas = [],
  17. uuidMapping = {};
  18. //将ID替换成UUID
  19. for (let temData of templateData) {
  20. uuidMapping[temData.ID] = uuidV1();
  21. }
  22. for(let temData of templateData) {
  23. let insertD = {
  24. userId: userId,
  25. compilationId: compilationId,
  26. name: temData.name,
  27. ID: uuidMapping[temData.ID],
  28. ParentID: uuidMapping[temData.ParentID] || -1,
  29. NextSiblingID: uuidMapping[temData.NextSiblingID] || -1,
  30. };
  31. insertDatas.push(insertD);
  32. }
  33. //插入数据
  34. await compleRationSectionTreeModel.insertMany(insertDatas);
  35. }
  36. }
  37. getNewTreeID(callback){
  38. counter.counterDAO.getIDAfterCount(counter.moduleName.rationTree, 1, function (err, result) {
  39. if(err){
  40. callback(err, null);
  41. }
  42. else {
  43. callback(0, result.sequence_value);
  44. }
  45. });
  46. }
  47. async getComplementaryTree (userId, compilationId) {
  48. const treeData = await compleRationSectionTreeModel.find({userId, compilationId}).lean();
  49. return treeData;
  50. }
  51. async getRationTree(userId, compilationId, rationRepId, type, callback) {
  52. //区分要获取的是标准的数据还是补充的数据
  53. const rationLibType = {
  54. complementary: 0,
  55. std: 1
  56. };
  57. try {
  58. let treeData;
  59. if (type === rationLibType.complementary) {
  60. treeData = await compleRationSectionTreeModel.find({userId: userId, compilationId: compilationId, $or: [{deleteInfo: null}, {'deleteInfo.deleted': false}]}).lean();
  61. } else {
  62. treeData = await stdSectionTreeModel.find({rationRepId: rationRepId}).lean();
  63. }
  64. callback(0, treeData);
  65. } catch (err) {
  66. console.log(err);
  67. callback(1, null);
  68. }
  69. }
  70. //获取补充定额拼接章节树
  71. /* async getRationTree(userID, rationRepId, callback){
  72. try{
  73. let stdSectionTree = await stdSectionTreeModel.find({rationRepId: rationRepId, $or: [{isDeleted: null}, {isDeleted: false}]});
  74. let compleSectionTree = await compleRationSectionTreeModel.find({userId: userID, rationRepId: rationRepId, deleteInfo: null});
  75. let dropPids = [], rstCompleSectionTree = [];
  76. //mark std
  77. for(let i = 0, len = stdSectionTree.length; i < len; i++){
  78. stdSectionTree[i]._doc.type = 'std';
  79. }
  80. for(let i = 0, len = compleSectionTree.length; i < len; i++){
  81. //mark complementary
  82. compleSectionTree[i]._doc.type = 'complementary';
  83. if(compleSectionTree[i]['isFirst']){
  84. let updateSection = getUpdateSection(compleSectionTree[i]['ParentID'], stdSectionTree);
  85. if(updateSection) {
  86. updateSection._doc.NextSiblingID = compleSectionTree[i]['ID'];
  87. }
  88. else if(!updateSection && compleSectionTree[i]['ParentID'] !== -1){
  89. dropPids.push(compleSectionTree[i]['ParentID']);
  90. }
  91. }
  92. }
  93. function getUpdateSection(pid, datas){
  94. for(let i = 0, len = datas.length; i < len; i++){
  95. if(datas[i]['ParentID'] === pid && datas[i]['NextSiblingID'] === -1){
  96. return datas[i];
  97. }
  98. }
  99. return null;
  100. }
  101. //返回父节点未被删除的
  102. for(let i = 0, len = compleSectionTree.length; i < len; i++){
  103. if(dropPids.indexOf(compleSectionTree[i]['ParentID']) === -1){
  104. rstCompleSectionTree.push(compleSectionTree[i]);
  105. }
  106. }
  107. callback(0, stdSectionTree.concat(rstCompleSectionTree));
  108. }
  109. catch (err){
  110. callback(err, null);
  111. }
  112. }*/
  113. async updateSection(userID, compilationId, updateData, callback){
  114. try{
  115. for(let i = 0, len = updateData.length; i < len; i++){
  116. let updateObj = updateData[i];
  117. if(updateObj.updateType === 'new'){
  118. updateObj.updateData.userId = userID;
  119. updateObj.updateData.compilationId = compilationId;
  120. await compleRationSectionTreeModel.create(updateObj.updateData);
  121. }
  122. else if(updateObj.updateType === 'update'){
  123. if(!updateObj.updateData.deleteInfo){
  124. await compleRationSectionTreeModel.update({userId: userID, ID: updateObj.updateData.ID}, updateObj.updateData);
  125. } else {
  126. await compleRationSectionTreeModel.remove({userId: userID, ID: updateObj.updateData.ID});
  127. }
  128. }
  129. }
  130. callback(0, 'success');
  131. }
  132. catch(err){
  133. callback(err, null);
  134. }
  135. }
  136. }
  137. export default SectionTreeDao;