tree_service.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819
  1. /*
  2. * @description:层级文件夹数据操作相关
  3. * @Author: CP
  4. * @Date: 2020-09-11 14:43:58
  5. * @FilePath: \construction_management\services\tree_service.go
  6. */
  7. package services
  8. import (
  9. "encoding/json"
  10. "errors"
  11. "fmt"
  12. "log"
  13. "strconv"
  14. "time"
  15. "github.com/kataras/iris/v12"
  16. "go.mod/comm"
  17. "go.mod/conf"
  18. "go.mod/dao"
  19. "go.mod/datasource"
  20. "go.mod/models"
  21. "go.mod/web/utils"
  22. "go.mod/web/viewmodels"
  23. )
  24. //定义项目Service接口
  25. type TreeService interface {
  26. //ValidManager(code string, account string, password string) error
  27. ValidRule(ctx iris.Context) (viewmodels.Tree, error)
  28. ValidRuleBidsectionType(ctx iris.Context) (viewmodels.Permission, error)
  29. Create(data viewmodels.Tree) error
  30. GetAllProject(projectId int) *viewmodels.Tree
  31. GetAllContract(projectId int, account *models.CmProjectAccount, bidsectionType int) *viewmodels.FolderContract
  32. GetAllContractByJL(projectId int) *viewmodels.FolderContract
  33. Rename(treevm viewmodels.Tree, projectId int) error
  34. GetFolderAndBid(id int, projectId int) ([]models.CmTree, error)
  35. DeleteFolderAndBid(id int, projectId int) error
  36. Move(id int, moveId int, projectId int) error
  37. }
  38. //返回service操作类
  39. type treeService struct {
  40. dao *dao.TreeDao
  41. permissionAccountDao *dao.PermissionAccountDao
  42. }
  43. //创建项目service
  44. func NewTreeService() TreeService {
  45. return &treeService{
  46. dao: dao.NewTreeDao(datasource.InstanceDbMaster()),
  47. permissionAccountDao: dao.NewPermissionAccountDao(datasource.InstanceDbMaster()),
  48. }
  49. }
  50. // 文件夹规则验证
  51. func (s *treeService) ValidRule(ctx iris.Context) (viewmodels.Tree, error) {
  52. folderVaild := viewmodels.Tree{}
  53. err := ctx.ReadJSON(&folderVaild)
  54. if err != nil {
  55. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  56. return folderVaild, err
  57. }
  58. err = folderVaild.Validate()
  59. if err != nil {
  60. log.Println("文件夹验证, error=", err)
  61. return folderVaild, err
  62. }
  63. return folderVaild, nil
  64. }
  65. // 文件夹规则验证
  66. func (s *treeService) ValidRuleBidsectionType(ctx iris.Context) (viewmodels.Permission, error) {
  67. folderVaild := viewmodels.Permission{}
  68. err := ctx.ReadForm(&folderVaild)
  69. if err != nil {
  70. log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
  71. return folderVaild, err
  72. }
  73. err = folderVaild.ValidateType()
  74. if err != nil {
  75. log.Println("请求标段类型验证, error=", err)
  76. return folderVaild, err
  77. }
  78. return folderVaild, nil
  79. }
  80. // 获得项目下 相关文件夹-整个树结构
  81. func (s *treeService) GetAllProject(projectId int) *viewmodels.Tree {
  82. datalist := s.dao.GetAllTree(projectId)
  83. folderlist := make([]*viewmodels.Tree, 0)
  84. // 生成根
  85. folder := &viewmodels.Tree{}
  86. id, _ := comm.AesEncrypt(strconv.Itoa(0), conf.SignSecret)
  87. parentId, _ := comm.AesEncrypt(strconv.Itoa(-1), conf.SignSecret)
  88. folder.Id = id
  89. folder.Name = "根目录"
  90. folder.Isfolder = 1
  91. folder.ParentId = parentId
  92. folder.Children = make([]*viewmodels.Tree, 0)
  93. folderlist = append(folderlist, folder)
  94. // 加入数据
  95. for _, data := range datalist {
  96. folder := &viewmodels.Tree{}
  97. //folder.Id = comm.Encrypt([]byte(conf.SignSecret), []byte(strconv.Itoa(data.Id)))
  98. id, _ := comm.AesEncrypt(strconv.Itoa(data.Id), conf.SignSecret)
  99. parentId, _ := comm.AesEncrypt(strconv.Itoa(data.ParentId), conf.SignSecret)
  100. projectId, _ := comm.AesEncrypt(strconv.Itoa(data.ProjectId), conf.SignSecret)
  101. serial, _ := comm.AesEncrypt(strconv.Itoa(data.Serial), conf.SignSecret)
  102. bidsectionId, _ := comm.AesEncrypt(strconv.Itoa(data.BidsectionId), conf.SignSecret)
  103. folder.Id = id
  104. folder.Name = data.Name
  105. folder.ParentId = parentId
  106. folder.ProjectId = projectId
  107. folder.BidsectionId = bidsectionId
  108. folder.Depth = data.Depth + 1
  109. folder.Serial = serial
  110. folder.Accounts = data.Accounts
  111. folder.Isfolder = data.Isfolder
  112. folder.IsEnd = false
  113. //folder.Leaf = true
  114. folder.HasFolder = false
  115. folder.IsBid = false
  116. folder.CreateTime = data.CreateTime.Format(conf.SysTimeform)
  117. folderlist = append(folderlist, folder)
  118. }
  119. //fmt.Println(folderlist)
  120. //var data []*viewmodels.Tree
  121. // data := make([]*viewmodels.Tree, 0)
  122. // for i, _ := range folderlist {
  123. // var a *viewmodels.Tree
  124. // a = &folderlist[i]
  125. // data = append(data, a)
  126. // }
  127. node := folderlist[0] //父节点
  128. maketree(folderlist, node) //调用生成tree
  129. //transformjson(node) //转化为json
  130. return node
  131. }
  132. // 获得合同管理的目录
  133. func (s *treeService) GetAllContract(projectId int, account *models.CmProjectAccount, bidsectionType int) *viewmodels.FolderContract {
  134. datalist := s.dao.GetAllTree(projectId)
  135. folderlist := make([]viewmodels.FolderContract, 0)
  136. // 2.获得该账号的权限
  137. permissionData := s.permissionAccountDao.GetProjectIdAccountId(projectId, account.Id)
  138. // 生成根
  139. folder := viewmodels.FolderContract{}
  140. id, _ := comm.AesEncrypt(strconv.Itoa(0), conf.SignSecret)
  141. parentId, _ := comm.AesEncrypt(strconv.Itoa(-1), conf.SignSecret)
  142. folder.Id = id
  143. folder.Name = "root"
  144. folder.ParentId = parentId
  145. folder.Isfolder = 1
  146. // folder.Children = make([]*viewmodels.FolderContract, 0)
  147. folderlist = append(folderlist, folder)
  148. // 加入数据
  149. for _, data := range datalist {
  150. flag := true
  151. // 标段过滤开始
  152. if data.BidsectionId != 0 && account.IsAdmin != 1 {
  153. flag = false
  154. }
  155. // 过滤没有权限访问的标段-管理员不需要过滤
  156. if data.BidsectionId != 0 && account.IsAdmin != 1 {
  157. permission := map[string]int{}
  158. for _, item := range permissionData {
  159. if data.BidsectionId == item.BidsectionId {
  160. // 区别合同,安全,质量
  161. if bidsectionType == 1 {
  162. json.Unmarshal([]byte(item.SafePermission), &permission)
  163. } else if bidsectionType == 2 {
  164. json.Unmarshal([]byte(item.QualityPermission), &permission)
  165. } else if bidsectionType == 0 {
  166. json.Unmarshal([]byte(item.ContractPermission), &permission)
  167. // fmt.Println(permission)
  168. // if permission["access"] == 1 {
  169. // flag = true
  170. // break
  171. // }
  172. } else {
  173. return nil
  174. }
  175. if permission["access"] == 1 {
  176. flag = true
  177. break
  178. }
  179. }
  180. }
  181. }
  182. // 标段过滤结束
  183. if flag {
  184. folder := viewmodels.FolderContract{}
  185. id, _ := comm.AesEncrypt(strconv.Itoa(data.Id), conf.SignSecret)
  186. parentId, _ := comm.AesEncrypt(strconv.Itoa(data.ParentId), conf.SignSecret)
  187. projectId, _ := comm.AesEncrypt(strconv.Itoa(data.ProjectId), conf.SignSecret)
  188. bidsectionId, _ := comm.AesEncrypt(strconv.Itoa(data.BidsectionId), conf.SignSecret)
  189. folder.Id = id
  190. folder.Name = data.Name
  191. folder.ParentId = parentId
  192. folder.ProjectId = projectId
  193. folder.BidsectionId = bidsectionId
  194. // 合同数据
  195. folder.Contracts = data.Contracts
  196. contractsIncome, _ := strconv.ParseFloat(data.ContractsIncome, 64)
  197. contractsReturned, _ := strconv.ParseFloat(data.ContractsReturned, 64)
  198. folder.ContractsIncome = utils.PriceToStringFormat(contractsIncome)
  199. folder.ContractsReturned = utils.PriceToStringFormat(contractsReturned)
  200. // 汇款进度
  201. ContractsIncome, err := strconv.ParseFloat(data.ContractsIncome, 64)
  202. if err != nil {
  203. ContractsIncome = 0
  204. }
  205. ContractsReturned, err := strconv.ParseFloat(data.ContractsReturned, 64)
  206. if err != nil || ContractsReturned == 0 {
  207. ContractsReturned = 0
  208. folder.ContractsIncomeProgress = "0"
  209. } else {
  210. folder.ContractsIncomeProgress = fmt.Sprintf("%.2f", (ContractsReturned/ContractsIncome)*100)
  211. }
  212. contractsPay, _ := strconv.ParseFloat(data.ContractsPay, 64)
  213. contractsPaid, _ := strconv.ParseFloat(data.ContractsPaid, 64)
  214. folder.ContractsPay = utils.PriceToStringFormat(contractsPay)
  215. folder.ContractsPaid = utils.PriceToStringFormat(contractsPaid)
  216. // 支付进度
  217. ContractsPay, err := strconv.ParseFloat(data.ContractsPay, 64)
  218. if err != nil {
  219. ContractsPay = 0
  220. }
  221. ContractsPaid, err := strconv.ParseFloat(data.ContractsPaid, 64)
  222. if err != nil || ContractsPaid == 0 {
  223. ContractsPaid = 0
  224. folder.ContractsPayProgress = "0"
  225. } else {
  226. folder.ContractsPayProgress = fmt.Sprintf("%.2f", (ContractsPaid/ContractsPay)*100)
  227. }
  228. // 安全
  229. folder.SafeTotal = data.SafeTotal
  230. folder.SafeRectification = data.SafeRectification
  231. folder.SafeRectificationFinish = data.SafeRectificationFinish
  232. folder.SafeRectificationIn = data.SafeRectificationIn
  233. folder.SafeApproval = data.SafeApproval
  234. folder.SafeClose = data.SafeClose
  235. // 质量
  236. folder.QualityTotal = data.QualityTotal
  237. folder.QualityApproval = data.QualityApproval
  238. folder.QualityRectification = data.QualityRectification
  239. folder.QualityRectificationFinish = data.QualityRectificationFinish
  240. folder.QualityRectificationIn = data.QualityRectificationIn
  241. folder.QualityClose = data.QualityClose
  242. folder.Isfolder = data.Isfolder
  243. folder.IsEnd = false
  244. folder.HasFolder = false
  245. folder.IsBid = false
  246. folderlist = append(folderlist, folder)
  247. }
  248. }
  249. var data []*viewmodels.FolderContract
  250. data = make([]*viewmodels.FolderContract, 0)
  251. for i, _ := range folderlist {
  252. var a *viewmodels.FolderContract
  253. a = &folderlist[i]
  254. data = append(data, a)
  255. }
  256. node := &folderlist[0] //父节点
  257. comm.MakeFolderContract(data, node)
  258. // 求和
  259. // dataLenght := len(data) + 4
  260. s.filterFolder(node)
  261. // 删除空目录
  262. ChildrenShow := make([]*viewmodels.FolderContract, 0)
  263. for _, item := range node.Children {
  264. if item.IsHidden == 0 {
  265. ChildrenShow = append(ChildrenShow, item)
  266. }
  267. }
  268. node.Children = ChildrenShow
  269. // node.Children = s.copyShowFolder(node.Children)
  270. // fmt.Println(node)
  271. return node
  272. }
  273. // 复制出需要展示的目录
  274. // func (s *treeService) copyShowFolder(children []*viewmodels.FolderContract) []*viewmodels.FolderContract {
  275. // ChildrenShow := make([]*viewmodels.FolderContract, 0)
  276. // for _, item := range children {
  277. // if item.IsHidden == 0 {
  278. // ChildrenShow = append(ChildrenShow, item)
  279. // }
  280. // }
  281. // return ChildrenShow
  282. // }
  283. func (s *treeService) filterFolder(node *viewmodels.FolderContract) {
  284. if node == nil {
  285. return
  286. }
  287. // 目录下是否为空目录 1空目录
  288. isEmptyFolder := 1
  289. length := len(node.Children)
  290. // 如果length为0叶子节点---为目录才需要遍历
  291. if length == 0 {
  292. // 1.如果是目录-已经为叶子目录不需要遍历
  293. // 2.是标段 目录不为空
  294. if node.Isfolder == 0 {
  295. isEmptyFolder = 0
  296. }
  297. } else {
  298. // 1.遍历目录下
  299. for i := 0; i < length; i++ {
  300. // 2.该节点下子对象 从0-length 依次进入
  301. s.filterFolder(node.Children[i])
  302. // 后序-父节点逻辑 处理完0 ... -length
  303. // 2-1 是标段 设置目录不为空
  304. if node.Children[i].Isfolder == 0 {
  305. // 有标段
  306. isEmptyFolder = 0
  307. } else {
  308. // 2-2 是目录 ----后序遍历,最后一层下的孩子们是否能删除 走完三个点后返回到父亲
  309. // 2-2-1 可展示目录 集合
  310. ChildrenShow := make([]*viewmodels.FolderContract, 0)
  311. // 2-2-2 目录下内容
  312. if len(node.Children[i].Children) != 0 {
  313. // 2-2-3 内容中是否有需要显示的目录
  314. for _, item := range node.Children[i].Children {
  315. // 2-2-4 有 设置为需要显示并拷贝显示内容
  316. if item.IsHidden == 0 {
  317. ChildrenShow = append(ChildrenShow, item)
  318. node.Children[i].IsHidden = 0
  319. }
  320. }
  321. }
  322. // 2-2-5 覆盖父亲节点的显示内容
  323. node.Children[i].Children = ChildrenShow
  324. // s.copyShowFolder(node.Children[i].Children)
  325. // fmt.Println("==========filterFolder执行完成")
  326. // fmt.Println(node.Children[i].Name)
  327. // fmt.Println(fmt.Sprintf("删除:%d", node.Children[i].IsHidden))
  328. // fmt.Println(node.Children[i].Children)
  329. }
  330. }
  331. }
  332. // fmt.Println("=========================进入")
  333. // fmt.Println(node.Name)
  334. // fmt.Println(fmt.Sprintf("空目录:%d", isEmptyFolder))
  335. // 后序的第一位叶子节点 如果目录为空,设置不显示
  336. if isEmptyFolder == 1 {
  337. node.IsHidden = 1
  338. }
  339. }
  340. // 新增一个文件夹
  341. func (s *treeService) Create(data viewmodels.Tree) error {
  342. // 类型校验
  343. folder := models.CmTree{}
  344. folder.Name = data.Name
  345. // 项目ID
  346. ProjectId, err := strconv.Atoi(data.ProjectId)
  347. if err != nil {
  348. return err
  349. }
  350. // 获得该目录ID
  351. Id, err := comm.AesDecrypt(data.Id, conf.SignSecret)
  352. if err != nil {
  353. return err
  354. }
  355. IdInt, err1 := strconv.Atoi(Id)
  356. if err1 != nil {
  357. return err1
  358. }
  359. // 该目录中是否有标段
  360. bidlist := s.dao.GetBidsection(IdInt)
  361. if len(bidlist) > 0 {
  362. return errors.New("该文件夹已存在标段,不能在新增目录")
  363. }
  364. // 获得该深度的文件夹最大序号
  365. serial := 0
  366. depth := data.Depth + 1
  367. if data.Depth == -1 {
  368. // 1-创建顶级目录--获得最大序号
  369. datalist := s.dao.GetAllDepth(depth, ProjectId)
  370. maxIndex := len(datalist)
  371. if maxIndex != 0 {
  372. serial = datalist[maxIndex-1].Serial + 1
  373. }
  374. } else {
  375. // 1-创建次目录-获得次级目录最大序号
  376. treeNode := s.dao.Get(IdInt, ProjectId)
  377. // 去除前端depth作用-在这里获得下级目录的深度
  378. depth = treeNode.Depth + 1
  379. if treeNode.Id == 0 {
  380. return errors.New("上级目录不正确")
  381. }
  382. attribution := fmt.Sprintf("%s%d-", treeNode.Attribution, treeNode.Serial)
  383. datalist := s.dao.GetALLDepthByAttribution(depth, ProjectId, attribution)
  384. maxIndex := len(datalist)
  385. if maxIndex != 0 {
  386. serial = datalist[maxIndex-1].Serial + 1
  387. }
  388. // 设置归属
  389. folder.Attribution = attribution //treeNode.Attribution + strconv.Itoa(treeNode.Serial) + "-"
  390. folder.ParentId = IdInt
  391. }
  392. folder.ProjectId = ProjectId
  393. folder.Serial = serial
  394. folder.Depth = depth
  395. folder.Isfolder = 1
  396. folder.CreateTime = time.Now()
  397. folder.UpdateTime = time.Now()
  398. folder.ContractsIncome = "0"
  399. folder.ContractsPaid = "0"
  400. folder.ContractsPay = "0"
  401. folder.ContractsReturned = "0"
  402. err = s.dao.Create(&folder)
  403. if err != nil {
  404. log.Println("添加目录出错, error=", err)
  405. return errors.New("添加目录出错")
  406. }
  407. return nil
  408. }
  409. // 重命名
  410. func (s *treeService) Rename(data viewmodels.Tree, projectId int) error {
  411. // 获得该目录ID
  412. Id, err := comm.AesDecrypt(data.Id, conf.SignSecret)
  413. if err != nil {
  414. return err
  415. }
  416. IdInt, err1 := strconv.Atoi(Id)
  417. if err1 != nil {
  418. return err1
  419. }
  420. // 获得树信息
  421. treeData := s.dao.Get(IdInt, projectId)
  422. if treeData.Id == 0 {
  423. return errors.New("修改的目录不合法")
  424. }
  425. treeNode := models.CmTree{}
  426. treeNode.Id = IdInt
  427. treeNode.Name = data.Name
  428. if treeData.Isfolder == 1 {
  429. // 目录
  430. err = s.dao.Update(&treeNode, []string{"Name"})
  431. } else {
  432. // 标段
  433. treeNode.BidsectionId = treeData.BidsectionId
  434. err = s.dao.Rename(&treeNode, []string{"Name"})
  435. }
  436. if err != nil {
  437. return err
  438. }
  439. return nil
  440. }
  441. // 获得该目录下所有的目录和标段
  442. func (s *treeService) GetFolderAndBid(id int, projectId int) ([]models.CmTree, error) {
  443. // 获得该目录ID
  444. // id, err := comm.AesDecrypt(id, conf.SignSecret)
  445. // if err != nil {
  446. // return nil, errors.New("目录解析错误")
  447. // }
  448. // IdInt, err := strconv.Atoi(id)
  449. // if err != nil {
  450. // return nil, errors.New("目录解析错误")
  451. // }
  452. // 获得该目录下所有数据
  453. folder := s.dao.Get(id, projectId)
  454. if folder.Id == 0 {
  455. return nil, errors.New("目录不合法")
  456. }
  457. attribution := fmt.Sprintf("%s%d-", folder.Attribution, folder.Serial)
  458. // 获得目录归属
  459. dataList := s.dao.GetFolderAndBid(folder.ProjectId, attribution)
  460. return dataList, nil
  461. }
  462. // 删除该目录和所属的目录和标段
  463. func (s *treeService) DeleteFolderAndBid(id int, projectId int) error {
  464. // 获得该目录ID
  465. // id, err := comm.AesDecrypt(id, conf.SignSecret)
  466. // if err != nil {
  467. // return errors.New("目录解析错误")
  468. // }
  469. // IdInt, err := strconv.Atoi(id)
  470. // if err != nil {
  471. // return errors.New("目录解析错误")
  472. // }
  473. // 获得该目录下 归属关系
  474. folder := s.dao.Get(id, projectId)
  475. if folder.Id == 0 {
  476. return errors.New("目录不合法")
  477. }
  478. attribution := fmt.Sprintf("%s%d-", folder.Attribution, folder.Serial)
  479. // --逻辑删除 目录和数据
  480. err := s.dao.DeleteFolderAndBid(folder.Id, folder.ProjectId, attribution)
  481. if err != nil {
  482. return err
  483. }
  484. // TODO -加入堆栈 物理删除数据
  485. return nil
  486. }
  487. // 移动目录
  488. func (s *treeService) Move(id int, moveId int, projectId int) error {
  489. // 获得项目目录
  490. treeNode := s.dao.Get(id, projectId)
  491. if treeNode.Id == 0 {
  492. return errors.New("目录或标段解析错误")
  493. }
  494. targetFolder := &models.CmTree{}
  495. // id为0是跟目录
  496. if moveId == 0 {
  497. // id
  498. targetFolder.Id = 0
  499. // 1.获得最大序号
  500. // seriallist := s.dao.GetALLDepthByAttribution(0, projectId, "")
  501. // maxIndex := len(seriallist)
  502. // serial := 0
  503. // if maxIndex != 0 {
  504. // serial = seriallist[maxIndex-1].Serial + 1
  505. // }
  506. // targetFolder.Serial = serial
  507. targetFolder.Serial = 0
  508. // 2.归属
  509. targetFolder.Attribution = ""
  510. // 3.深度depth-根目录为-1
  511. targetFolder.Depth = -1
  512. // 4.为目录
  513. targetFolder.Isfolder = 1
  514. // 5.项目ID
  515. targetFolder.ProjectId = projectId
  516. // fmt.Println(targetFolder)
  517. // if targetFolder.Id == 0 {
  518. // return errors.New("目录解析错误-测试")
  519. // }
  520. } else {
  521. targetFolder = s.dao.Get(moveId, projectId)
  522. if targetFolder.Id == 0 {
  523. return errors.New("目录解析错误")
  524. }
  525. }
  526. // 目录必须是目录
  527. if targetFolder.Isfolder == 0 {
  528. return errors.New("目标不是文件夹")
  529. }
  530. // 目录的目标不能是自己
  531. if id == moveId {
  532. return errors.New("目标目录是原目录-不能移动")
  533. }
  534. // 移动的是目录
  535. if treeNode.Isfolder == 1 {
  536. // 1.目录下不能有标段
  537. // 被移动的目录存在标段
  538. moveBidList := s.dao.GetBidsection(moveId)
  539. if len(moveBidList) > 0 {
  540. return errors.New("该目录中存在标段,不能移动")
  541. }
  542. } else {
  543. // 标段
  544. // 1.移动的目录是否为叶子节点
  545. moveBidList := s.dao.GetChildFolder(moveId)
  546. if len(moveBidList) > 0 {
  547. return errors.New("该目录中存在其他目录,不能移动")
  548. }
  549. }
  550. err := s.dao.Move(treeNode, targetFolder)
  551. if err != nil {
  552. return err
  553. }
  554. return nil
  555. }
  556. // 获得合同管理的目录
  557. func (s *treeService) GetAllContractByJL(projectId int) *viewmodels.FolderContract {
  558. datalist := s.dao.GetAllTree(projectId)
  559. folderlist := make([]viewmodels.FolderContract, 0)
  560. // 生成根
  561. folder := viewmodels.FolderContract{}
  562. id, _ := comm.AesEncrypt(strconv.Itoa(0), conf.SignSecret)
  563. parentId, _ := comm.AesEncrypt(strconv.Itoa(-1), conf.SignSecret)
  564. folder.Id = id
  565. folder.Name = "root"
  566. folder.ParentId = parentId
  567. folder.Isfolder = 1
  568. // folder.Children = make([]*viewmodels.FolderContract, 0)
  569. folderlist = append(folderlist, folder)
  570. // 加入数据
  571. for _, data := range datalist {
  572. folder := viewmodels.FolderContract{}
  573. id, _ := comm.AesEncrypt(strconv.Itoa(data.Id), conf.SignSecret)
  574. parentId, _ := comm.AesEncrypt(strconv.Itoa(data.ParentId), conf.SignSecret)
  575. projectId, _ := comm.AesEncrypt(strconv.Itoa(data.ProjectId), conf.SignSecret)
  576. bidsectionId, _ := comm.AesEncrypt(strconv.Itoa(data.BidsectionId), conf.SignSecret)
  577. folder.Id = id
  578. folder.Name = data.Name
  579. folder.ParentId = parentId
  580. folder.ProjectId = projectId
  581. folder.BidsectionId = bidsectionId
  582. // 合同数据
  583. folder.Contracts = data.Contracts
  584. contractsIncome, _ := strconv.ParseFloat(data.ContractsIncome, 64)
  585. contractsReturned, _ := strconv.ParseFloat(data.ContractsReturned, 64)
  586. folder.ContractsIncome = utils.PriceToStringFormat(contractsIncome)
  587. folder.ContractsReturned = utils.PriceToStringFormat(contractsReturned)
  588. // 汇款进度
  589. ContractsIncome, err := strconv.ParseFloat(data.ContractsIncome, 64)
  590. if err != nil {
  591. ContractsIncome = 0
  592. }
  593. ContractsReturned, err := strconv.ParseFloat(data.ContractsReturned, 64)
  594. if err != nil || ContractsReturned == 0 {
  595. ContractsReturned = 0
  596. folder.ContractsIncomeProgress = "0"
  597. } else {
  598. folder.ContractsIncomeProgress = fmt.Sprintf("%.2f", (ContractsReturned/ContractsIncome)*100)
  599. }
  600. contractsPay, _ := strconv.ParseFloat(data.ContractsPay, 64)
  601. contractsPaid, _ := strconv.ParseFloat(data.ContractsPaid, 64)
  602. folder.ContractsPay = utils.PriceToStringFormat(contractsPay)
  603. folder.ContractsPaid = utils.PriceToStringFormat(contractsPaid)
  604. // 支付进度
  605. ContractsPay, err := strconv.ParseFloat(data.ContractsPay, 64)
  606. if err != nil {
  607. ContractsPay = 0
  608. }
  609. ContractsPaid, err := strconv.ParseFloat(data.ContractsPaid, 64)
  610. if err != nil || ContractsPaid == 0 {
  611. ContractsPaid = 0
  612. folder.ContractsPayProgress = "0"
  613. } else {
  614. folder.ContractsPayProgress = fmt.Sprintf("%.2f", (ContractsPaid/ContractsPay)*100)
  615. }
  616. // 安全
  617. folder.SafeTotal = data.SafeTotal
  618. folder.SafeRectification = data.SafeRectification
  619. folder.SafeRectificationFinish = data.SafeRectificationFinish
  620. folder.SafeRectificationIn = data.SafeRectificationIn
  621. folder.SafeApproval = data.SafeApproval
  622. folder.SafeClose = data.SafeClose
  623. // 质量
  624. folder.QualityTotal = data.QualityTotal
  625. folder.QualityApproval = data.QualityApproval
  626. folder.QualityRectification = data.QualityRectification
  627. folder.QualityRectificationFinish = data.QualityRectificationFinish
  628. folder.QualityRectificationIn = data.QualityRectificationIn
  629. folder.QualityClose = data.QualityClose
  630. folder.Isfolder = data.Isfolder
  631. folder.IsEnd = false
  632. folder.HasFolder = false
  633. folder.IsBid = false
  634. folderlist = append(folderlist, folder)
  635. }
  636. var data []*viewmodels.FolderContract
  637. data = make([]*viewmodels.FolderContract, 0)
  638. for i, _ := range folderlist {
  639. var a *viewmodels.FolderContract
  640. a = &folderlist[i]
  641. data = append(data, a)
  642. }
  643. node := &folderlist[0] //父节点
  644. comm.MakeFolderContract(data, node)
  645. // 求和
  646. // dataLenght := len(data) + 4
  647. s.filterFolder(node)
  648. // 删除空目录
  649. ChildrenShow := make([]*viewmodels.FolderContract, 0)
  650. for _, item := range node.Children {
  651. if item.IsHidden == 0 {
  652. ChildrenShow = append(ChildrenShow, item)
  653. }
  654. }
  655. node.Children = ChildrenShow
  656. // node.Children = s.copyShowFolder(node.Children)
  657. // fmt.Println(node)
  658. return node
  659. }
  660. // 创建一颗树...interface{}
  661. func maketree(Data []*viewmodels.Tree, node *viewmodels.Tree) { //参数为父节点,添加父节点的子节点指针切片
  662. childs, _ := havechild(Data, node) //判断节点是否有子节点并返回
  663. if childs != nil {
  664. // fmt.Printf("\n")
  665. // fmt.Println(*node)
  666. // fmt.Println("子节点:")
  667. // for _, v := range childs {
  668. // fmt.Println(*v)
  669. // } //打印
  670. // 子节点总数
  671. Total := len(childs)
  672. node.ChildsTotal = Total
  673. // 标记最后一个元素
  674. end := Total - 1
  675. childs[end].IsEnd = true
  676. // 往父节点添加子节点
  677. node.Children = append(node.Children, childs[0:]...) //添加子节点
  678. for _, v := range childs { //查询子节点的子节点,并添加到子节点
  679. _, has := havechild(Data, v)
  680. if has {
  681. // 递归添加节点
  682. maketree(Data, v)
  683. }
  684. // else {
  685. // // 叶子节点
  686. // node.Leaf = false
  687. // }
  688. // 目录下是否包含标段
  689. if v.Isfolder == 0 {
  690. node.IsBid = true
  691. } else {
  692. node.HasFolder = true
  693. }
  694. }
  695. }
  696. }
  697. // 是否有子树
  698. func havechild(Data []*viewmodels.Tree, node *viewmodels.Tree) (child []*viewmodels.Tree, yes bool) {
  699. for _, v := range Data {
  700. if v.ParentId == node.Id {
  701. child = append(child, v)
  702. }
  703. }
  704. if child != nil {
  705. yes = true
  706. }
  707. return
  708. }
  709. func transformjson(Data *viewmodels.Tree) { //转为json
  710. Jsondata, _ := json.Marshal(Data)
  711. fmt.Println(string(Jsondata))
  712. }
  713. //now := comm.NowUnix()
  714. //comm.FormatFromUnixTime(now)
  715. // ParentIdInt := 0
  716. // if data.ParentId != "" {
  717. // ParentId, err := comm.AesDecrypt(data.ParentId, conf.SignSecret)
  718. // if err != nil {
  719. // return err
  720. // }
  721. // ParentIdInt, err = strconv.Atoi(ParentId)
  722. // if err != nil {
  723. // ParentIdInt = 0
  724. // }
  725. // }