123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556 |
- /*
- * @description: 合同项目节相关数据库操作
- * @Author: CP
- * @Date: 2020-11-02 11:37:32
- * @FilePath: \construction_management\dao\tree_contract_dao.go
- */
- package dao
- import (
- "errors"
- "fmt"
- "log"
- "strings"
- "github.com/go-xorm/xorm"
- "go.mod/models"
- )
- //数据库操作引擎
- type TreeContractDao struct {
- engine *xorm.Engine
- }
- //获得一个DAO对象
- func NewTreeContractDao(engine *xorm.Engine) *TreeContractDao {
- return &TreeContractDao{
- engine: engine,
- }
- }
- // 获得本项目的合同项目节
- func (d *TreeContractDao) Get(treeId int, bidsectionId int, projectId int, treeType int) *models.CmTreeContracts {
- data := &models.CmTreeContracts{}
- _, err := d.engine.
- Where("tree_id=? and bidsection_id =? and project_id=? and tree_type=?", treeId, bidsectionId, projectId, treeType).
- Get(data)
- if err != nil {
- data.Id = 0
- return data
- }
- return data
- }
- // 获得项目下的项目节
- func (d *TreeContractDao) GetAll(bidsectionId int, projectId int, treeType int) []models.CmTreeContracts {
- datalist := make([]models.CmTreeContracts, 0)
- err := d.engine.
- Asc("id").
- Where("bidsection_id =? and project_id=? and tree_type=?", bidsectionId, projectId, treeType).
- Limit(5000, 0).
- Find(&datalist)
- if err != nil {
- return datalist
- } else {
- return datalist
- }
- }
- // 获得项目下的项目节不包含合同
- func (d *TreeContractDao) GetAllNotContract(bidsectionId int, projectId int, treeType int) []models.CmTreeContracts {
- datalist := make([]models.CmTreeContracts, 0)
- err := d.engine.
- Asc("id").
- Where("bidsection_id =? and project_id=? and tree_type=? and contract_id=0", bidsectionId, projectId, treeType).
- Limit(5000, 0).
- Find(&datalist)
- if err != nil {
- return datalist
- } else {
- return datalist
- }
- }
- // 获得最新ID
- func (d *TreeContractDao) GetLastId() *models.CmTreeContracts {
- data := &models.CmTreeContracts{}
- _, err := d.engine.
- Desc("id").
- Get(data)
- if err != nil {
- data.Id = 0
- return data
- }
- return data
- }
- // 获得项目节所有合同
- func (d *TreeContractDao) GetContractAll(bidsectionId int, projectId int) []models.CmTreeContracts {
- datalist := make([]models.CmTreeContracts, 0)
- err := d.engine.
- Asc("id").
- Where("bidsection_id =? and project_id=? and contract_id!=0 ", bidsectionId, projectId).
- Find(&datalist)
- if err != nil {
- return datalist
- } else {
- return datalist
- }
- }
- // 获得标段 项目节中已有的合同
- func (d *TreeContractDao) GetContract(bidsectionId int, projectId int, treeType int) []models.CmTreeContracts {
- datalist := make([]models.CmTreeContracts, 0)
- err := d.engine.
- Asc("id").
- Where("bidsection_id =? and project_id=? and contract_id!=0 and tree_type=?", bidsectionId, projectId, treeType).
- Find(&datalist)
- if err != nil {
- return datalist
- } else {
- return datalist
- }
- }
- // 获得节点的孩子
- func (d *TreeContractDao) GetChildren(parentId int, bidsectionId int, projectId int, treeType int) []models.CmTreeContracts {
- datalist := make([]models.CmTreeContracts, 0)
- err := d.engine.
- Asc("serial").
- Where("parent_id=? and bidsection_id=? and project_id=? and tree_type=?", parentId, bidsectionId, projectId, treeType).
- Find(&datalist)
- if err != nil {
- return datalist
- } else {
- return datalist
- }
- }
- //根据序号和深度获得前一个兄弟节点
- func (d *TreeContractDao) GetElderBrother(serial int, depth int, parentId int, bidsectionId int, projectId int, treeType int) []models.CmTreeContracts {
- datalist := make([]models.CmTreeContracts, 0)
- err := d.engine.
- Desc("serial").
- Where("serial < ? and depth = ? and parent_id =? and bidsection_id=? and project_id=? and tree_type=?", serial, depth, parentId, bidsectionId, projectId, treeType).
- Find(&datalist)
- if err != nil {
- return datalist
- } else {
- return datalist
- }
- }
- //根据序号和深度获得后一个兄弟节点
- func (d *TreeContractDao) GetYoungerBrother(serial int, depth int, parentId int, bidsectionId int, projectId int, treeType int) []models.CmTreeContracts {
- datalist := make([]models.CmTreeContracts, 0)
- err := d.engine.
- Asc("serial").
- Where("serial > ? and depth = ? and parent_id =? and bidsection_id=? and project_id=? and tree_type=?", serial, depth, parentId, bidsectionId, projectId, treeType).
- Find(&datalist)
- if err != nil {
- return datalist
- } else {
- return datalist
- }
- }
- // 获得最后一条项目节
- func (d *TreeContractDao) GetLast(projectId int, treeType int) *models.CmTreeContracts {
- data := &models.CmTreeContracts{}
- _, err := d.engine.
- Desc("id").
- Where("project_id=? and tree_type=?", projectId, treeType).
- Get(data)
- if err != nil {
- data.Id = 0
- return data
- }
- return data
- }
- // 获得谋归属下的项目节
- func (d *TreeContractDao) GetAttribution(attribution string, projectId int, bidsectionId int, treeType int) []models.CmTreeContracts {
- datalist := make([]models.CmTreeContracts, 0)
- err := d.engine.
- Asc("serial").
- Where("attribution like ? and project_id=? and bidsection_id=? and tree_type=?", attribution+"%", projectId, bidsectionId, treeType).
- Find(&datalist)
- if err != nil {
- return datalist
- } else {
- return datalist
- }
- }
- // 获得谋归属下的项目节-合同
- func (d *TreeContractDao) GetAttributionContract(section *models.CmTreeContracts, treeType int) []models.CmTreeContracts {
- // attribution := section.Attribution
- attribution := fmt.Sprintf("%s-", section.Code)
- projectId := section.ProjectId
- bidsectionId := section.BidsectionId
- datalist := make([]models.CmTreeContracts, 0)
- err := d.engine.
- Asc("serial").
- Where("(attribution like ? or id =? ) and project_id=? and bidsection_id=? and tree_type=? and contract_id!=0", attribution+"%", section.Id, projectId, bidsectionId, treeType).
- Find(&datalist)
- if err != nil {
- return datalist
- } else {
- return datalist
- }
- }
- // 项目节升降级
- func (d *TreeContractDao) MoveDepth(section *models.CmTreeContracts, elderBrother *models.CmTreeContracts, operation string, bidsectionId int, projectId int, treeType int) error {
- session := d.engine.NewSession()
- defer session.Close()
- err := session.Begin()
- if err != nil {
- return errors.New("操作失败-db")
- }
- // 降级
- if operation == "downDepth" {
- // 1.前一个兄弟节点
- // fmt.Println(elderBrother)
- // 2.把节点移动到兄弟节点的下级,成为兄弟的孩子
- // 2-1 节点的父亲为兄弟的ID
- // 2-2 节点的深度 +1
- // 2-3 节点归属为兄弟归属+序号
- attribution := fmt.Sprintf("%s%d-", elderBrother.Attribution, elderBrother.Serial)
- // 2-4 序号 没有孩子序号为1,有孩子 最大孩子序号+1
- // 2-4-1 获得上一位哥的孩子们
- elderBrotherChildren := d.GetChildren(elderBrother.TreeId, bidsectionId, projectId, treeType)
- serial := 1
- if len(elderBrotherChildren) != 0 {
- serial = elderBrotherChildren[len(elderBrotherChildren)-1].Serial + 1
- }
- // 2-5 项目节编号
- // 原编号
- // section.Code
- // 移动后编号
- moveCode := fmt.Sprintf("%s%d", attribution, serial)
- _, err = session.Exec("UPDATE cm_tree_contracts SET `parent_id` = ?,attribution= ? , serial = ? ,`code` = replace(`code`, '"+section.Code+"', '"+moveCode+"')"+
- ",`depth` =`depth` + ? where id = ? and tree_type=? ", elderBrother.TreeId, attribution, serial, 1, section.Id, treeType)
- if err != nil {
- session.Rollback()
- return errors.New("降级失败")
- }
- // 3.更新节点下的归属
- // 3-1 节点的所有孩子的归属
- // 原节点 孩子的归属
- attributionChildren := fmt.Sprintf("%s%d-", section.Attribution, section.Serial)
- // 降级后的 孩子归属
- moveAttributionChildren := fmt.Sprintf("%s%d-", attribution, serial)
- // 3-2 降级 深度+1
- _, err = session.Exec("UPDATE cm_tree_contracts SET "+
- "`depth` =`depth` + ? where attribution like ? and project_id=? and bidsection_id=? and tree_type=? ", 1, attributionChildren+"%", projectId, bidsectionId, treeType)
- if err != nil {
- session.Rollback()
- return errors.New("降级失败")
- }
- // 3-3--替换 归属和编号
- // "`attribution` = replace(`attribution`, '"+attributionChildren+"', '"+moveAttributionChildren+"') ,`code` = replace(`code`, '"+section.Code+"', '"+moveCode+"'),"
- err = d.replaceContractAttribution(session, attributionChildren, moveAttributionChildren, section.Code, moveCode, projectId, bidsectionId, treeType)
- if err != nil {
- session.Rollback()
- return errors.New("降级失败")
- }
- } else if operation == "upDepth" {
- // 升级
- // 1.父亲节点
- sectionFather := d.Get(section.ParentId, bidsectionId, projectId, treeType)
- if sectionFather.Id == 0 {
- session.Rollback()
- return errors.New("升级-未找到上级项目节")
- }
- // 2.原节点的父亲ID字段升级为爷爷ID
- // 2-1 升级 深度-1
- // 2-2 序号 爷爷的孩子们的 序号+1
- grandpaChildren := d.GetChildren(sectionFather.ParentId, bidsectionId, projectId, treeType)
- serial := 1
- if len(grandpaChildren) != 0 {
- serial = grandpaChildren[len(grandpaChildren)-1].Serial + 1
- }
- // 2-3 归属 原父亲的归属
- // 移动后编号-需替换原编号 节点2-1-1 升级后的父节点归属 2- 加上爷爷最大孩子的序号+1
- moveCode := fmt.Sprintf("%s%d", sectionFather.Attribution, serial)
- // 升级的项目节
- _, err = session.Exec("UPDATE cm_tree_contracts SET `parent_id` = ?,attribution= ? , serial = ? ,`code` = replace(`code`, '"+section.Code+"', '"+moveCode+"')"+
- ",`depth` =`depth` - ? where id = ? and tree_type=? ", sectionFather.ParentId, sectionFather.Attribution, serial, 1, section.Id, treeType)
- if err != nil {
- session.Rollback()
- return errors.New("升级失败")
- }
- // 3.更新节点下的归属,深度
- // 原节点 孩子的归属
- attributionChildren := fmt.Sprintf("%s%d-", section.Attribution, section.Serial)
- // 升级后的 孩子归属
- moveAttributionChildren := fmt.Sprintf("%s%d-", sectionFather.Attribution, serial)
- // 深度 -1
- _, err = session.Exec("UPDATE cm_tree_contracts SET `attribution` = replace(`attribution`, '"+attributionChildren+"', '"+moveAttributionChildren+"') "+
- ",`depth` =`depth` - ? where attribution like ? and project_id=? and bidsection_id=? and tree_type=? ", 1, attributionChildren+"%", projectId, bidsectionId, treeType)
- if err != nil {
- session.Rollback()
- return errors.New("升级失败")
- }
- // 3-1
- err = d.replaceContractAttribution(session, attributionChildren, moveAttributionChildren, section.Code, moveCode, projectId, bidsectionId, treeType)
- if err != nil {
- session.Rollback()
- return errors.New("升级失败")
- }
- } else {
- return errors.New("参数错误")
- }
- err = session.Commit()
- if err != nil {
- session.Rollback()
- return errors.New("操作失败-db")
- }
- return nil
- }
- // 合同项目节上下移动
- func (d *TreeContractDao) MoveSerial(section *models.CmTreeContracts, brother *models.CmTreeContracts, operation string, bidsectionId int, projectId int, treeType int) error {
- session := d.engine.NewSession()
- defer session.Close()
- err := session.Begin()
- if err != nil {
- return errors.New("操作失败-db")
- }
- //1.上下移
- // 1.项目节序号替换为兄弟序号
- _, err = session.Exec("UPDATE cm_tree_contracts SET serial = ? , `code` = replace(`code`, '"+section.Code+"', '"+brother.Code+"') where id = ? and tree_type=? ", brother.Serial, section.Id, treeType)
- if err != nil {
- session.Rollback()
- return errors.New("移动失败")
- }
- // 兄弟序号替换为项目节序号
- _, err = session.Exec("UPDATE cm_tree_contracts SET serial = ? , `code` = replace(`code`, '"+brother.Code+"', '"+section.Code+"') where id = ? and tree_type=? ", section.Serial, brother.Id, treeType)
- if err != nil {
- session.Rollback()
- return errors.New("移动失败")
- }
- //2.项目节孩子们 归属设置
- // 原节点 孩子的归属
- attributionChildren := fmt.Sprintf("%s%d-", section.Attribution, section.Serial)
- // 移动后的 孩子归属和编号
- moveAttributionChildren := fmt.Sprintf("%s%d-", brother.Attribution, brother.Serial)
- err = d.replaceContractAttribution(session, attributionChildren, moveAttributionChildren, section.Code, brother.Code, projectId, bidsectionId, treeType)
- if err != nil {
- session.Rollback()
- return errors.New("移动失败")
- }
- // _, err = session.Exec("UPDATE cm_tree_contracts SET `attribution` = replace(`attribution`, '"+attributionChildren+"', '"+moveAttributionChildren+"') "+
- // "`code` = replace(`code`, '"+attributionChildren+"', '"+moveAttributionChildren+"') where attribution like ? and project_id=? and bidsection_id=? ", attributionChildren+"%", projectId, bidsectionId)
- // if err != nil {
- // session.Rollback()
- // return errors.New("移动失败")
- // }
- // 3.兄弟节点孩子们 归属设置
- // 兄弟节点 孩子的归属
- attributionChildren = fmt.Sprintf("%s%d-", brother.Attribution, brother.Serial)
- // 移动后的 孩子归属
- moveAttributionChildren = fmt.Sprintf("%s%d-", section.Attribution, section.Serial)
- err = d.replaceContractAttribution(session, attributionChildren, moveAttributionChildren, brother.Code, section.Code, projectId, bidsectionId, treeType)
- if err != nil {
- session.Rollback()
- return errors.New("移动失败")
- }
- err = session.Commit()
- if err != nil {
- session.Rollback()
- return errors.New("操作失败-db")
- }
- return nil
- }
- // 修改项目节序号
- func (d *TreeContractDao) UpdateSerial(section *models.CmTreeContracts, serial int, treeType int) error {
- session := d.engine.NewSession()
- defer session.Close()
- err := session.Begin()
- if err != nil {
- return errors.New("操作失败-db")
- }
- // 1.更新项目节序号和项目节编号
- moveCode := fmt.Sprintf("%s%d", section.Attribution, serial)
- _, err = session.Exec("UPDATE cm_tree_contracts SET serial = ? , `code` = ? where id = ? and tree_type=? ", serial, moveCode, section.Id, treeType)
- if err != nil {
- session.Rollback()
- log.Println("合同项目节序号更新 error=", err)
- return errors.New("更新序号失败")
- }
- // 2.更新项目节子孙们的序号和归属
- // 2-1归属 项目节 孩子归属
- attributionChildren := fmt.Sprintf("%s%d-", section.Attribution, section.Serial)
- // 2-2 序号更新后的 孩子归属
- moveAttributionChildren := fmt.Sprintf("%s%d-", section.Attribution, serial)
- // 2-3 项目节 孩子的编号
- code := fmt.Sprintf("%s%d", section.Attribution, section.Serial)
- err = d.replaceContractAttribution(session, attributionChildren, moveAttributionChildren, code, moveCode, section.ProjectId, section.BidsectionId, treeType)
- if err != nil {
- session.Rollback()
- log.Println("合同项目节序号更新 error=", err)
- return errors.New("更新序号失败")
- }
- err = session.Commit()
- if err != nil {
- session.Rollback()
- return errors.New("操作失败-db")
- }
- return nil
- }
- // 插入多条数据
- func (d *TreeContractDao) CreateAll(data []*models.CmTreeContracts) error {
- session := d.engine.NewSession()
- defer session.Close()
- err := session.Begin()
- if err != nil {
- return errors.New("新增失败-db")
- }
- // 新增
- for _, item := range data {
- _, err := session.Insert(item)
- if err != nil {
- log.Println(" error=", err)
- session.Rollback()
- return errors.New("新增失败")
- }
- }
- err = session.Commit()
- if err != nil {
- session.Rollback()
- return errors.New("新增失败-db")
- }
- return nil
- }
- // 新增项目节
- func (d *TreeContractDao) Create(data *models.CmTreeContracts) (*models.CmTreeContracts, error) {
- session := d.engine.NewSession()
- defer session.Close()
- err := session.Begin()
- if err != nil {
- return nil, errors.New("新增失败-db")
- }
- _, err = session.Insert(data)
- if err != nil {
- log.Println(" error=", err)
- return nil, errors.New("新增失败")
- }
- // 插入成功后,ID自动赋值到data.Id里
- // fmt.Println("=======================================")
- // fmt.Println(data)
- // 更新合同节树ID
- // data.TreeId = data.Id
- // _, err = session.Id(data.Id).Update(data)
- // if err != nil {
- // log.Println(" error=", err)
- // return errors.New("新增失败")
- // }
- err = session.Commit()
- if err != nil {
- session.Rollback()
- return nil, errors.New("新增失败-db")
- }
- return data, nil
- }
- // 保存项目节
- func (d *TreeContractDao) Save(section *models.CmTreeContracts, columns []string) error {
- _, err := d.engine.Id(section.Id).MustCols(columns...).Update(section)
- if err != nil {
- return errors.New("保存失败")
- }
- return nil
- }
- // 删除项目节
- func (d *TreeContractDao) Delete(section *models.CmTreeContracts) error {
- session := d.engine.NewSession()
- defer session.Close()
- err := session.Begin()
- if err != nil {
- return errors.New("删除失败-db")
- }
- // 1. 删除项目节
- _, err = session.Where("id = ? ", section.Id).Delete(section)
- if err != nil {
- session.Rollback()
- return errors.New("删除失败")
- }
- // 孩子们的归属
- attribution := fmt.Sprintf("%s%d-", section.Attribution, section.Serial)
- // 2. 删除项目节孩子们
- _, err = session.Exec("DELETE FROM `cm_tree_contracts` WHERE attribution like ? and project_id=? and bidsection_id=? and tree_type=? ", attribution+"%", section.ProjectId, section.BidsectionId, section.TreeType)
- if err != nil {
- session.Rollback()
- return errors.New("删除失败")
- }
- err = session.Commit()
- if err != nil {
- session.Rollback()
- return errors.New("删除失败-db")
- }
- return nil
- }
- //替换项目节归属
- func (d *TreeContractDao) replaceContractAttribution(session *xorm.Session, attributionChildren string, moveAttributionChildren string, code string, moveCode string, projectId int, bidsectionId int, treeType int) error {
- // 1.获得需要替换的数据
- sectionData := d.GetAttribution(attributionChildren, projectId, bidsectionId, treeType)
- if len(sectionData) == 0 {
- return nil
- }
- attributionSql := " attribution = case id "
- codeSql := " code = case id "
- idList := make([]int, 0)
- for _, item := range sectionData {
- // section := &models.CmTreeContracts{}
- // section.Id = item.Id
- // 替换归属
- attributionSql += fmt.Sprintf("when %d then '%s' ", item.Id, strings.Replace(item.Attribution, attributionChildren, moveAttributionChildren, 1))
- //section.Attribution = strings.Replace(item.Attribution, attributionChildren, moveAttributionChildren, 1)
- // 替换编号
- codeSql += fmt.Sprintf("when %d then '%s' ", item.Id, strings.Replace(item.Code, code, moveCode, 1))
- // section.Code = strings.Replace(item.Code, code, moveCode, 1)
- idList = append(idList, item.Id)
- }
- attributionSql += " end, "
- codeSql += " end "
- id := strings.Replace(strings.Trim(fmt.Sprint(idList), "[]"), " ", ",", -1)
- sql := "update cm_tree_contracts set " + attributionSql + codeSql + " WHERE id IN (" + id + ")"
- _, err := session.Exec(sql)
- if err != nil {
- log.Println("替换项目节归属, error=", err)
- return err
- }
- return nil
- }
|