sectionTreeModel.js 6.1 KB

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