cust_customer.go 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669
  1. package cust
  2. import (
  3. "bytes"
  4. "context"
  5. "math"
  6. "strconv"
  7. "dashoo.cn/opms_libary/myerrors"
  8. "github.com/360EntSecGroup-Skylar/excelize"
  9. "github.com/gogf/gf/errors/gerror"
  10. "github.com/gogf/gf/frame/g"
  11. "github.com/gogf/gf/os/gtime"
  12. "github.com/gogf/gf/text/gstr"
  13. "github.com/gogf/gf/util/gconv"
  14. "dashoo.cn/micro/app/dao/cust"
  15. platdao "dashoo.cn/micro/app/dao/plat"
  16. model "dashoo.cn/micro/app/model/cust"
  17. "dashoo.cn/micro/app/service"
  18. )
  19. const TIME_LAYOUT = "2006-01-02 15:04:05"
  20. type CustomerService struct {
  21. *service.ContextService
  22. Dao *cust.CustCustomerDao
  23. BelongDao *cust.CustCustomerBelongDao
  24. DynamicsDao *cust.CustCustomerDynamicsDao
  25. ContactDao *cust.CustCustomerContactDao
  26. FollowDao *platdao.PlatFollowupDao
  27. BelongServer *CustomerbelongService
  28. ContanctServer *CustomercontactService
  29. }
  30. var isPublic, noPublic = "10", "20" // 公海,非公海
  31. var isTransfer int8 = 1 //转移
  32. var isAllocation int8 = 2 //分配 Allocation
  33. var OperaTion, AllocaTion, Receive, Merge = "20", "10", "30", "40" // 10分配20转移30领取40合并
  34. type OpnType struct {
  35. OperaTion string
  36. }
  37. func NewCustomerService(ctx context.Context) (svc *CustomerService, err error) {
  38. svc = new(CustomerService)
  39. if svc.ContextService, err = svc.Init(ctx); err != nil {
  40. return nil, err
  41. }
  42. svc.Dao = cust.NewCustCustomerDao(svc.Tenant)
  43. svc.BelongDao = cust.NewCustCustomerBelongDao(svc.Tenant)
  44. svc.DynamicsDao = cust.NewCustCustomerDynamicsDao(svc.Tenant)
  45. svc.ContactDao = cust.NewCustCustomerContactDao(svc.Tenant)
  46. svc.FollowDao = platdao.NewPlatFollowupDao(svc.Tenant)
  47. svc.BelongServer, _ = NewCustomerBelongService(ctx)
  48. svc.ContanctServer, _ = NewCustomerContactService(ctx)
  49. return svc, nil
  50. }
  51. //创建客户
  52. func (c *CustomerService) Create(req *model.CustomerAddSeq) (insertId int64, err error) {
  53. cusTomer := new(model.CustCustomer)
  54. count, err := c.Dao.Where(g.Map{"cust_name": req.CustName}).Count()
  55. if err != nil {
  56. err = myerrors.New("Sql执行异常", err)
  57. return
  58. }
  59. if count > 0 {
  60. err = myerrors.NewMsgError(nil, "该客户信息已存在,不可重复添加")
  61. return
  62. }
  63. if err = gconv.Struct(req, cusTomer); err != nil {
  64. return
  65. }
  66. service.SetCreatedInfo(cusTomer, c.GetCxtUserId(), c.GetCxtUserName())
  67. cusTomer.CustCode = "CT" + strconv.Itoa(int(gtime.Timestamp()))
  68. cusTomer.CustStatus = "10"
  69. Roles := c.GetCxtUserRoles()
  70. roles := false
  71. for _, v := range Roles {
  72. if v == "Sales" { // 销售角色
  73. roles = true
  74. }
  75. }
  76. // 销售角色
  77. if roles {
  78. cusTomer.IsPublic = noPublic
  79. cusTomer.SalesId = c.GetCxtUserId()
  80. cusTomer.SalesName = c.GetCxtUserName()
  81. insertId, err = c.Dao.InsertAndGetId(cusTomer)
  82. if err != nil {
  83. return
  84. }
  85. c.CreateContact(int(insertId))
  86. } else {
  87. cusTomer.IsPublic = isPublic
  88. insertId, err = c.Dao.InsertAndGetId(cusTomer)
  89. if err != nil {
  90. err = gerror.New("创建失败")
  91. return
  92. }
  93. }
  94. return
  95. }
  96. func (c *CustomerService) CreateContact(cust_id int) (err error) {
  97. belongData := new(model.CustCustomerBelong)
  98. service.SetCreatedInfo(belongData, c.GetCxtUserId(), c.GetCxtUserName())
  99. belongData.CustId = cust_id
  100. belongData.SaleName = c.GetCxtUserName()
  101. belongData.OpnType = AllocaTion
  102. belongData.OpnPeople = c.GetCxtUserName()
  103. belongData.OpnDatetime = gtime.Now()
  104. _, err = c.BelongDao.InsertAndGetId(belongData)
  105. if err != nil {
  106. return
  107. }
  108. return
  109. }
  110. //删除客户
  111. func (c *CustomerService) DeleteById(id int) error {
  112. Model := c.Dao
  113. ContactModel := c.ContactDao //联系人
  114. BelongModel := c.BelongDao
  115. regionDetail := new(model.CustCustomer)
  116. err := Model.Where(c.Dao.Columns.Id, id).Scan(&regionDetail)
  117. if err != nil {
  118. g.Log().Error(err)
  119. err = gerror.New("没有要删除的数据")
  120. return err
  121. }
  122. //删除客户表
  123. _, err = Model.Where(c.Dao.Columns.Id, id).Delete()
  124. if err == nil {
  125. _, err = ContactModel.Where(c.ContactDao.Columns.CustId, id).Delete()
  126. if err == nil {
  127. _, err = BelongModel.Where(c.BelongDao.Columns.CustId, id).Delete()
  128. if err != nil {
  129. err = gerror.New("删除数据失败")
  130. return err
  131. }
  132. } else {
  133. err = gerror.New("删除数据失败")
  134. return err
  135. }
  136. } else {
  137. g.Log().Error(err)
  138. err = gerror.New("删除数据失败")
  139. return err
  140. }
  141. return nil
  142. }
  143. //修改客户
  144. func (c *CustomerService) UpdateById(req *model.UpdateCustomer) (err error) {
  145. db := c.Dao.M
  146. count, err := db.Where(c.Dao.Columns.Id, req.Id).Count()
  147. if err != nil {
  148. err = gerror.New("修改用户信息失败")
  149. return
  150. }
  151. if count == 0 {
  152. err = gerror.New("用户信息不存在")
  153. return
  154. }
  155. countNums, err := db.Where(c.Dao.Columns.CustName+" = ", req.CustName).Where(c.Dao.Columns.Id+" not in (?)", req.Id).Count()
  156. if err != nil {
  157. err = gerror.New("修改用户信息失败")
  158. return
  159. }
  160. if countNums > 0 {
  161. err = gerror.New("客户名称已存在")
  162. return
  163. }
  164. CustomertData := new(model.CustomerAddSeq)
  165. if err = gconv.Struct(req, CustomertData); err != nil {
  166. return
  167. }
  168. service.SetUpdatedInfo(CustomertData, c.GetCxtUserId(), c.GetCxtUserName())
  169. _, err = db.FieldsEx(c.Dao.Columns.CreatedTime, c.Dao.Columns.CreatedBy, c.Dao.Columns.CreatedName, c.Dao.Columns.Id, c.Dao.Columns.CustCode, c.Dao.Columns.SalesName, c.Dao.Columns.SalesId).
  170. WherePri(c.Dao.Columns.Id, req.Id).Update(CustomertData)
  171. if err != nil {
  172. err = gerror.New("修改用户信息失败")
  173. return
  174. }
  175. return
  176. }
  177. //客户列表列表
  178. func (c *CustomerService) GetList(req *model.CustCustomerSearchReq) (total int, customerList []*model.CustList, err error) {
  179. Model := c.Dao.M
  180. if req.TargetType == "" {
  181. if !req.IsPublic {
  182. Model = Model.Where(c.Dao.Columns.SalesId, c.CxtUser.Id).Where(c.Dao.Columns.IsPublic, noPublic)
  183. } else {
  184. Model = Model.Where(c.Dao.Columns.IsPublic, isPublic)
  185. }
  186. }
  187. //客户名称
  188. if req.CustName != "" {
  189. Model = Model.Where(c.Dao.Columns.CustName+" like ?", "%"+req.CustName+"%")
  190. }
  191. //客户编码
  192. if req.CustCode != "" {
  193. Model = Model.Where(c.Dao.Columns.CustCode+" like ?", "%"+req.CustCode+"%")
  194. }
  195. //客户行业
  196. if req.CustIndustry != "" {
  197. Model = Model.Where(c.Dao.Columns.CustIndustry+" like ?", "%"+req.CustIndustry+"%")
  198. }
  199. //客户级别
  200. if req.CustLevel != "" {
  201. g.Log().Info("level", req.CustLevel)
  202. Model = Model.Where(c.Dao.Columns.CustLevel, req.CustLevel)
  203. }
  204. //
  205. if req.FollowUpDate != "" {
  206. Model = Model.Where(c.Dao.Columns.FollowUpDate+" like ? ", req.FollowUpDate+"%")
  207. }
  208. total, err = Model.Count()
  209. if err != nil {
  210. g.Log().Error(err)
  211. err = gerror.New("获取总行数失败")
  212. return
  213. }
  214. if req.PageNum == 0 {
  215. req.PageNum = 1
  216. }
  217. err = Model.Page(req.GetPage()).Order("id desc").Scan(&customerList)
  218. for _, v := range customerList {
  219. times := gconv.String(v.CreatedTime)
  220. v.FollowUpDate = gstr.SubStr(v.FollowUpDate, 0, 16)
  221. v.CreatedTime = gstr.SubStr(times, 0, 16)
  222. }
  223. return
  224. }
  225. //移入公海
  226. func (c *CustomerService) MoveToPubic(ids []int64) error {
  227. Model := c.Dao.M
  228. list, err := Model.Fields(c.Dao.Columns.CreatedTime).Where(c.Dao.Columns.Id+" in (?) ", ids).All()
  229. if err != nil || list.Len() <= 0 {
  230. g.Log().Error(err)
  231. err = gerror.New("没有要移除的数据")
  232. return err
  233. }
  234. _, err = Model.Data(g.Map{
  235. "is_public": isPublic,
  236. "sales_id": 0,
  237. "sales_name": "",
  238. "dept_id": 0,
  239. "dept_name": "",
  240. "create_time": gtime.Now(),
  241. "updated_by": c.GetCxtUserId(),
  242. "updated_name": c.GetCxtUserName(),
  243. "updated_time": gtime.Now(),
  244. }).Where(c.ContactDao.Columns.Id+" in (?)", ids).Update()
  245. if err != nil {
  246. g.Log().Error(err)
  247. err = gerror.New("移入公海失败")
  248. return err
  249. }
  250. return nil
  251. }
  252. //分配客户
  253. func (c *CustomerService) DistriCustomer(req *model.DistriCustomer) error {
  254. custModel := c.Dao
  255. rep, err := custModel.Where(cust.CustCustomer.Columns.Id+" in (?) ", req.Ids).Where(cust.CustCustomer.Columns.IsPublic, isPublic).All()
  256. if err != nil {
  257. err = myerrors.New("DistriCustomer Sql执行异常", err)
  258. return err
  259. }
  260. var custMap = make(map[int]string)
  261. for _, v := range rep {
  262. custMap[v.Id] = v.SalesName
  263. }
  264. g.Log().Info("fadfadsfaf", custMap)
  265. for _, v := range req.Ids {
  266. g.Log().Info("req.Ids", v)
  267. if custMap[gconv.Int(v)] != "" {
  268. err = gerror.New("ID" + gconv.String(v) + "已被认领")
  269. return err
  270. }
  271. }
  272. err = c.updateCustomer(req.Ids, req.SalesId, req.SalesName)
  273. if err != nil {
  274. err = gerror.New("可配客户失败")
  275. return err
  276. }
  277. if req.Receive != "" {
  278. req.Receive = Receive
  279. } else {
  280. req.Receive = AllocaTion
  281. }
  282. err = c.Createbelong(rep, req)
  283. g.Log().Info("errr", req)
  284. if err != nil {
  285. err = gerror.New("分配客户失败")
  286. return err
  287. }
  288. return nil
  289. }
  290. //客户详情
  291. func (c *CustomerService) GetEntityById(ids []int64) (entityInfo []*model.CustList, err error) {
  292. Model := c.Dao //
  293. err = Model.Where(cust.CustCustomer.Columns.Id+" in (?)", ids).Scan(&entityInfo)
  294. if err != nil {
  295. return nil, myerrors.New("获取用户数据失败", err)
  296. }
  297. for _, v := range entityInfo {
  298. v.FollowUpDate = gstr.SubStr(v.FollowUpDate, 0, 16)
  299. v.CreatedTime = gstr.SubStr(v.CreatedTime, 0, 16)
  300. }
  301. return
  302. }
  303. //客户摘要
  304. func (c *CustomerService) CustAbstract(id int64) (followInfo *model.Follow, err error) {
  305. custModel := c.Dao //
  306. Model := c.FollowDao
  307. count, err := Model.Where(c.FollowDao.Columns.CustId, id).Count()
  308. if err != nil {
  309. err = myerrors.New("获取用户数据失败", err)
  310. return
  311. }
  312. followInfo = new(model.Follow)
  313. followInfo.FollowCount = count
  314. //
  315. followUpdate, err := custModel.Fields(c.Dao.Columns.FollowUpDate).FindOne(id)
  316. if err != nil {
  317. err = myerrors.New("CustAbstract Sql执行错误", err)
  318. return
  319. }
  320. if followUpdate.FollowUpDate == nil {
  321. followInfo.NotFollowDay = 0
  322. return
  323. }
  324. g.Log().Info("fasa1211", followUpdate)
  325. //g.Log().Info("find", followUpdate)
  326. //if err != nil {
  327. // g.Log().Error(err)
  328. // myerrors.New("获取用户数据失败", err)
  329. // return
  330. //}
  331. //if followUpdate.FollowUpDate != nil {
  332. // followInfo.NotFollowDay = 0
  333. // return
  334. //}
  335. upDate := gtime.New(followUpdate.FollowUpDate)
  336. now := gtime.Now()
  337. //
  338. poor := now.Sub(upDate)
  339. hours := float64(poor.Hours() / 24)
  340. if hours < 0 {
  341. followInfo.NotFollowDay = 0
  342. } else {
  343. followInfo.NotFollowDay = int(math.Floor(hours))
  344. }
  345. return
  346. }
  347. //转移客户
  348. func (c *CustomerService) UpdateBytransfer(req *model.DistriCustomer) (err error) {
  349. custModel := c.Dao
  350. data, err := custModel.Fields("sales_id,sales_name,id").Where(cust.CustCustomer.Columns.Id+" in (?)", req.Ids).All()
  351. if err != nil {
  352. err = myerrors.New("Sql执行异常", err)
  353. return err
  354. }
  355. if len(data) == 0 {
  356. err = myerrors.New("该数据不存在", err)
  357. return
  358. }
  359. err = c.updateCustomer(req.Ids, req.SalesId, req.SalesName)
  360. if err != nil {
  361. return
  362. }
  363. err = c.Createbelong(data, req)
  364. g.Log().Info("err", err)
  365. if err != nil {
  366. return
  367. }
  368. return
  369. }
  370. //变更客户所属关系
  371. func (c *CustomerService) updateCustomer(ids []int64, salesId int64, salesName string) error {
  372. custModel := c.Dao.M
  373. _, err := custModel.Data(g.Map{
  374. "sales_id": salesId,
  375. "is_public": noPublic,
  376. "sales_name": salesName,
  377. "updated_by": c.GetCxtUserId(),
  378. "updated_name": c.GetCxtUserName(),
  379. "updated_time": gtime.Now(),
  380. }).Where(cust.CustCustomer.Columns.Id+" in (?)", ids).Update()
  381. if err != nil {
  382. err = myerrors.New("updateCustomer 变更执行失败", err)
  383. return err
  384. }
  385. return nil
  386. }
  387. //客户操作日志
  388. func (c *CustomerService) OperationLog(ctx context.Context, ids []int64, req *model.AddCustomerDynameicsReq) (err error) {
  389. g.Log("fdasfsa", gconv.String(ctx))
  390. cusDynameics := new(model.CustCustomerDynamics)
  391. if err = gconv.Struct(req, cusDynameics); err != nil {
  392. g.Log().Info("error", err)
  393. return
  394. }
  395. Model := c.DynamicsDao.M
  396. g.Log().Info("IDS", ids)
  397. maps := []map[string]interface{}{}
  398. for _, v := range ids {
  399. contact := map[string]interface{}{}
  400. contact["cust_id"] = v
  401. contact["opn_people_id"] = c.GetCxtUserId()
  402. contact["opn_people"] = c.GetCxtUserName()
  403. contact["opn_date"] = req.OpnDate
  404. contact["opn_type"] = req.OpnType
  405. contact["remark"] = ""
  406. contact["created_by"] = c.GetCxtUserId()
  407. contact["created_name"] = c.GetCxtUserName()
  408. contact["created_by"] = c.GetCxtUserId()
  409. contact["created_time"] = gtime.Now()
  410. contact["opn_content"] = req.OpnContent
  411. maps = append(maps, contact)
  412. }
  413. lastId, err := Model.InsertAndGetId(maps)
  414. if err != nil {
  415. err = myerrors.New("OperationLog Sql执行失败", err)
  416. return
  417. }
  418. if lastId == 0 {
  419. err = myerrors.New("客户操作日志失败", err)
  420. return
  421. }
  422. return
  423. }
  424. //客户动态
  425. func (c *CustomerService) DynamicsList(req *model.CustomerDynameicsReq) (total int, result []interface{}, err error) {
  426. Model := c.DynamicsDao.M
  427. total, err = Model.Fields().Count()
  428. if err != nil {
  429. g.Log().Error(err)
  430. err = gerror.New("获取总行数失败")
  431. return
  432. }
  433. //if req.PageNum == 0 {
  434. // req.PageNum = 1
  435. //}
  436. dynamics := []*model.CustomerDynameicsRep{}
  437. err = Model.Where("cust_id = ", req.CustId).Order("created_time desc").Scan(&dynamics)
  438. dynamicsList := make(map[string][]*model.CustomerDynameicsRep)
  439. for _, v := range dynamics {
  440. gt1 := gtime.New(v.OpnDate)
  441. opnDate := gt1.Format("Y-m-d")
  442. dynamicsList[opnDate] = append(dynamicsList[opnDate], &model.CustomerDynameicsRep{
  443. OpnPeople: v.OpnPeople,
  444. OpnDate: v.OpnDate,
  445. OpnType: v.OpnType,
  446. OpnContent: v.OpnContent,
  447. })
  448. }
  449. //g.Log().Info("DynamicsList----", dynamicsList)
  450. //result = append(result, dynamicsList)
  451. //g.Log().Info("Dy---", result)
  452. return
  453. }
  454. //合并客户
  455. func (c *CustomerService) Mergecustomer(req *model.MergecustomerRep) (err error) {
  456. Model := c.Dao.M
  457. //ContactModel := c.ContactDao.M
  458. BelongDao := c.BelongDao
  459. //当前目标客户是否存在
  460. FindOne, err := Model.Where(c.Dao.Columns.DeletedTime+" is null").Where(c.Dao.Columns.Id, req.Id).FindOne()
  461. if err != nil || FindOne.IsEmpty() {
  462. err = gerror.New("该数据不存在")
  463. return
  464. }
  465. ContactModel := c.ContactDao
  466. _, err = ContactModel.Where(c.ContactDao.Columns.CustId+" in (?)", req.ChooseId).WhereOr(c.ContactDao.Columns.CustId, req.Id).Delete()
  467. if err != nil {
  468. err = gerror.New("合并联系人失败")
  469. return
  470. }
  471. CustomertData := new(model.CustomerAddSeq)
  472. if err = gconv.Struct(req, CustomertData); err != nil {
  473. return
  474. }
  475. service.SetUpdatedInfo(CustomertData, c.GetCxtUserId(), c.GetCxtUserName())
  476. _, err = Model.FieldsEx(c.Dao.Columns.CreatedTime, c.Dao.Columns.CreatedBy,
  477. c.Dao.Columns.CreatedName, c.Dao.Columns.Id,
  478. c.Dao.Columns.CustCode).WherePri(c.Dao.Columns.Id, req.Id).Update(CustomertData)
  479. if err != nil {
  480. err = gerror.New("合并失败")
  481. return
  482. }
  483. //删除被合并的客户信息
  484. Model.Where(c.Dao.Columns.Id+" in (?)", req.ChooseId).Delete()
  485. //删除 所选客户销售联系人
  486. _, err = BelongDao.Where(c.BelongDao.Columns.CustId+" in (?)", req.ChooseId).WhereOr(c.BelongDao.Columns.CustId, req.Id).Delete()
  487. if err != nil {
  488. err = gerror.New("合并失败")
  489. return
  490. }
  491. //插入一条合并成功的归属记录
  492. belongService := c.BelongServer //
  493. req.AddCustomerBelong.CustId = req.Id
  494. req.AddCustomerBelong.OpnType = Merge
  495. req.AddCustomerBelong.OpnPeople = c.GetCxtUserName()
  496. req.AddCustomerBelong.OpnDatetime = gtime.Now()
  497. req.AddCustomerBelong.SaleName = req.SalesName
  498. err = belongService.Create(req.AddCustomerBelong)
  499. if err != nil {
  500. err = gerror.New("合并失败")
  501. return
  502. }
  503. return
  504. }
  505. //联系人(合并)
  506. func (c *CustomerService) Createcontact(id int, Ids []int64, req *model.CustCustomerContactSeq) (err error) {
  507. ContactModel := c.ContactDao.M
  508. _, err = ContactModel.Where(c.ContactDao.Columns.CustId+" in (?)", Ids).WhereOr(c.ContactDao.Columns.CustId, id).Delete()
  509. if err != nil {
  510. err = gerror.New("合并联系人失败")
  511. return
  512. }
  513. contactService := c.ContanctServer //new(CustomercontactService)
  514. err = contactService.Create(req)
  515. if err != nil {
  516. err = gerror.New("合并联系人失败")
  517. return
  518. }
  519. return
  520. }
  521. //批量插入客户归属记录表//parameter map[string]string
  522. func (c *CustomerService) Createbelong(rep []*model.CustCustomer, req *model.DistriCustomer) (err error) {
  523. belongModel := c.BelongDao
  524. var belongData []*model.CustCustomerBelong
  525. date_time := gtime.Now()
  526. userName := c.GetCxtUserName()
  527. for _, v := range rep {
  528. orig_sale_name := v.SalesName
  529. belong := new(model.CustCustomerBelong) //map[string]interface{}{}
  530. belong.CustId = v.Id
  531. belong.SaleName = req.SalesName
  532. belong.OrigSaleName = orig_sale_name
  533. belong.OpnType = req.Receive
  534. belong.OpnPeople = userName
  535. belong.CreatedName = userName
  536. belong.OpnDatetime = date_time
  537. belong.Remark = req.Remark
  538. belong.CreatedBy = c.GetCxtUserId()
  539. belongData = append(belongData, belong)
  540. }
  541. lastId, err := belongModel.InsertAndGetId(belongData)
  542. g.Log().Info("belong", belongData)
  543. if err != nil {
  544. err = myerrors.New("Createbelong Sql执行异常", err)
  545. return err
  546. }
  547. if lastId == 0 {
  548. err = myerrors.New("Createbelong 插入", err)
  549. return err
  550. }
  551. return
  552. }
  553. //导出数据
  554. func (c *CustomerService) Export(req *model.CustCustomerExport) (content *model.CustExport, err error) {
  555. var con model.CustExport
  556. total, data, err := c.GetList(&req.CustCustomerSearchReq)
  557. if err != nil {
  558. return
  559. }
  560. f := excelize.NewFile()
  561. index := f.NewSheet("Sheet1")
  562. for index, item := range req.Columns {
  563. sheetPosition := service.Div(index+1) + "1"
  564. f.SetCellValue("Sheet1", sheetPosition, item)
  565. }
  566. if total > 0 {
  567. for ck, item := range data {
  568. for index, v := range req.Columns {
  569. // "CustCode":客户编码
  570. if v == "客户编码" {
  571. f.SetCellValue("Sheet1", service.Div(index+1)+strconv.Itoa(ck+2), item.CustCode)
  572. }
  573. if v == "客户名称" {
  574. f.SetCellValue("Sheet1", service.Div(index+1)+strconv.Itoa(ck+2), item.CustName)
  575. }
  576. if v == "助记名" {
  577. f.SetCellValue("Sheet1", service.Div(index+1)+strconv.Itoa(ck+2), item.AbbrName)
  578. }
  579. if v == "助记名" {
  580. f.SetCellValue("Sheet1", service.Div(index+1)+strconv.Itoa(ck+2), item.AbbrName)
  581. }
  582. if v == "所在地区" {
  583. f.SetCellValue("Sheet1", service.Div(index+1)+strconv.Itoa(ck+2), item.CustLocation)
  584. }
  585. if v == "客户行业" {
  586. f.SetCellValue("Sheet1", service.Div(index+1)+strconv.Itoa(ck+2), item.CustIndustry)
  587. }
  588. if v == "客户级别" {
  589. f.SetCellValue("Sheet1", service.Div(index+1)+strconv.Itoa(ck+2), item.CustLevel)
  590. }
  591. if v == "客户状态" {
  592. var CustStatus string
  593. CustStatus = "正常"
  594. if item.CustStatus != "10" {
  595. CustStatus = "异常"
  596. }
  597. f.SetCellValue("Sheet1", service.Div(index+1)+strconv.Itoa(ck+2), CustStatus)
  598. }
  599. if v == "最后跟进时间" {
  600. f.SetCellValue("Sheet1", service.Div(index+1)+strconv.Itoa(ck+2), item.FollowUpDate)
  601. }
  602. if v == "创建时间" {
  603. f.SetCellValue("Sheet1", service.Div(index+1)+strconv.Itoa(ck+2), item.CreatedTime)
  604. }
  605. }
  606. }
  607. }
  608. f.SetActiveSheet(index)
  609. var buffer *bytes.Buffer
  610. buffer, _ = f.WriteToBuffer()
  611. con.Content = buffer.Bytes()
  612. return &con, err
  613. }