project_account_dao.go 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145
  1. /*
  2. * @description:项目用户数据库操作
  3. * @Author: CP
  4. * @Date: 2020-08-31 15:57:04
  5. * @FilePath: \construction_management\dao\project_account_dao.go
  6. */
  7. package dao
  8. import (
  9. "errors"
  10. "github.com/go-xorm/xorm"
  11. "go.mod/models"
  12. "go.mod/web/viewmodels"
  13. )
  14. //数据库操作引擎
  15. type ProjectAccountDao struct {
  16. engine *xorm.Engine
  17. }
  18. //获得一个DAO对象
  19. func NewProjectAccountDao(engine *xorm.Engine) *ProjectAccountDao {
  20. return &ProjectAccountDao{
  21. engine: engine,
  22. }
  23. }
  24. //CmProjectAccount 设置的值获得数据-弃用
  25. // func (d *ProjectAccountDao) GetOne(data *models.CmProjectAccount) *models.CmProjectAccount {
  26. // //data := data
  27. // //Get取到值后,会自动赋值到data中
  28. // ok, err := d.engine.
  29. // Where("account = ?", data.Account).
  30. // Get(data)
  31. // if ok && err == nil {
  32. // return data
  33. // } else {
  34. // data.Id = 0
  35. // return data
  36. // }
  37. // }
  38. // 根据账号获得数据
  39. func (d *ProjectAccountDao) GetAccount(account string) *models.CmProjectAccount {
  40. data := &models.CmProjectAccount{}
  41. ok, err := d.engine.
  42. Where("account = ?", account).
  43. Get(data)
  44. if ok && err == nil {
  45. return data
  46. } else {
  47. data.Id = 0
  48. return data
  49. }
  50. }
  51. //id获得数据
  52. func (d *ProjectAccountDao) Get(id int, projectId int) *models.CmProjectAccount {
  53. data := &models.CmProjectAccount{Id: id, ProjectId: projectId}
  54. ok, err := d.engine.Get(data)
  55. if ok && err == nil {
  56. return data
  57. } else {
  58. data.Id = 0
  59. return data
  60. }
  61. }
  62. // 获得项目下的账号信息
  63. func (d *ProjectAccountDao) GetAll(projectId int) []models.CmProjectAccount {
  64. datalist := make([]models.CmProjectAccount, 0)
  65. err := d.engine.
  66. Where("project_id = ?", projectId).
  67. Desc("id").
  68. Find(&datalist)
  69. if err != nil {
  70. return datalist
  71. } else {
  72. return datalist
  73. }
  74. }
  75. func (d *ProjectAccountDao) GetInId(ids string, name string) []models.CmProjectAccount {
  76. datalist := make([]models.CmProjectAccount, 0)
  77. err := d.engine.
  78. Where("id in ( "+ids+" ) ").
  79. And("account like ? or name like ? or company like ? or mobile like ?", name+"%", name+"%", name+"%", name+"%").
  80. Desc("id").
  81. Find(&datalist)
  82. if err != nil {
  83. return datalist
  84. } else {
  85. return datalist
  86. }
  87. }
  88. // 检索 账号姓名单位手机
  89. func (d *ProjectAccountDao) Search(name string, projectId int) []models.CmProjectAccount {
  90. datalist := make([]models.CmProjectAccount, 0)
  91. err := d.engine.
  92. Where("project_id = ? ", projectId).
  93. And("account like ? or name like ? or company like ? or mobile like ?", name+"%", name+"%", name+"%", name+"%").
  94. Desc("id").
  95. Find(&datalist)
  96. if err != nil {
  97. return datalist
  98. } else {
  99. return datalist
  100. }
  101. }
  102. // 新增
  103. func (d *ProjectAccountDao) Add(data *models.CmProjectAccount) error {
  104. _, err := d.engine.Insert(data)
  105. return err
  106. }
  107. //更新
  108. func (d *ProjectAccountDao) Update(data *models.CmProjectAccount, columns []string) error {
  109. //_, err := d.engine.Id(data.Id).MustCols(columns...).Update(data)
  110. _, err := d.engine.Where("id = ? and project_id = ? ", data.Id, data.ProjectId).MustCols(columns...).Update(data)
  111. if err != nil {
  112. return errors.New("更新账号失败")
  113. }
  114. return nil
  115. }
  116. // 物理删除
  117. func (d *ProjectAccountDao) Delete(data *models.CmProjectAccount) error {
  118. _, err := d.engine.Where("id = ? and project_id = ? ", data.Id, data.ProjectId).Delete(data)
  119. if err != nil {
  120. return errors.New("删除到账号失败")
  121. }
  122. return nil
  123. }
  124. // 查找
  125. func (d *ProjectAccountDao) FindById(id int) (viewmodels.ProjectInfo, error) {
  126. projectInfo := viewmodels.ProjectInfo{}
  127. _, err := d.engine.Sql("select p.`name` as project_name, p.`code`, p.`create_time`, a.`mobile`, a.`name` from `cm_project` as p, `cm_project_account` as a where p.`user_id` = a.`id` and p.`id` = ?", id).Get(&projectInfo)
  128. return projectInfo, err
  129. }