supplier.go 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810
  1. package service
  2. import (
  3. "context"
  4. "fmt"
  5. "github.com/issue9/conv"
  6. log "github.com/sirupsen/logrus"
  7. "strconv"
  8. "strings"
  9. "time"
  10. "youngee_b_api/db"
  11. "youngee_b_api/model/gorm_model"
  12. "youngee_b_api/model/http_model"
  13. )
  14. var Supplier *supplier
  15. type supplier struct {
  16. }
  17. // CreateSupplier 创建younggee平台用户账号与服务商用户信息
  18. func (*supplier) CreateSupplier(ctx context.Context, phone string) (*http_model.RegisterData, error) {
  19. // 1. 创建YG平台用户
  20. userInfo := gorm_model.YounggeeUser{
  21. Phone: phone,
  22. User: "1003",
  23. Username: phone,
  24. Password: "1003",
  25. RealName: "",
  26. Role: "6",
  27. Email: "",
  28. LastLoginTime: time.Now().UTC().Local(),
  29. }
  30. userId, createUserErr := db.CreateUser(ctx, userInfo)
  31. if createUserErr != nil {
  32. log.Infof("[CreateSupplierUser] fail,err:%+v", createUserErr)
  33. return nil, createUserErr
  34. } else {
  35. // 2. 创建服务商信息
  36. supplierInfo := &gorm_model.YoungeeSupplier{
  37. SupplierName: phone,
  38. PhoneNumber: phone,
  39. UserId: *userId,
  40. }
  41. supplierId, createSupplierErr := db.CreateSupplier(ctx, *supplierInfo)
  42. fmt.Println(supplierId)
  43. if createSupplierErr != nil {
  44. log.Infof("[CreateSupplierUser] fail,err:%+v", createSupplierErr)
  45. return nil, createSupplierErr
  46. }
  47. res := &http_model.RegisterData{
  48. UserID: *userId,
  49. }
  50. return res, nil
  51. }
  52. }
  53. // GetSupplierIncomeList 查询服务商收入列表
  54. func (*supplier) GetSupplierIncomeList(ctx context.Context, req *http_model.FullSProjectIncomeListRequest) (*http_model.FullSProjectIncomeData, error) {
  55. var sProjectIncomeData *http_model.FullSProjectIncomeData
  56. sProjectIncomeData = &http_model.FullSProjectIncomeData{}
  57. // 1. 查询
  58. supplierIncome, total, err := db.GetSupplierIncomeList(ctx, req.PageSize, req.PageNum, req.SupplierId, req.IncomeStatus)
  59. if err != nil {
  60. return nil, nil
  61. }
  62. // 2. 补充种草/本地生活任务信息
  63. if supplierIncome != nil {
  64. sProjectIncomeData.Total = total
  65. for _, income := range supplierIncome {
  66. // 2.1. 种草任务基本信息
  67. if income.IncomeType == 1 {
  68. var sProjectInfo *http_model.FullSProjectIncomeListResponse
  69. sProjectInfo = &http_model.FullSProjectIncomeListResponse{}
  70. sProjectData, sProjectErr := db.GetSProjectDetail(ctx, income.SProjectID)
  71. if sProjectErr != nil {
  72. log.Infof("[GetSProjectDetail] fail,err:%+v", sProjectErr)
  73. return nil, sProjectErr
  74. }
  75. if sProjectData != nil {
  76. sProjectInfo.SProjectId = sProjectData.SProjectId
  77. sProjectInfo.IncomeId = income.IncomeID
  78. sProjectInfo.IncomeType = income.IncomeType
  79. sProjectInfo.SProjectId = sProjectData.SProjectId
  80. sProjectInfo.ProjectName = sProjectData.ProjectName
  81. sProjectInfo.ProjectPlatform = sProjectData.ProjectPlatform
  82. sProjectInfo.ServiceCharge = sProjectData.ServiceCharge
  83. sProjectInfo.ServiceChargeSettle = sProjectData.ServiceChargeSettle
  84. sProjectInfo.FinishTime = conv.MustString(sProjectData.FinishTime)
  85. // 2.2. 商品基本信息
  86. productInfo, productErr := db.GetProductByID(ctx, sProjectData.ProductId)
  87. if productErr != nil {
  88. log.Infof("[GetProductByID] fail,err:%+v", productErr)
  89. return nil, productErr
  90. }
  91. if productInfo != nil {
  92. sProjectInfo.ProductName = productInfo.ProductName
  93. sProjectInfo.ProductPrice = productInfo.ProductPrice
  94. }
  95. // 2.3. 商品图片信息
  96. productPhotoInfo, productPhotoErr := db.GetProductPhotoByProductID(ctx, sProjectData.ProductId)
  97. if productPhotoErr != nil {
  98. log.Infof("[GetProductPhotoByProductID] fail,err:%+v", productPhotoErr)
  99. return nil, productPhotoErr
  100. }
  101. if productPhotoInfo != nil {
  102. for _, photo := range productPhotoInfo {
  103. fmt.Println(photo)
  104. if photo.Symbol == 1 {
  105. sProjectInfo.ProductPhotoSymbol = 1
  106. sProjectInfo.ProductPhotoUrl = photo.PhotoUrl
  107. sProjectInfo.ProductPhotoUid = photo.PhotoUid
  108. }
  109. }
  110. }
  111. }
  112. }
  113. // 2.1. 本地生活任务基本信息
  114. if income.IncomeType == 2 {
  115. var sLocalInfo *http_model.FullSProjectIncomeListResponse
  116. sLocalInfo = &http_model.FullSProjectIncomeListResponse{}
  117. sLocalData, sLocalErr := db.GetSLocalLifeDetail(ctx, income.SLocalID)
  118. if sLocalErr != nil {
  119. log.Infof("[GetSLocalLifeDetail] fail,err:%+v", sLocalErr)
  120. return nil, sLocalErr
  121. }
  122. if sLocalData != nil {
  123. sLocalInfo.IncomeId = income.IncomeID
  124. sLocalInfo.IncomeType = income.IncomeType
  125. sLocalInfo.SLocalId = sLocalData.SLocalId
  126. sLocalInfo.LocalName = sLocalData.LocalName
  127. sLocalInfo.LocalPlatform = sLocalData.LocalPlatform
  128. sLocalInfo.ServiceCharge = sLocalData.ServiceCharge
  129. sLocalInfo.ServiceChargeSettle = sLocalData.ServiceChargeSettle
  130. sLocalInfo.FinishTime = conv.MustString(sLocalData.FinishTime)
  131. // 2.2. 门店基本信息
  132. storeInfo, storeErr := db.FindStoreById(ctx, sLocalData.StoreId)
  133. if storeErr != nil {
  134. log.Infof("[GetProductByID] fail,err:%+v", storeErr)
  135. return nil, storeErr
  136. }
  137. if storeInfo != nil {
  138. sLocalInfo.StoreName = storeInfo.StoreName
  139. }
  140. // 2.3. 门店图片信息
  141. storePhotoInfo, storePhotoErr := db.GetStorePhotoByStoreID(ctx, sLocalData.StoreId)
  142. if storePhotoErr != nil {
  143. log.Infof("[GetProductPhotoByProductID] fail,err:%+v", storePhotoErr)
  144. return nil, storePhotoErr
  145. }
  146. if storePhotoInfo != nil {
  147. for _, photo := range storePhotoInfo {
  148. fmt.Println(photo)
  149. if photo.Symbol == 1 {
  150. sLocalInfo.StoreMainPhotoSymbol = 1
  151. sLocalInfo.StoreMainPhotoUrl = photo.PhotoUrl
  152. sLocalInfo.StoreMainPhotoUid = photo.PhotoUid
  153. }
  154. }
  155. }
  156. }
  157. }
  158. }
  159. } else {
  160. sProjectIncomeData.Total = 0
  161. }
  162. return sProjectIncomeData, nil
  163. }
  164. // CreateSupplierInvoice 创建服务商发票
  165. func (*supplier) CreateSupplierInvoice(ctx context.Context, req *http_model.CreateSupplierInvoiceRequest) error {
  166. // 1. 数据转换
  167. var invoiceData *gorm_model.YounggeeSupplierInvoice
  168. invoiceData = &gorm_model.YounggeeSupplierInvoice{
  169. SupplierId: req.SupplierId,
  170. InvoiceStatus: 1,
  171. IncomeIds: req.IncomeIds,
  172. }
  173. if req.SOperatorType == 1 {
  174. invoiceData.SOperatorType = 1
  175. invoiceData.SOperator = req.SupplierId
  176. } else if req.SOperatorType == 2 {
  177. invoiceData.SOperatorType = 2
  178. invoiceData.SOperator = req.SubAccountId
  179. }
  180. // 2. 插入数据库
  181. err := db.CreateSupplierInvoice(ctx, invoiceData)
  182. if err != nil {
  183. return err
  184. }
  185. return nil
  186. }
  187. // UpdateSupplierInvoice 更新服务商发票
  188. func (*supplier) UpdateSupplierInvoice(ctx context.Context, req *http_model.UpdateSupplierInvoiceRequest) error {
  189. // 1. 数据转换
  190. var invoiceData *gorm_model.YounggeeSupplierInvoice
  191. var currentTime time.Time
  192. currentTime = time.Now()
  193. invoiceData = &gorm_model.YounggeeSupplierInvoice{
  194. InvoiceId: req.InvoiceId,
  195. InvoiceUrl: req.InvoiceUrl,
  196. Company: req.Company,
  197. UploadInvoiceTime: &currentTime,
  198. }
  199. // 2. 更新
  200. err := db.UpdateSupplierInvoice(ctx, invoiceData)
  201. if err != nil {
  202. return err
  203. }
  204. // 3. 更新服务商收入状态
  205. incomeIds, dbErr := db.GetIncomeIdsByInvoiceId(ctx, req.InvoiceId)
  206. if dbErr != nil {
  207. return dbErr
  208. }
  209. if incomeIds == "" {
  210. return nil
  211. }
  212. strSlice := strings.Split(incomeIds, ",")
  213. intSlice := make([]int, len(strSlice))
  214. for i, s := range strSlice {
  215. num, err := strconv.Atoi(s)
  216. if err != nil {
  217. fmt.Println("转换错误:", err)
  218. return err
  219. }
  220. intSlice[i] = num
  221. }
  222. updateSupplierIncomeErr := db.UpdateSupplierIncomeStatus(ctx, intSlice, 3)
  223. if updateSupplierIncomeErr != nil {
  224. return updateSupplierIncomeErr
  225. }
  226. return nil
  227. }
  228. // UpdateSupplierIncomeStatus 修改服务商收入状态
  229. func (*supplier) UpdateSupplierIncomeStatus(ctx context.Context, incomeIds string, incomeStatus int) error {
  230. // 1. 转换incomeIds为数组
  231. strSlice := strings.Split(incomeIds, ",")
  232. intSlice := make([]int, len(strSlice))
  233. for i, s := range strSlice {
  234. num, err := strconv.Atoi(s)
  235. if err != nil {
  236. fmt.Println("转换错误:", err)
  237. return err
  238. }
  239. intSlice[i] = num
  240. }
  241. // 2. 修改数据库中数据
  242. updateSupplierIncomeErr := db.UpdateSupplierIncomeStatus(ctx, intSlice, incomeStatus)
  243. if updateSupplierIncomeErr != nil {
  244. return updateSupplierIncomeErr
  245. }
  246. return nil
  247. }
  248. // GetSupplierInvoiceList 查找服务商发票列表
  249. func (*supplier) GetSupplierInvoiceList(ctx context.Context, req *http_model.SupplierInvoiceListRequest) (*http_model.SupplierInvoiceListData, error) {
  250. // 1. 查询服务商发票信息
  251. supplierInvoiceList, total, err := db.GetInvoiceListBySupplierId(ctx, req.SupplierId, req.InvoiceStatus, 4, req.PageSize, req.PageNum)
  252. if err != nil {
  253. return nil, err
  254. }
  255. // 2. 根据发票中的incomeIds去查找任务及其服务费收入
  256. var supplierInvoiceData *http_model.SupplierInvoiceListData
  257. supplierInvoiceData = &http_model.SupplierInvoiceListData{}
  258. for _, supplierInvoice := range supplierInvoiceList {
  259. var supplierInvoiceInfo *http_model.SupplierInvoiceInfo
  260. supplierInvoiceInfo = &http_model.SupplierInvoiceInfo{}
  261. // 2.1. 基础信息填入
  262. supplierInvoiceInfo.UploadInvoiceTime = supplierInvoice.UploadInvoiceTime
  263. supplierInvoiceInfo.InvoiceUrl = supplierInvoice.InvoiceUrl
  264. supplierInvoiceInfo.AgreeTime = supplierInvoice.AgreeTime
  265. supplierInvoiceInfo.RejectTime = supplierInvoice.RejectTime
  266. supplierInvoiceInfo.Company = supplierInvoice.Company
  267. supplierInvoiceInfo.SOperator = supplierInvoice.SOperator
  268. supplierInvoiceInfo.FailReason = supplierInvoice.FailReason
  269. // 2.2. 任务及其收入信息填入
  270. incomeIds := supplierInvoice.IncomeIds
  271. strSlice := strings.Split(incomeIds, ",")
  272. intSlice := make([]int, len(strSlice))
  273. for i, s := range strSlice {
  274. num, err := strconv.Atoi(s)
  275. if err != nil {
  276. fmt.Println("转换错误:", err)
  277. return nil, err
  278. }
  279. intSlice[i] = num
  280. }
  281. for _, incomeId := range intSlice {
  282. var sTaskInfo *http_model.STaskInfo
  283. sTaskInfo = &http_model.STaskInfo{}
  284. currIncome, incomeErr := db.GetIncomeInfoByIncomeId(ctx, incomeId)
  285. if incomeErr != nil {
  286. return nil, incomeErr
  287. }
  288. sTaskInfo.ServiceCharge = currIncome.ServiceChargeSettle
  289. supplierInvoiceInfo.Amount += currIncome.ServiceChargeSettle
  290. if currIncome.IncomeType == 1 {
  291. sTaskInfo.Id = currIncome.SProjectID
  292. } else if currIncome.IncomeType == 3 {
  293. sTaskInfo.Id = currIncome.SLocalID
  294. }
  295. supplierInvoiceInfo.STaskInfo = append(supplierInvoiceInfo.STaskInfo, sTaskInfo)
  296. }
  297. supplierInvoiceData.SupplierInvoiceList = append(supplierInvoiceData.SupplierInvoiceList, supplierInvoiceInfo)
  298. }
  299. supplierInvoiceData.Total = total
  300. return supplierInvoiceData, nil
  301. }
  302. // GetSupplierToWithdrawList 服务商待提现列表
  303. func (*supplier) GetSupplierToWithdrawList(ctx context.Context, req *http_model.SupplierToWithdrawListRequest) (*http_model.SupplierToWithdrawListData, error) {
  304. // 1. 判断服务商类型
  305. var supplierInvoiceData *http_model.SupplierToWithdrawListData
  306. supplierInvoiceData = &http_model.SupplierToWithdrawListData{}
  307. supplierInfo, supplierErr := db.GetSupplierById(ctx, req.SupplierId)
  308. if supplierErr != nil {
  309. return nil, supplierErr
  310. }
  311. if supplierInfo != nil {
  312. // 企业服务商
  313. if supplierInfo.SupplierType == 2 {
  314. // 查询企业服务商发票信息
  315. supplierInvoiceList, total, supplierInvoiceErr := db.GetInvoiceListBySupplierId(ctx, req.SupplierId, 3, 1, req.PageSize, req.PageNum)
  316. if supplierInvoiceErr != nil {
  317. return nil, supplierInvoiceErr
  318. }
  319. if supplierInvoiceList != nil {
  320. for _, supplierInvoice := range supplierInvoiceList {
  321. // 2. 根据发票中的incomeIds去查找任务及其服务费收入
  322. var supplierInvoiceInfo *http_model.SupplierToWithdrawInfo
  323. supplierInvoiceInfo = &http_model.SupplierToWithdrawInfo{}
  324. // supplierInvoiceInfo.Amount = 0.0
  325. // 2.1. 基础信息填入
  326. supplierInvoiceInfo.AgreeTime = supplierInvoice.AgreeTime
  327. supplierInvoiceInfo.Company = supplierInvoice.Company
  328. supplierInvoiceInfo.SupplierType = supplierInfo.SupplierType
  329. // 2.2. 任务及其收入信息填入
  330. incomeIds := supplierInvoice.IncomeIds
  331. strSlice := strings.Split(incomeIds, ",")
  332. intSlice := make([]int, len(strSlice))
  333. for i, s := range strSlice {
  334. num, err := strconv.Atoi(s)
  335. if err != nil {
  336. fmt.Println("转换错误:", err)
  337. return nil, err
  338. }
  339. intSlice[i] = num
  340. }
  341. for _, incomeId := range intSlice {
  342. var sTaskInfo *http_model.STaskInfo
  343. sTaskInfo = &http_model.STaskInfo{}
  344. currIncome, incomeErr := db.GetIncomeInfoByIncomeId(ctx, incomeId)
  345. if incomeErr != nil {
  346. return nil, incomeErr
  347. }
  348. sTaskInfo.ServiceCharge = currIncome.ServiceChargeSettle
  349. supplierInvoiceInfo.Amount += currIncome.ServiceChargeSettle
  350. if currIncome.IncomeType == 1 {
  351. sTaskInfo.Id = currIncome.SProjectID
  352. } else if currIncome.IncomeType == 3 {
  353. sTaskInfo.Id = currIncome.SLocalID
  354. }
  355. supplierInvoiceInfo.STaskInfo = append(supplierInvoiceInfo.STaskInfo, sTaskInfo)
  356. }
  357. supplierInvoiceData.ToWithdrawList = append(supplierInvoiceData.ToWithdrawList, supplierInvoiceInfo)
  358. }
  359. supplierInvoiceData.Total = total
  360. }
  361. } else if supplierInfo.SupplierType == 1 {
  362. // 个人服务商
  363. // 查询个人服务商收入信息
  364. supplierIncomeList, supplierIncomeTotal, supplierIncomeErr := db.GetSupplierIncomeList(ctx, req.PageSize, req.PageNum, req.SupplierId, 5)
  365. if supplierIncomeErr != nil {
  366. return nil, supplierIncomeErr
  367. }
  368. if supplierIncomeList != nil {
  369. supplierInvoiceData.Total = supplierIncomeTotal
  370. for _, supplierIncome := range supplierIncomeList {
  371. var supplierInvoiceInfo *http_model.SupplierToWithdrawInfo
  372. supplierInvoiceInfo = &http_model.SupplierToWithdrawInfo{}
  373. supplierInvoiceInfo.SupplierType = supplierInfo.SupplierType
  374. supplierInvoiceInfo.IncomeId = supplierIncome.IncomeID
  375. supplierInvoiceInfo.Amount = supplierIncome.ServiceChargeSettle
  376. supplierInvoiceInfo.AmountActual = supplierIncome.ServiceChargeSettle * 0.05
  377. var sTaskInfo *http_model.STaskInfo
  378. sTaskInfo = &http_model.STaskInfo{}
  379. sTaskInfo.ServiceCharge = supplierIncome.ServiceChargeSettle
  380. if supplierIncome.IncomeType == 1 {
  381. sTaskInfo.Id = supplierIncome.SProjectID
  382. } else if supplierIncome.IncomeType == 3 {
  383. sTaskInfo.Id = supplierIncome.SLocalID
  384. }
  385. supplierInvoiceInfo.STaskInfo = append(supplierInvoiceInfo.STaskInfo, sTaskInfo)
  386. supplierInvoiceData.ToWithdrawList = append(supplierInvoiceData.ToWithdrawList, supplierInvoiceInfo)
  387. }
  388. }
  389. }
  390. }
  391. return supplierInvoiceData, nil
  392. }
  393. // CreateSupplierWithdraw 创建服务商提现信息
  394. func (*supplier) CreateSupplierWithdraw(ctx context.Context, req *http_model.CreateSupplierWithdrawRequest) error {
  395. var supplierWithdrawInfoList []*gorm_model.YounggeeSupplierWithdraw
  396. // 1. 判断服务商类型
  397. supplierInfo, supplierErr := db.GetSupplierById(ctx, req.SupplierId)
  398. if supplierErr != nil {
  399. return supplierErr
  400. }
  401. if supplierInfo != nil {
  402. if supplierInfo.SupplierType == 1 {
  403. // 1.1. 个人服务商
  404. for _, withdrawInfo := range req.CreatePersonSupplierWithdraw {
  405. var supplierWithdrawInfo *gorm_model.YounggeeSupplierWithdraw
  406. supplierWithdrawInfo = &gorm_model.YounggeeSupplierWithdraw{}
  407. // 1.2.1. 接口传入信息填入
  408. supplierWithdrawInfo.SupplierId = req.SupplierId
  409. supplierWithdrawInfo.WithdrawStatus = 2
  410. supplierWithdrawInfo.BankName = withdrawInfo.BankName
  411. supplierWithdrawInfo.BankNumber = withdrawInfo.BankNumber
  412. var currentTime time.Time
  413. currentTime = time.Now()
  414. supplierWithdrawInfo.SupplyTime = &currentTime
  415. supplierWithdrawInfo.WithdrawAmount = 0.0
  416. // 1.2.2. 查找服务商信息填入
  417. supplierWithdrawInfo.Name = supplierInfo.Name
  418. supplierWithdrawInfo.Phone = withdrawInfo.Phone
  419. // 1.2.3. 收入信息填入
  420. currIncome, incomeInfoErr := db.GetIncomeInfoByIncomeId(ctx, withdrawInfo.IncomeId)
  421. if incomeInfoErr != nil {
  422. return incomeInfoErr
  423. }
  424. if currIncome != nil {
  425. supplierWithdrawInfo.WithdrawAmount += currIncome.ServiceChargeSettle
  426. }
  427. supplierWithdrawInfo.AmountPayable = supplierWithdrawInfo.WithdrawAmount
  428. supplierWithdrawInfoList = append(supplierWithdrawInfoList, supplierWithdrawInfo)
  429. }
  430. } else if supplierInfo.SupplierType == 2 {
  431. // 1.2. 机构服务商
  432. for _, withdrawInfo := range req.CreateCompanySupplierWithdraw {
  433. var supplierWithdrawInfo *gorm_model.YounggeeSupplierWithdraw
  434. supplierWithdrawInfo = &gorm_model.YounggeeSupplierWithdraw{}
  435. // 1.2.1. 接口传入信息填入
  436. supplierWithdrawInfo.SupplierId = req.SupplierId
  437. supplierWithdrawInfo.WithdrawStatus = 2
  438. supplierWithdrawInfo.BankName = withdrawInfo.BankName
  439. supplierWithdrawInfo.BankNumber = withdrawInfo.BankNumber
  440. var currentTime time.Time
  441. currentTime = time.Now()
  442. supplierWithdrawInfo.SupplyTime = &currentTime
  443. supplierWithdrawInfo.WithdrawAmount = 0.0
  444. // 1.2.2. 查找服务商信息填入
  445. supplierWithdrawInfo.Name = supplierInfo.Name
  446. supplierWithdrawInfo.Phone = supplierInfo.PhoneNumber
  447. supplierWithdrawInfo.Company = supplierInfo.CompanyName
  448. // 1.2.3. 收入信息填入
  449. incomeIds, incomeErr := db.GetIncomeIdsByInvoiceId(ctx, withdrawInfo.InvoiceId)
  450. if incomeErr != nil {
  451. return incomeErr
  452. }
  453. if incomeIds != "" {
  454. strSlice := strings.Split(incomeIds, ",")
  455. intSlice := make([]int, len(strSlice))
  456. for i, s := range strSlice {
  457. num, err := strconv.Atoi(s)
  458. if err != nil {
  459. fmt.Println("转换错误:", err)
  460. return err
  461. }
  462. intSlice[i] = num
  463. }
  464. for _, incomeId := range intSlice {
  465. currIncome, incomeInfoErr := db.GetIncomeInfoByIncomeId(ctx, incomeId)
  466. if incomeInfoErr != nil {
  467. return incomeInfoErr
  468. }
  469. if currIncome != nil {
  470. supplierWithdrawInfo.WithdrawAmount += currIncome.ServiceChargeSettle
  471. }
  472. }
  473. supplierWithdrawInfo.AmountPayable = supplierWithdrawInfo.WithdrawAmount
  474. }
  475. supplierWithdrawInfoList = append(supplierWithdrawInfoList, supplierWithdrawInfo)
  476. }
  477. }
  478. }
  479. // 2. 数据库插入
  480. err := db.CreateSupplierWithdraw(ctx, supplierWithdrawInfoList)
  481. if err != nil {
  482. return err
  483. }
  484. return nil
  485. }
  486. // GetSupplierWithdrawList 服务商提现列表
  487. func (*supplier) GetSupplierWithdrawList(ctx context.Context, req *http_model.SupplierWithdrawListRequest) (*http_model.SupplierWithdrawListData, error) {
  488. var supplierWithdrawListData *http_model.SupplierWithdrawListData
  489. supplierWithdrawListData = &http_model.SupplierWithdrawListData{}
  490. // 1. 根据服务商ID和提现状态去查找提现信息列表
  491. supplierWithdrawList, total, supplierWithdrawErr := db.GetSupplierWithdrawList(ctx, req.PageSize, req.PageNum, req.SupplierId, req.WithdrawStatus)
  492. if supplierWithdrawErr != nil {
  493. return nil, supplierWithdrawErr
  494. }
  495. if supplierWithdrawList != nil {
  496. supplierWithdrawListData.Total = total
  497. supplierInfo, supplierErr := db.GetSupplierById(ctx, req.SupplierId)
  498. if supplierErr != nil {
  499. return nil, supplierErr
  500. }
  501. if supplierInfo != nil {
  502. if supplierInfo.SupplierId == 1 {
  503. // 2. 个人服务商
  504. var supplierWithdrawInfo *http_model.SupplierWithdrawInfo
  505. supplierWithdrawInfo = &http_model.SupplierWithdrawInfo{}
  506. for _, withdrawInfo := range supplierWithdrawList {
  507. IncomeId, IncomeIdErr := strconv.Atoi(withdrawInfo.IncomeIds)
  508. if IncomeIdErr != nil {
  509. return nil, IncomeIdErr
  510. }
  511. incomeInfo, incomeErr := db.GetIncomeInfoByIncomeId(ctx, IncomeId)
  512. if incomeErr != nil {
  513. return nil, incomeErr
  514. }
  515. if incomeInfo != nil {
  516. var sTaskInfo *http_model.STaskInfo
  517. sTaskInfo = &http_model.STaskInfo{}
  518. if incomeInfo.IncomeType == 1 {
  519. sTaskInfo.Id = incomeInfo.SProjectID
  520. sTaskInfo.ServiceCharge = incomeInfo.ServiceChargeSettle
  521. } else if incomeInfo.IncomeType == 3 {
  522. sTaskInfo.Id = incomeInfo.SLocalID
  523. sTaskInfo.ServiceCharge = incomeInfo.ServiceChargeSettle
  524. }
  525. supplierWithdrawInfo.STaskInfo = append(supplierWithdrawInfo.STaskInfo, sTaskInfo)
  526. }
  527. supplierWithdrawInfo.SupplierType = 1
  528. supplierWithdrawInfo.SupplierWithdrawId = withdrawInfo.SupplierWithdrawId
  529. supplierWithdrawInfo.WithdrawAmount = withdrawInfo.WithdrawAmount
  530. supplierWithdrawInfo.AmountPayable = withdrawInfo.AmountPayable
  531. supplierWithdrawInfo.Name = withdrawInfo.Name
  532. supplierWithdrawInfo.IDNumber = withdrawInfo.IDNumber
  533. supplierWithdrawInfo.BankName = withdrawInfo.BankName
  534. supplierWithdrawInfo.BankNumber = withdrawInfo.BankNumber
  535. supplierWithdrawInfo.Phone = withdrawInfo.Phone
  536. // supplierWithdrawInfo.Company = withdrawInfo.Company
  537. supplierWithdrawInfo.SupplyTime = conv.MustString(withdrawInfo.SupplyTime)
  538. supplierWithdrawInfo.AgreeTime = conv.MustString(withdrawInfo.AgreeTime)
  539. supplierWithdrawInfo.RejectTime = conv.MustString(withdrawInfo.RejectTime)
  540. supplierWithdrawInfo.FailReason = withdrawInfo.FailReason
  541. supplierWithdrawListData.WithdrawList = append(supplierWithdrawListData.WithdrawList, supplierWithdrawInfo)
  542. }
  543. } else if supplierInfo.SupplierId == 2 {
  544. // 3. 企业服务商
  545. for _, withdrawInfo := range supplierWithdrawList {
  546. var supplierWithdrawInfo *http_model.SupplierWithdrawInfo
  547. supplierWithdrawInfo = &http_model.SupplierWithdrawInfo{}
  548. InvoiceId, InvoiceIdErr := strconv.Atoi(withdrawInfo.InvoiceIds)
  549. if InvoiceIdErr != nil {
  550. return nil, InvoiceIdErr
  551. }
  552. incomeIds, incomeIdsErr := db.GetIncomeIdsByInvoiceId(ctx, InvoiceId)
  553. if incomeIdsErr != nil {
  554. return nil, incomeIdsErr
  555. }
  556. if incomeIds != "" {
  557. strSlice := strings.Split(incomeIds, ",")
  558. intSlice := make([]int, len(strSlice))
  559. for i, s := range strSlice {
  560. num, err := strconv.Atoi(s)
  561. if err != nil {
  562. fmt.Println("转换错误:", err)
  563. return nil, err
  564. }
  565. intSlice[i] = num
  566. }
  567. for _, incomeId := range intSlice {
  568. var sTaskInfo *http_model.STaskInfo
  569. sTaskInfo = &http_model.STaskInfo{}
  570. currIncome, incomeInfoErr := db.GetIncomeInfoByIncomeId(ctx, incomeId)
  571. if incomeInfoErr != nil {
  572. return nil, incomeInfoErr
  573. }
  574. if currIncome != nil {
  575. if currIncome.IncomeType == 1 {
  576. sTaskInfo.Id = currIncome.SProjectID
  577. sTaskInfo.ServiceCharge = currIncome.ServiceChargeSettle
  578. } else if currIncome.IncomeType == 3 {
  579. sTaskInfo.Id = currIncome.SLocalID
  580. sTaskInfo.ServiceCharge = currIncome.ServiceChargeSettle
  581. }
  582. supplierWithdrawInfo.STaskInfo = append(supplierWithdrawInfo.STaskInfo, sTaskInfo)
  583. }
  584. }
  585. }
  586. supplierWithdrawInfo.SupplierType = 1
  587. supplierWithdrawInfo.SupplierWithdrawId = withdrawInfo.SupplierWithdrawId
  588. supplierWithdrawInfo.WithdrawAmount = withdrawInfo.WithdrawAmount
  589. supplierWithdrawInfo.AmountPayable = withdrawInfo.AmountPayable
  590. supplierWithdrawInfo.Name = withdrawInfo.Name
  591. supplierWithdrawInfo.IDNumber = withdrawInfo.IDNumber
  592. supplierWithdrawInfo.BankName = withdrawInfo.BankName
  593. supplierWithdrawInfo.BankNumber = withdrawInfo.BankNumber
  594. supplierWithdrawInfo.Phone = withdrawInfo.Phone
  595. supplierWithdrawInfo.Company = withdrawInfo.Company
  596. supplierWithdrawInfo.SupplyTime = conv.MustString(withdrawInfo.SupplyTime)
  597. supplierWithdrawInfo.AgreeTime = conv.MustString(withdrawInfo.AgreeTime)
  598. supplierWithdrawInfo.RejectTime = conv.MustString(withdrawInfo.RejectTime)
  599. supplierWithdrawInfo.FailReason = withdrawInfo.FailReason
  600. supplierWithdrawListData.WithdrawList = append(supplierWithdrawListData.WithdrawList, supplierWithdrawInfo)
  601. }
  602. }
  603. }
  604. }
  605. return supplierWithdrawListData, nil
  606. }
  607. func (*supplier) GetSupplierAmountBillList(ctx context.Context, req *http_model.SupplierAmountBillListRequest) (*http_model.SupplierAmountBillData, error) {
  608. return nil, nil
  609. }
  610. // GetManageInvoiceInfo 查找后台回票信息
  611. func (*supplier) GetManageInvoiceInfo(ctx context.Context, req *http_model.ManageInvoiceInfoRequest) (*http_model.ManageInvoiceInfoData, error) {
  612. var invoiceInfo *http_model.ManageInvoiceInfoData
  613. invoiceInfo = &http_model.ManageInvoiceInfoData{}
  614. manageInvoiceInfo, mamageInvoiceErr := db.GetManageInvoice(ctx)
  615. if mamageInvoiceErr != nil {
  616. return nil, mamageInvoiceErr
  617. }
  618. if manageInvoiceInfo != nil {
  619. invoiceInfo.InvoiceInfoID = manageInvoiceInfo.InvoiceInfoID
  620. invoiceInfo.Address = manageInvoiceInfo.Address
  621. invoiceInfo.Phone = manageInvoiceInfo.Phone
  622. invoiceInfo.BankNumber = manageInvoiceInfo.BankNumber
  623. invoiceInfo.EnterpriseName = manageInvoiceInfo.EnterpriseName
  624. invoiceInfo.ProjectNameToChoose = manageInvoiceInfo.ProjectNameToChoose
  625. invoiceInfo.BankName = manageInvoiceInfo.BankName
  626. }
  627. return invoiceInfo, nil
  628. }
  629. // GetWithdrawAmount 查找可提现、提现中、已提现金额
  630. func (*supplier) GetWithdrawAmount(ctx context.Context, req *http_model.WithdrawAmountRequest) (*http_model.WithdrawAmountData, error) {
  631. var amountInfo *http_model.WithdrawAmountData
  632. amountInfo = &http_model.WithdrawAmountData{}
  633. amountInfo.WithdrawAble = 0.0
  634. amountInfo.PendingWithdraw = 0.0
  635. amountInfo.Withdrawn = 0.0
  636. return amountInfo, nil
  637. }
  638. // GetSupplierBillAmount 服务商账单 总余额、可提现金额
  639. func (*supplier) GetSupplierBillAmount(ctx context.Context, req *http_model.SupplierAmountRequest) (*http_model.SupplierBillAmountData, error) {
  640. var incomeData *http_model.SupplierBillAmountData
  641. incomeData = &http_model.SupplierBillAmountData{}
  642. incomeData.FullAmount = 0.0
  643. incomeData.Settle = 0.0
  644. // 1. 个人服务商
  645. if req.SupplierType == 1 {
  646. supplierIncome, supplierIncomeErr := db.GetFullSupplierIncomeList(ctx, req.SupplierId, 5)
  647. if supplierIncomeErr != nil {
  648. return nil, supplierIncomeErr
  649. }
  650. if supplierIncome != nil {
  651. for _, income := range supplierIncome {
  652. incomeData.FullAmount += income.ServiceChargeSettle
  653. }
  654. incomeData.Settle = incomeData.FullAmount
  655. }
  656. } else if req.SupplierType == 2 {
  657. // 2. 企业服务商
  658. // 可提现
  659. supplierIncome, supplierIncomeErr := db.GetFullSupplierIncomeList(ctx, req.SupplierId, 5)
  660. if supplierIncomeErr != nil {
  661. return nil, supplierIncomeErr
  662. }
  663. if supplierIncome != nil {
  664. for _, income := range supplierIncome {
  665. incomeData.Settle += income.ServiceChargeSettle
  666. }
  667. incomeData.FullAmount = incomeData.Settle
  668. }
  669. // 可回票
  670. supplierToInvoice, supplierToInvoiceErr := db.GetFullSupplierIncomeList(ctx, req.SupplierId, 1)
  671. if supplierToInvoiceErr != nil {
  672. return nil, supplierToInvoiceErr
  673. }
  674. if supplierToInvoice != nil {
  675. for _, invoice := range supplierToInvoice {
  676. incomeData.FullAmount += invoice.ServiceChargeSettle
  677. }
  678. }
  679. }
  680. // fmt.Println(incomeData)
  681. return incomeData, nil
  682. }
  683. // GetWithdrawPaymentInfo 提现收款信息
  684. func (*supplier) GetWithdrawPaymentInfo(ctx context.Context, req *http_model.WithdrawPaymentInfoRequest) (*http_model.WithdrawPaymentInfoData, error) {
  685. var paymentInfo *http_model.WithdrawPaymentInfoData
  686. paymentInfo = &http_model.WithdrawPaymentInfoData{}
  687. // 1. 个人服务商
  688. if req.SupplierType == 1 {
  689. }
  690. return paymentInfo, nil
  691. }
  692. // UpdateWithdrawPaymentInfo 提现收款信息
  693. func (*supplier) UpdateWithdrawPaymentInfo(ctx context.Context, req *http_model.UpdateWithdrawPaymentInfoRequest) error {
  694. return nil
  695. }
  696. // GetSupplierInvoiceAmount 可回发票、待传发票、平台确认中、已回发票金额
  697. func (*supplier) GetSupplierInvoiceAmount(ctx context.Context, req *http_model.InvoiceAmountRequest) (*http_model.InvoiceAmountData, error) {
  698. var invoiceAmount *http_model.InvoiceAmountData
  699. invoiceAmount = &http_model.InvoiceAmountData{}
  700. invoiceAmount.InvoiceReturned = 0.0
  701. invoiceAmount.PendingUpload = 0.0
  702. invoiceAmount.CanUpload = 0.0
  703. invoiceAmount.PlatformConfirming = 0.0
  704. // 可回发票
  705. supplierIncome, supplierIncomeErr := db.GetFullSupplierIncomeList(ctx, req.SupplierId, 1)
  706. if supplierIncomeErr != nil {
  707. return nil, supplierIncomeErr
  708. }
  709. if supplierIncome != nil {
  710. for _, income := range supplierIncome {
  711. invoiceAmount.CanUpload += income.ServiceChargeSettle
  712. }
  713. }
  714. // 待传发票
  715. pendingUploadIncome, pendingUploadErr := db.GetFullSupplierIncomeList(ctx, req.SupplierId, 2)
  716. if pendingUploadErr != nil {
  717. return nil, pendingUploadErr
  718. }
  719. if pendingUploadIncome != nil {
  720. for _, income := range pendingUploadIncome {
  721. invoiceAmount.PendingUpload += income.ServiceChargeSettle
  722. }
  723. }
  724. // 平台确认中
  725. platformConfirmingIncome, platformConfirmingErr := db.GetFullSupplierIncomeList(ctx, req.SupplierId, 3)
  726. if platformConfirmingErr != nil {
  727. return nil, platformConfirmingErr
  728. }
  729. if platformConfirmingIncome != nil {
  730. for _, income := range platformConfirmingIncome {
  731. invoiceAmount.PlatformConfirming += income.ServiceChargeSettle
  732. }
  733. }
  734. // 已回票
  735. InvoiceReturnedIncome, InvoiceReturnedErr := db.GetFullSupplierIncomeList(ctx, req.SupplierId, 5)
  736. if InvoiceReturnedErr != nil {
  737. return nil, InvoiceReturnedErr
  738. }
  739. if InvoiceReturnedIncome != nil {
  740. for _, income := range InvoiceReturnedIncome {
  741. invoiceAmount.InvoiceReturned += income.ServiceChargeSettle
  742. }
  743. }
  744. return invoiceAmount, nil
  745. }