eta_business.go 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321
  1. package eta_business
  2. import (
  3. "fmt"
  4. "github.com/beego/beego/v2/client/orm"
  5. "github.com/rdlucklib/rdluck_tools/paging"
  6. "hongze/hz_crm_api/utils"
  7. "strings"
  8. "time"
  9. )
  10. const (
  11. EtaBusinessSigningStatusFirst = iota + 1
  12. EtaBusinessSigningStatusContinue
  13. EtaBusinessSigningStatusTerminate
  14. )
  15. type EtaBusiness struct {
  16. EtaBusinessId int `orm:"column(eta_business_id);pk"`
  17. BusinessName string `description:"商家名称"`
  18. BusinessCode string `description:"商家编码"`
  19. CreditCode string `description:"社会统一信用码"`
  20. RegionType string `description:"所属区域:国内;海外"`
  21. Province string `description:"省份"`
  22. City string `description:"城市"`
  23. Address string `description:"商家地址"`
  24. SellerId int `description:"销售ID"`
  25. SellerName string `description:"销售名称"`
  26. Leader string `description:"决策人"`
  27. IndustryId int `description:"行业ID"`
  28. IndustryName string `description:"行业名称"`
  29. CapitalScale string `description:"资金规模"`
  30. ResearchTeamSize string `description:"研究团队规模"`
  31. UserMax int `description:"用户上限"`
  32. SigningStatus int `description:"签约状态:1-首次签约;2-续约中;3-已终止"`
  33. Enable int `description:"状态:0-禁用;1-启用"`
  34. ContractId int `description:"当前合约ID"`
  35. SigningTime time.Time `description:"当前合约的签约时间"`
  36. ExpiredTime time.Time `description:"当前合约的到期时间"`
  37. CreateTime time.Time `description:"创建时间"`
  38. ModifyTime time.Time `description:"更新时间"`
  39. }
  40. func (m *EtaBusiness) TableName() string {
  41. return "eta_business"
  42. }
  43. func (m *EtaBusiness) PrimaryId() string {
  44. return EtaBusinessColumns.EtaBusinessId
  45. }
  46. var EtaBusinessColumns = struct {
  47. EtaBusinessId string
  48. BusinessName string
  49. BusinessCode string
  50. CreditCode string
  51. RegionType string
  52. Province string
  53. City string
  54. Address string
  55. SellerId string
  56. SellerName string
  57. Leader string
  58. IndustryId string
  59. IndustryName string
  60. CapitalScale string
  61. ResearchTeamSize string
  62. UserMax string
  63. SigningStatus string
  64. Enable string
  65. ContractId string
  66. SigningTime string
  67. ExpiredTime string
  68. CreateTime string
  69. ModifyTime string
  70. }{
  71. EtaBusinessId: "eta_business_id",
  72. BusinessName: "business_name",
  73. BusinessCode: "business_code",
  74. CreditCode: "credit_code",
  75. RegionType: "region_type",
  76. Province: "province",
  77. City: "city",
  78. Address: "address",
  79. SellerId: "seller_id",
  80. SellerName: "seller_name",
  81. Leader: "leader",
  82. IndustryId: "industry_id",
  83. IndustryName: "industry_name",
  84. CapitalScale: "capital_scale",
  85. ResearchTeamSize: "research_team_size",
  86. UserMax: "user_max",
  87. SigningStatus: "signing_status",
  88. Enable: "enable",
  89. ContractId: "contract_id",
  90. SigningTime: "signing_time",
  91. ExpiredTime: "expired_time",
  92. CreateTime: "create_time",
  93. ModifyTime: "modify_time",
  94. }
  95. func (m *EtaBusiness) Create() (err error) {
  96. o := orm.NewOrm()
  97. id, err := o.Insert(m)
  98. if err != nil {
  99. return
  100. }
  101. m.EtaBusinessId = int(id)
  102. return
  103. }
  104. func (m *EtaBusiness) CreateMulti(items []*EtaBusiness) (err error) {
  105. if len(items) == 0 {
  106. return
  107. }
  108. o := orm.NewOrm()
  109. _, err = o.InsertMulti(len(items), items)
  110. return
  111. }
  112. func (m *EtaBusiness) Update(cols []string) (err error) {
  113. o := orm.NewOrm()
  114. _, err = o.Update(m, cols...)
  115. return
  116. }
  117. func (m *EtaBusiness) Del() (err error) {
  118. o := orm.NewOrm()
  119. sql := fmt.Sprintf(`DELETE FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.PrimaryId())
  120. _, err = o.Raw(sql, m.EtaBusinessId).Exec()
  121. return
  122. }
  123. func (m *EtaBusiness) GetItemById(id int) (item *EtaBusiness, err error) {
  124. o := orm.NewOrm()
  125. sql := fmt.Sprintf(`SELECT * FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.PrimaryId())
  126. err = o.Raw(sql, id).QueryRow(&item)
  127. return
  128. }
  129. func (m *EtaBusiness) GetItemByCondition(condition string, pars []interface{}) (item *EtaBusiness, err error) {
  130. o := orm.NewOrm()
  131. sql := fmt.Sprintf(`SELECT * FROM %s WHERE 1=1 %s LIMIT 1`, m.TableName(), condition)
  132. err = o.Raw(sql, pars).QueryRow(&item)
  133. return
  134. }
  135. func (m *EtaBusiness) GetCountByCondition(condition string, pars []interface{}) (count int, err error) {
  136. o := orm.NewOrm()
  137. sql := fmt.Sprintf(`SELECT COUNT(1) FROM %s WHERE 1=1 %s`, m.TableName(), condition)
  138. err = o.Raw(sql, pars).QueryRow(&count)
  139. return
  140. }
  141. func (m *EtaBusiness) GetItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string) (items []*EtaBusiness, err error) {
  142. o := orm.NewOrm()
  143. fields := strings.Join(fieldArr, ",")
  144. if len(fieldArr) == 0 {
  145. fields = `*`
  146. }
  147. order := `ORDER BY create_time DESC`
  148. if orderRule != "" {
  149. order = ` ORDER BY ` + orderRule
  150. }
  151. sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s`, fields, m.TableName(), condition, order)
  152. _, err = o.Raw(sql, pars).QueryRows(&items)
  153. return
  154. }
  155. func (m *EtaBusiness) GetPageItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string, startSize, pageSize int) (items []*EtaBusiness, err error) {
  156. o := orm.NewOrm()
  157. fields := strings.Join(fieldArr, ",")
  158. if len(fieldArr) == 0 {
  159. fields = `*`
  160. }
  161. order := `ORDER BY create_time DESC`
  162. if orderRule != "" {
  163. order = ` ORDER BY ` + orderRule
  164. }
  165. sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s LIMIT ?,?`, fields, m.TableName(), condition, order)
  166. _, err = o.Raw(sql, pars, startSize, pageSize).QueryRows(&items)
  167. return
  168. }
  169. // CreateEtaBusinessAndContract 新增商家和签约
  170. func CreateEtaBusinessAndContract(businessItem *EtaBusiness, contractItem *EtaBusinessContract) (err error) {
  171. if businessItem == nil || contractItem == nil {
  172. err = fmt.Errorf("item empty")
  173. return
  174. }
  175. o := orm.NewOrm()
  176. tx, err := o.Begin()
  177. if err != nil {
  178. return
  179. }
  180. defer func() {
  181. if err != nil {
  182. _ = tx.Rollback()
  183. return
  184. }
  185. _ = tx.Commit()
  186. }()
  187. // 商家
  188. businessId, e := tx.Insert(businessItem)
  189. if e != nil {
  190. err = fmt.Errorf("business insert err: %s", e.Error())
  191. return
  192. }
  193. businessItem.EtaBusinessId = int(businessId)
  194. // 签约
  195. contractItem.EtaBusinessId = businessItem.EtaBusinessId
  196. contractId, e := tx.Insert(contractItem)
  197. if e != nil {
  198. err = fmt.Errorf("contract insert err: %s", e.Error())
  199. return
  200. }
  201. // 更新商家当前签约ID
  202. businessItem.ContractId = int(contractId)
  203. cols := []string{"ContractId"}
  204. _, err = tx.Update(businessItem, cols...)
  205. return
  206. }
  207. // EtaBusinessAddReq 新增商家请求体
  208. type EtaBusinessAddReq struct {
  209. BusinessName string `description:"商家名称"`
  210. CreditCode string `description:"社会统一信用码"`
  211. RegionType string `description:"所属区域:国内;海外"`
  212. Province string `description:"省份"`
  213. City string `description:"城市"`
  214. SellerId int `description:"销售ID"`
  215. SellerName string `description:"销售名称"`
  216. Leader string `description:"决策人"`
  217. IndustryId int `description:"行业ID"`
  218. IndustryName string `description:"行业名称"`
  219. CapitalScale string `description:"资金规模"`
  220. ResearchTeamSize string `description:"研究团队规模"`
  221. UserMax int `description:"用户上限"`
  222. SigningTime string `description:"签约时间"`
  223. ExpiredTime string `description:"到期时间"`
  224. IsCheck bool `description:"是否只做校验而不实际新增(业务操作上基础信息和签约时间分成两个步骤了)"`
  225. }
  226. // EtaBusinessEditReq 编辑商家请求体
  227. type EtaBusinessEditReq struct {
  228. EtaBusinessId int `description:"商家ID"`
  229. Province string `description:"省份"`
  230. City string `description:"城市"`
  231. Leader string `description:"决策人"`
  232. IndustryId int `description:"行业ID"`
  233. IndustryName string `description:"行业名称"`
  234. CapitalScale string `description:"资金规模"`
  235. ResearchTeamSize string `description:"研究团队规模"`
  236. UserMax int `description:"用户上限"`
  237. }
  238. // EtaBusinessSigningReq 商家签约请求体
  239. type EtaBusinessSigningReq struct {
  240. EtaBusinessId int `description:"商家ID"`
  241. SigningTime string `description:"当前合约的签约时间"`
  242. ExpiredTime string `description:"当前合约的到期时间"`
  243. }
  244. // EtaBusinessEnableReq 禁启用商家请求体
  245. type EtaBusinessEnableReq struct {
  246. EtaBusinessId int `description:"商家ID"`
  247. }
  248. // EtaBusinessMoveSellerReq 移动商家销售请求体
  249. type EtaBusinessMoveSellerReq struct {
  250. EtaBusinessId int `description:"商家ID"`
  251. SellerId int `description:"销售ID"`
  252. SellerName string `description:"销售名称"`
  253. }
  254. // CreateEtaBusinessCode 生成ETA商家编码
  255. func CreateEtaBusinessCode() (code string, err error) {
  256. var num int
  257. o := orm.NewOrm()
  258. sql := `SELECT COUNT(1) AS num FROM eta_business WHERE create_time >= ? `
  259. err = o.Raw(sql, time.Now().Format(utils.FormatDate)).QueryRow(&num)
  260. if err != nil {
  261. return
  262. }
  263. code = "E" + time.Now().Format("20060102") + fmt.Sprintf("%02d", num)
  264. return
  265. }
  266. // EtaBusinessListResp 商家分页列表响应体
  267. type EtaBusinessListResp struct {
  268. List []*EtaBusinessItem `description:"商家列表数据"`
  269. Paging *paging.PagingItem `description:"分页数据"`
  270. }
  271. // EtaBusinessItem ETA商家信息
  272. type EtaBusinessItem struct {
  273. EtaBusinessId int
  274. BusinessName string `description:"商家名称"`
  275. BusinessCode string `description:"商家编码"`
  276. CreditCode string `description:"社会统一信用码"`
  277. RegionType string `description:"所属区域:国内;海外"`
  278. Address string `description:"商家地址"`
  279. SellerId int `description:"销售ID"`
  280. SellerName string `description:"销售名称"`
  281. Leader string `description:"决策人"`
  282. IndustryId int `description:"行业ID"`
  283. IndustryName string `description:"行业名称"`
  284. CapitalScale string `description:"资金规模"`
  285. ResearchTeamSize string `description:"研究团队规模"`
  286. UserMax int `description:"用户上限"`
  287. SigningStatus int `description:"签约状态:1-首次签约;2-续约中;3-已终止"`
  288. Enable int `description:"状态:0-禁用;1-启用"`
  289. SigningTime string `description:"当前合约的签约时间"`
  290. ExpiredTime string `description:"当前合约的到期时间"`
  291. CreateTime string `description:"创建时间"`
  292. ModifyTime string `description:"更新时间"`
  293. }