zhimeng
/
ad_platform
2
0
Fork 0
Code Issues 0 Pull Requests 0 Releases 0 Wiki Activity
广告平台(站长使用)
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
104 Commits
1 Branch
5.6 MiB
 
 
 
 
 
Tree: b0ffdd6a85
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'b0ffdd6a85'
${ noResults }
ad_platform/app/svc/svc_wx_data.go

531 lines
21 KiB
Raw Blame History 

  1. package svc

  2. 

  3. import (

  4. 	"applet/app/enum"

  5. 	"applet/app/md"

  6. 	"applet/app/utils"

  7. 	db "code.fnuoos.com/zhimeng/model.git/src"

  8. 	"code.fnuoos.com/zhimeng/model.git/src/super/implement"

  9. 	"code.fnuoos.com/zhimeng/model.git/src/super/model"

  10. 	"errors"

  11. 	"fmt"

  12. 	"github.com/shopspring/decimal"

  13. 	"time"

  14. )

  15. 

  16. func GenerateWxAdData(req md.GenerateWxAdData) (err error, generateWxAdData model.GenerateWxAdData) {

  17. 	// 1、查找原始数据记录

  18. 	originalWxAdDataDb := implement.NewOriginalWxAdDataDb(db.Db)

  19. 	originalWxAdData, err := originalWxAdDataDb.GetOriginalWxAdData(req.OriginalDataId)

  20. 	if err != nil {

  21. 		return

  22. 	}

  23. 	if originalWxAdData == nil {

  24. 		err = errors.New("未查询到原始数据记录")

  25. 		return

  26. 	}

  27. 

  28. 	// 2、查询对应媒体、代理的分成策略

  29. 	mediumDivisionStrategyDb := implement.NewMediumDivisionStrategyDb(db.Db)

  30. 	mediumDivisionStrategy, err := mediumDivisionStrategyDb.GetOriginalWxAdDataByMediumId(originalWxAdData.MediumId)

  31. 	if err != nil {

  32. 		return

  33. 	}

  34. 	if mediumDivisionStrategy == nil {

  35. 		err = errors.New("未查询到对应代理的分成策略")

  36. 		return

  37. 	}

  38. 	mediumDivisionStrategyWithAgentFlowDb := implement.NewMediumDivisionStrategyWithAgentFlowDb(db.Db)

  39. 	mediumDivisionStrategyWithAgentFlows, err := mediumDivisionStrategyWithAgentFlowDb.FindMediumDivisionStrategyWithAgentFlowByStrategyId(mediumDivisionStrategy.MediumId)

  40. 	if err != nil {

  41. 		return

  42. 	}

  43. 

  44. 	// 3、计算媒体、代理收益、平台留存、佣金留存、协议分成、协议总分成

  45. 	publisherIncome := float64(originalWxAdData.PublisherIncome)

  46. 	mediaRevenue := publisherIncome * float64(mediumDivisionStrategy.MediaRevenueRate) / 100                                      // 媒体收益

  47. 	agentRevenue := publisherIncome * float64(mediumDivisionStrategy.AgentRevenueRate) / 100                                      // 代理收益

  48. 	platformRetention := publisherIncome * float64(mediumDivisionStrategy.PlatformRetentionRate) / 100                            // 平台留存

  49. 	commissionRetention := publisherIncome * float64(mediumDivisionStrategy.CommissionRetentionRate) / 100                        // 佣金留存

  50. 	agreementSharingTotal := (mediaRevenue + agentRevenue) / ((100 - float64(mediumDivisionStrategy.AgreementSharingRate)) / 100) // 协议总分成（倒推）

  51. 	agreementSharing := agreementSharingTotal - (mediaRevenue + agentRevenue)                                                     // 协议分成

  52. 

  53. 	// 3、判断是否有调价留存

  54. 	var priceAdjustmentRetention int

  55. 	if req.NowEcpm != req.OriginalEcpm || req.NowExposureCount != req.OriginalExposureCount {

  56. 		tmpMediaRevenue := int(utils.StrToFloat64(req.NowEcpm) * float64(req.NowExposureCount) / 1000)

  57. 		priceAdjustmentRetention = int(float64(tmpMediaRevenue) - mediaRevenue)

  58. 		mediaRevenue = float64(tmpMediaRevenue)

  59. 	}

  60. 

  61. 	// 4、计算各代理收益

  62. 	var extraRevenue float64

  63. 	var agentRevenueFlows []struct {

  64. 		AgentId          int `json:"agent_id"`

  65. 		AgentRevenueRate int `json:"agent_revenue_rate"`

  66. 		ExtraRevenueRate int `json:"extra_revenue_rate"`

  67. 		AgentRevenue     int `json:"agent_revenue"`

  68. 		ExtraRevenue     int `json:"extra_revenue"`

  69. 	}

  70. 	for _, v := range *mediumDivisionStrategyWithAgentFlows {

  71. 		tmpAgentRevenue := agentRevenue * float64(v.AgentRevenueRate) / 100

  72. 		tmpExtraRevenue := commissionRetention * float64(v.ExtraRevenueRate) / 100

  73. 		extraRevenue += tmpExtraRevenue

  74. 		agentRevenueFlows = append(agentRevenueFlows, struct {

  75. 			AgentId          int `json:"agent_id"`

  76. 			AgentRevenueRate int `json:"agent_revenue_rate"`

  77. 			ExtraRevenueRate int `json:"extra_revenue_rate"`

  78. 			AgentRevenue     int `json:"agent_revenue"`

  79. 			ExtraRevenue     int `json:"extra_revenue"`

  80. 		}{

  81. 			AgentId:          v.AgentId,

  82. 			AgentRevenueRate: v.AgentRevenueRate,

  83. 			ExtraRevenueRate: v.ExtraRevenueRate,

  84. 			AgentRevenue:     int(tmpAgentRevenue),

  85. 			ExtraRevenue:     int(tmpExtraRevenue),

  86. 		})

  87. 	}

  88. 

  89. 	// 5、插入 generate_wx_ad_data 、generate_wx_ad_data_with_agent_flow 数据

  90. 	session := db.Db.NewSession()

  91. 	defer session.Close()

  92. 	session.Begin()

  93. 

  94. 	now := time.Now()

  95. 	generateWxAdData = model.GenerateWxAdData{

  96. 		MediumId:                 originalWxAdData.MediumId,

  97. 		Uuid:                     originalWxAdData.Uuid,

  98. 		AppId:                    originalWxAdData.AppId,

  99. 		PlatformAppId:            originalWxAdData.PlatformAppId,

  100. 		OriginalDataId:           originalWxAdData.Id,

  101. 		SlotId:                   originalWxAdData.SlotId,

  102. 		AdSlot:                   originalWxAdData.AdSlot,

  103. 		Date:                     originalWxAdData.Date,

  104. 		ReqSuccCount:             originalWxAdData.ReqSuccCount,

  105. 		ExposureCount:            req.NowExposureCount, // 现-曝光量

  106. 		ExposureRate:             originalWxAdData.ExposureRate,

  107. 		ClickCount:               originalWxAdData.ClickCount,

  108. 		ClickRate:                originalWxAdData.ClickRate,

  109. 		Ecpm:                     req.NowEcpm, // 现-ecpm

  110. 		PlatformRetention:        int(platformRetention),

  111. 		CommissionRetention:      int(commissionRetention),

  112. 		PriceAdjustmentRetention: priceAdjustmentRetention,

  113. 		MediaRevenue:             int(mediaRevenue),

  114. 		AgentRevenue:             int(agentRevenue),

  115. 		ExtraRevenue:             int(extraRevenue),

  116. 		AgreementSharing:         int(agreementSharing),

  117. 		AgreementSharingTotal:    int(agreementSharingTotal),

  118. 		PlatformRetentionRate:    mediumDivisionStrategy.PlatformRetentionRate,

  119. 		CommissionRetentionRate:  mediumDivisionStrategy.CommissionRetentionRate,

  120. 		MediaRevenueRate:         mediumDivisionStrategy.MediaRevenueRate,

  121. 		AgentRevenueRate:         mediumDivisionStrategy.AgentRevenueRate,

  122. 		ExtraRevenueRate:         mediumDivisionStrategy.ExtraRevenueRate,

  123. 		AgreementSharingRate:     mediumDivisionStrategy.AgreementSharingRate,

  124. 		IsGenerateReport:         0,

  125. 		CreateAt:                 now.Format("2006-01-02 15:04:05"),

  126. 		UpdateAt:                 now.Format("2006-01-02 15:04:05"),

  127. 		Platform:                 originalWxAdData.Platform,

  128. 	}

  129. 	generateWxAdDataDb := implement.NewGenerateWxAdDataDb(db.Db)

  130. 	_, err = generateWxAdDataDb.GenerateWxAdDataInsertBySession(session, &generateWxAdData)

  131. 	if err != nil {

  132. 		_ = session.Rollback()

  133. 		return

  134. 	}

  135. 

  136. 	var generateWxAdDataWithAgentFlows []*model.GenerateWxAdDataWithAgentFlow

  137. 	for _, v := range agentRevenueFlows {

  138. 		generateWxAdDataWithAgentFlows = append(generateWxAdDataWithAgentFlows, &model.GenerateWxAdDataWithAgentFlow{

  139. 			Uuid:             generateWxAdData.Uuid,

  140. 			AppId:            generateWxAdData.AppId,

  141. 			PlatformAppId:    generateWxAdData.PlatformAppId,

  142. 			AgentId:          v.AgentId,

  143. 			GenerateDataId:   generateWxAdData.Id,

  144. 			SlotId:           generateWxAdData.SlotId,

  145. 			AdSlot:           generateWxAdData.AdSlot,

  146. 			Date:             generateWxAdData.Date,

  147. 			AgentRevenue:     v.AgentRevenue,

  148. 			AgentRevenueRate: v.AgentRevenueRate,

  149. 			ExtraRevenue:     v.ExtraRevenue,

  150. 			ExtraRevenueRate: v.ExtraRevenueRate,

  151. 			CreateAt:         now.Format("2006-01-02 15:04:05"),

  152. 			UpdateAt:         now.Format("2006-01-02 15:04:05"),

  153. 		})

  154. 	}

  155. 	if len(generateWxAdDataWithAgentFlows) > 0 {

  156. 		generateWxAdDataWithAgentFlowDb := implement.NewGenerateWxAdDataWithAgentFlowDb(db.Db)

  157. 		_, err = generateWxAdDataWithAgentFlowDb.BatchAddGenerateWxAdDataWithAgentFlow(session, generateWxAdDataWithAgentFlows)

  158. 		if err != nil {

  159. 			_ = session.Rollback()

  160. 			return

  161. 		}

  162. 	}

  163. 

  164. 	// 6、修改 original_wx_ad_data 记录中的 is_apply(是否已应用(0:未 1:已) )

  165. 	originalWxAdData.IsApply = 1

  166. 	_, err = originalWxAdDataDb.UpdateOriginalWxAdDataBySession(session, originalWxAdData, "is_apply")

  167. 	if err != nil {

  168. 		_ = session.Rollback()

  169. 		return

  170. 	}

  171. 	return session.Commit(), generateWxAdData

  172. }

  173. 

  174. func ClacEcpm(req md.ClacEcpmReq) (err error, ecpm string) {

  175. 	// 1、查找原始数据记录

  176. 	originalWxAdDataDb := implement.NewOriginalWxAdDataDb(db.Db)

  177. 	originalWxAdData, err := originalWxAdDataDb.GetOriginalWxAdData(req.GenerateDataId)

  178. 	if err != nil {

  179. 		return

  180. 	}

  181. 	if originalWxAdData == nil {

  182. 		err = errors.New("未查询到原始数据记录")

  183. 		return

  184. 	}

  185. 

  186. 	// 2、查询对应媒体分成策略

  187. 	mediumDivisionStrategyDb := implement.NewMediumDivisionStrategyDb(db.Db)

  188. 	mediumDivisionStrategy, err := mediumDivisionStrategyDb.GetOriginalWxAdDataByMediumId(originalWxAdData.MediumId)

  189. 	if err != nil {

  190. 		return

  191. 	}

  192. 	if mediumDivisionStrategy == nil {

  193. 		err = errors.New("未查询到对应代理的分成策略")

  194. 		return

  195. 	}

  196. 

  197. 	// 3、计算媒体、代理收益、协议总分成

  198. 	publisherIncome := decimal.NewFromInt(int64(originalWxAdData.PublisherIncome))

  199. 	mediaRevenueRate := decimal.NewFromInt(int64(mediumDivisionStrategy.MediaRevenueRate))

  200. 	agentRevenueRate := decimal.NewFromInt(int64(mediumDivisionStrategy.AgentRevenueRate))

  201. 	agreementSharingRate := decimal.NewFromInt(int64(mediumDivisionStrategy.AgreementSharingRate))

  202. 	exposureCount := decimal.NewFromInt(int64(originalWxAdData.ExposureCount))

  203. 	rateValue := decimal.NewFromInt(100)

  204. 	thousandRateValue := decimal.NewFromInt(1000)

  205. 

  206. 	mediaRevenue := publisherIncome.Mul(mediaRevenueRate.Div(rateValue)) // 媒体收益

  207. 	agentRevenue := publisherIncome.Mul(agentRevenueRate.Div(rateValue)) // 代理收益

  208. 

  209. 	agreementSharingTotal := mediaRevenue.Add(agentRevenue).Div(rateValue.Sub(agreementSharingRate).Div(rateValue)) // 协议总分成（倒推）

  210. 

  211. 	fmt.Println(mediaRevenue, agentRevenue, agreementSharingRate, rateValue.Sub(agreementSharingRate), agreementSharingTotal)

  212. 	// 4、倒退出当前ecpm值

  213. 	ecpm = agreementSharingTotal.Div(exposureCount.Div(thousandRateValue)).Round(2).String()

  214. 	return

  215. }

  216. 

  217. func SettlementWxAdData(req md.SettlementWxAdData) (err error) {

  218. 	// 1、查找生成数据记录

  219. 	generateWxAdDataDb := implement.NewGenerateWxAdDataDb(db.Db)

  220. 	generateWxAdData, err := generateWxAdDataDb.GetGenerateWxAdData(req.GenerateDataId)

  221. 	if err != nil {

  222. 		return

  223. 	}

  224. 	if generateWxAdData == nil {

  225. 		err = errors.New("未查询到生成数据记录")

  226. 		return

  227. 	}

  228. 

  229. 	originalWxAdDataDb := implement.NewOriginalWxAdDataDb(db.Db)

  230. 	originalWxAdData, err := originalWxAdDataDb.GetOriginalWxAdData(generateWxAdData.OriginalDataId)

  231. 	if err != nil {

  232. 		return

  233. 	}

  234. 	if originalWxAdData == nil {

  235. 		err = errors.New("未查询到原始数据记录")

  236. 		return

  237. 	}

  238. 

  239. 	// 2、查询 生成数据-代理明细 记录

  240. 	generateWxAdDataWithAgentFlowDb := implement.NewGenerateWxAdDataWithAgentFlowDb(db.Db)

  241. 	generateWxAdDataWithAgentFlow, err := generateWxAdDataWithAgentFlowDb.FindGenerateWxAdDataWithAgentFlowByStrategyId(generateWxAdData.Id)

  242. 	if err != nil {

  243. 		return

  244. 	}

  245. 

  246. 	// 3、查询 媒体、代理的结算方式

  247. 	mediumListDb := implement.NewMediumListDb(db.Db)

  248. 	medium, err := mediumListDb.GetMediumList(originalWxAdData.MediumId)

  249. 	if err != nil {

  250. 		return

  251. 	}

  252. 	if medium == nil {

  253. 		return errors.New("媒体：" + utils.IntToStr(originalWxAdData.MediumId) + "未查询到对应记录")

  254. 	}

  255. 	var agentMap = map[int]model.AgentList{}

  256. 	agentListDb := implement.NewAgentListDb(db.Db)

  257. 	for _, v := range *generateWxAdDataWithAgentFlow {

  258. 		agent, err1 := agentListDb.GetAgentList(v.AgentId)

  259. 		if err1 != nil {

  260. 			return err1

  261. 		}

  262. 		if agent == nil {

  263. 			return errors.New("代理：" + utils.IntToStr(v.AgentId) + "未查询到对应记录")

  264. 		}

  265. 		agentMap[v.AgentId] = *agent

  266. 	}

  267. 

  268. 	// 4、生成数据

  269. 	now := time.Now()

  270. 	session := db.Db.NewSession()

  271. 	defer session.Close()

  272. 	session.Begin()

  273. 

  274. 	// 4.1 新增/更新 medium_settlement 数据

  275. 	mediumSettlementDb := implement.NewMediumSettlementDb(db.Db)

  276. 	mediumSettlement, err := mediumSettlementDb.GetMediumSettlementForAvailable(originalWxAdData.MediumId)

  277. 	if err != nil {

  278. 		return

  279. 	}

  280. 	var basicIncomeBefore, mediumSettlementState, mediumSettlementPayState int

  281. 	// 判断是否为预付结算类型

  282. 	if medium.SettlementType == enum.MediumSettlementTypeForPaymentInAdvance {

  283. 		mediumSettlementState = enum.MediumSettlementStateForCompleteSign

  284. 		mediumSettlementPayState = enum.MediumSettlementPayStateForPaymentAlready

  285. 	}

  286. 

  287. 	if mediumSettlement == nil {

  288. 		// 新增一条数据

  289. 		mediumSettlement = &model.MediumSettlement{

  290. 			Uuid:                     originalWxAdData.Uuid,

  291. 			MediumId:                 originalWxAdData.MediumId,

  292. 			AppId:                    originalWxAdData.AppId,

  293. 			PlatformAppId:            originalWxAdData.PlatformAppId,

  294. 			BusinessKind:             1,

  295. 			Kind:                     medium.SettlementType,

  296. 			BasicIncome:              generateWxAdData.MediaRevenue,

  297. 			OtherIncome:              0,

  298. 			OriginalIncome:           originalWxAdData.PublisherIncome,

  299. 			PlatformRetention:        generateWxAdData.PlatformRetention,

  300. 			CommissionRetention:      generateWxAdData.CommissionRetention,

  301. 			PriceAdjustmentRetention: generateWxAdData.PriceAdjustmentRetention,

  302. 			State:                    mediumSettlementState,

  303. 			PayState:                 mediumSettlementPayState,

  304. 			StartDate:                now.Format("2006-01-02"),

  305. 			EndDate:                  "",

  306. 			CreateAt:                 now.Format("2006-01-02 15:04:05"),

  307. 			UpdateAt:                 now.Format("2006-01-02 15:04:05"),

  308. 			SettleFile:               "",

  309. 		}

  310. 		_, err = mediumSettlementDb.MediumSettlementInsertBySession(session, mediumSettlement)

  311. 		if err != nil {

  312. 			_ = session.Rollback()

  313. 			return

  314. 		}

  315. 	} else {

  316. 		// 更新数据

  317. 		if medium.SettlementType != mediumSettlement.Kind {

  318. 			// TODO::前后结算类型不一致，之前结算单状态自动变成"核算中"

  319. 			mediumSettlement.State = enum.MediumSettlementStateForAccountingInProgress

  320. 			// mediumSettlement.PayState = mediumSettlementPayState

  321. 			_, err = mediumSettlementDb.UpdateMediumSettlementBySession(session, mediumSettlement, "state")

  322. 			if err != nil {

  323. 				_ = session.Rollback()

  324. 				return

  325. 			}

  326. 

  327. 			// 新增一条数据

  328. 			mediumSettlement = &model.MediumSettlement{

  329. 				Uuid:                     originalWxAdData.Uuid,

  330. 				MediumId:                 originalWxAdData.MediumId,

  331. 				AppId:                    originalWxAdData.AppId,

  332. 				PlatformAppId:            originalWxAdData.PlatformAppId,

  333. 				BusinessKind:             1,

  334. 				Kind:                     medium.SettlementType,

  335. 				BasicIncome:              generateWxAdData.MediaRevenue,

  336. 				OtherIncome:              0,

  337. 				OriginalIncome:           originalWxAdData.PublisherIncome,

  338. 				PlatformRetention:        generateWxAdData.PlatformRetention,

  339. 				CommissionRetention:      generateWxAdData.CommissionRetention,

  340. 				PriceAdjustmentRetention: generateWxAdData.PriceAdjustmentRetention,

  341. 				State:                    mediumSettlementState,

  342. 				PayState:                 mediumSettlementPayState,

  343. 				StartDate:                now.Format("2006-01-02"),

  344. 				EndDate:                  "",

  345. 				CreateAt:                 now.Format("2006-01-02 15:04:05"),

  346. 				UpdateAt:                 now.Format("2006-01-02 15:04:05"),

  347. 				SettleFile:               "",

  348. 			}

  349. 			_, err = mediumSettlementDb.MediumSettlementInsertBySession(session, mediumSettlement)

  350. 			if err != nil {

  351. 				_ = session.Rollback()

  352. 				return

  353. 			}

  354. 		} else {

  355. 			basicIncomeBefore = mediumSettlement.BasicIncome

  356. 			mediumSettlement.BasicIncome += generateWxAdData.MediaRevenue

  357. 			mediumSettlement.OriginalIncome += originalWxAdData.PublisherIncome

  358. 			mediumSettlement.PlatformRetention += generateWxAdData.PlatformRetention

  359. 			mediumSettlement.CommissionRetention += generateWxAdData.CommissionRetention

  360. 			mediumSettlement.PriceAdjustmentRetention += generateWxAdData.PriceAdjustmentRetention

  361. 			_, err = mediumSettlementDb.UpdateMediumSettlementBySession(session, mediumSettlement, "basic_income", "original_income", "platform_retention", "commission_retention", "price_adjustment_retention")

  362. 			if err != nil {

  363. 				_ = session.Rollback()

  364. 				return

  365. 			}

  366. 		}

  367. 	}

  368. 	if medium.SettlementType == enum.MediumSettlementTypeForPaymentInAdvance {

  369. 		err = DealMediumAmount(session, md.DealMediumAmount{

  370. 			Mid:      utils.IntToStr(originalWxAdData.Uuid),

  371. 			Type:     md.FinMediumFlowDirectionExpenditure,

  372. 			Kind:     md.SettlementSubKindForMediumFlow,

  373. 			OrdId:    utils.IntToStr(mediumSettlement.Id),

  374. 			MediumId: originalWxAdData.MediumId,

  375. 			Amount:   float64(generateWxAdData.MediaRevenue) / 100,

  376. 			Memo:     md.SettlementSubTitleForMediumFlow,

  377. 		})

  378. 		if err != nil {

  379. 			_ = session.Rollback()

  380. 			return

  381. 		}

  382. 	}

  383. 

  384. 	// 4.2 新增 medium_settlement_with_flow 数据

  385. 	mediumSettlementWithFlowDb := implement.NewMediumSettlementWithFlowDb(db.Db)

  386. 	_, err = mediumSettlementWithFlowDb.MediumSettlementWithFlowInsertBySession(session, &model.MediumSettlementWithFlow{

  387. 		SettlementId:      mediumSettlement.Id,

  388. 		GenerateDataId:    generateWxAdData.Id,

  389. 		Amount:            generateWxAdData.MediaRevenue,

  390. 		BasicIncomeBefore: basicIncomeBefore,

  391. 		BasicIncomeAfter:  mediumSettlement.BasicIncome,

  392. 		OtherIncomeBefore: 0,

  393. 		OtherIncomeAfter:  0,

  394. 		Kind:              enum.MediumSettlementWithFlowKindForBasicIncome,

  395. 		CreateAt:          now.Format("2006-01-02 15:04:05"),

  396. 		UpdateAt:          now.Format("2006-01-02 15:04:05"),

  397. 	})

  398. 	if err != nil {

  399. 		_ = session.Rollback()

  400. 		return

  401. 	}

  402. 	// 4.3 新增 agent_settlement、agent_settlement_with_flow 数据

  403. 	agentSettlementDb := implement.NewAgentSettlementDb(db.Db)

  404. 	agentSettlementWithFlowDb := implement.NewAgentSettlementWithFlowDb(db.Db)

  405. 	for _, v := range *generateWxAdDataWithAgentFlow {

  406. 		agentSettlement, err1 := agentSettlementDb.GetAgentSettlementForAvailable(v.AgentId)

  407. 		if err1 != nil {

  408. 			_ = session.Rollback()

  409. 			return err1

  410. 		}

  411. 

  412. 		var agentBasicIncomeBefore, agentOtherIncomeBefore, agentSettlementState, agentSettlementPayState int

  413. 		// 判断是否为预付结算类型

  414. 		if agentMap[v.AgentId].SettlementType == enum.AgentSettlementTypeForPaymentInAdvance {

  415. 			agentSettlementState = enum.AgentSettlementStateForCompleteSign

  416. 			agentSettlementPayState = enum.AgentSettlementPayStateForPaymentAlready

  417. 		}

  418. 		if agentSettlement == nil {

  419. 			// 新增一条数据

  420. 			agentSettlement = &model.AgentSettlement{

  421. 				Uuid:          v.Uuid,

  422. 				AgentId:       v.AgentId,

  423. 				MediumId:      originalWxAdData.MediumId,

  424. 				AppId:         v.AppId,

  425. 				PlatformAppId: v.PlatformAppId,

  426. 				BusinessKind:  1,

  427. 				Kind:          agentMap[v.AgentId].SettlementType,

  428. 				BasicIncome:   v.AgentRevenue,

  429. 				OtherIncome:   v.ExtraRevenue,

  430. 				State:         agentSettlementState,

  431. 				PayState:      agentSettlementPayState,

  432. 				StartDate:     now.Format("2006-01-02"),

  433. 				EndDate:       "",

  434. 				CreateAt:      now.Format("2006-01-02 15:04:05"),

  435. 				UpdateAt:      now.Format("2006-01-02 15:04:05"),

  436. 			}

  437. 			_, err = agentSettlementDb.AgentSettlementInsertBySession(session, agentSettlement)

  438. 			if err != nil {

  439. 				_ = session.Rollback()

  440. 				return

  441. 			}

  442. 		} else {

  443. 			// 更新数据

  444. 			if agentMap[v.AgentId].SettlementType != agentSettlement.Kind {

  445. 				// TODO::前后结算类型不一致，之前结算单状态自动变成"核算中"

  446. 				agentSettlement.State = enum.AgentSettlementStateForAccountingInProgress

  447. 				_, err = agentSettlementDb.UpdateAgentSettlementBySession(session, agentSettlement, "state")

  448. 				if err != nil {

  449. 					_ = session.Rollback()

  450. 					return

  451. 				}

  452. 

  453. 				// 新增一条数据

  454. 				agentSettlement = &model.AgentSettlement{

  455. 					Uuid:          v.Uuid,

  456. 					AgentId:       v.AgentId,

  457. 					MediumId:      originalWxAdData.MediumId,

  458. 					AppId:         v.AppId,

  459. 					PlatformAppId: v.PlatformAppId,

  460. 					BusinessKind:  1,

  461. 					Kind:          agentMap[v.AgentId].SettlementType,

  462. 					BasicIncome:   v.AgentRevenue,

  463. 					OtherIncome:   v.ExtraRevenue,

  464. 					State:         agentSettlementState,

  465. 					PayState:      agentSettlementPayState,

  466. 					StartDate:     now.Format("2006-01-02"),

  467. 					EndDate:       "",

  468. 					CreateAt:      now.Format("2006-01-02 15:04:05"),

  469. 					UpdateAt:      now.Format("2006-01-02 15:04:05"),

  470. 				}

  471. 				_, err = agentSettlementDb.AgentSettlementInsertBySession(session, agentSettlement)

  472. 				if err != nil {

  473. 					_ = session.Rollback()

  474. 					return

  475. 				}

  476. 			} else {

  477. 				agentBasicIncomeBefore = agentSettlement.BasicIncome

  478. 				agentOtherIncomeBefore = agentSettlement.OtherIncome

  479. 				agentSettlement.BasicIncome += v.AgentRevenue

  480. 				agentSettlement.OtherIncome += v.ExtraRevenue

  481. 

  482. 				_, err = agentSettlementDb.UpdateAgentSettlementBySession(session, agentSettlement, "basic_income", "other_income")

  483. 				if err != nil {

  484. 					_ = session.Rollback()

  485. 					return

  486. 				}

  487. 			}

  488. 		}

  489. 

  490. 		if agentMap[v.AgentId].SettlementType == enum.AgentSettlementTypeForPaymentInAdvance {

  491. 			err = DealAgentAmount(session, md.DealAgentAmount{

  492. 				Mid:    utils.IntToStr(originalWxAdData.Uuid),

  493. 				Type:   md.FinAgentFlowDirectionExpenditure,

  494. 				Kind:   md.SettlementSubKindForAgentFlow,

  495. 				OrdId:  utils.IntToStr(v.Id),

  496. 				Amount: float64(v.AgentRevenue)/100 + float64(v.ExtraRevenue)/100,

  497. 				Memo:   md.SettlementSubTitleForMediumFlow,

  498. 			})

  499. 			if err != nil {

  500. 				_ = session.Rollback()

  501. 				return

  502. 			}

  503. 		}

  504. 		_, err = agentSettlementWithFlowDb.AgentSettlementWithFlowInsertBySession(session, &model.AgentSettlementWithFlow{

  505. 			SettlementId:      mediumSettlement.Id,

  506. 			GenerateDataId:    generateWxAdData.Id,

  507. 			Amount:            v.AgentRevenue,

  508. 			BasicIncomeBefore: agentBasicIncomeBefore,

  509. 			BasicIncomeAfter:  agentSettlement.BasicIncome,

  510. 			OtherIncomeBefore: agentOtherIncomeBefore,

  511. 			OtherIncomeAfter:  agentSettlement.OtherIncome,

  512. 			Kind:              enum.AgentSettlementWithFlowKindForBasicIncome,

  513. 			CreateAt:          now.Format("2006-01-02 15:04:05"),

  514. 			UpdateAt:          now.Format("2006-01-02 15:04:05"),

  515. 		})

  516. 		if err != nil {

  517. 			_ = session.Rollback()

  518. 			return

  519. 		}

  520. 	}

  521. 

  522. 	// 4、修改 generate_wx_ad_data 记录中的 is_generate_report (是否已应用(0:未 1:已) )

  523. 	generateWxAdData.IsGenerateReport = 1

  524. 	_, err = generateWxAdDataDb.UpdateGenerateWxAdDataBySession(session, generateWxAdData, "is_generate_report")

  525. 	if err != nil {

  526. 		_ = session.Rollback()

  527. 		return

  528. 	}

  529. 	return session.Commit()

  530. }