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.
102 Commits
1 Branch
3.4 MiB
 
 
 
 
 
Tree: cc0bd9d3cd
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'cc0bd9d3cd'
${ noResults }
ad_platform/app/svc/svc_wx_data.go

527 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. 	"github.com/shopspring/decimal"

  12. 	"time"

  13. )

  14. 

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

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

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

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

  19. 	if err != nil {

  20. 		return

  21. 	}

  22. 	if originalWxAdData == nil {

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

  24. 		return

  25. 	}

  26. 

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

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

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

  30. 	if err != nil {

  31. 		return

  32. 	}

  33. 	if mediumDivisionStrategy == nil {

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

  35. 		return

  36. 	}

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

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

  39. 	if err != nil {

  40. 		return

  41. 	}

  42. 

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

  44. 	publisherIncome := originalWxAdData.PublisherIncome

  45. 	mediaRevenue := publisherIncome * mediumDivisionStrategy.MediaRevenueRate / 100                              // 媒体收益

  46. 	agentRevenue := publisherIncome * mediumDivisionStrategy.AgentRevenueRate / 100                              // 代理收益

  47. 	platformRetention := publisherIncome * mediumDivisionStrategy.PlatformRetentionRate / 100                    // 平台留存

  48. 	commissionRetention := publisherIncome * mediumDivisionStrategy.CommissionRetentionRate / 100                // 佣金留存

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

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

  51. 

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

  53. 	var priceAdjustmentRetention int

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

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

  56. 		priceAdjustmentRetention = tmpMediaRevenue - mediaRevenue

  57. 		mediaRevenue = tmpMediaRevenue

  58. 	}

  59. 

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

  61. 	var extraRevenue int

  62. 	var agentRevenueFlows []struct {

  63. 		AgentId          int `json:"agent_id"`

  64. 		AgentRevenueRate int `json:"agent_revenue_rate"`

  65. 		ExtraRevenueRate int `json:"extra_revenue_rate"`

  66. 		AgentRevenue     int `json:"agent_revenue"`

  67. 		ExtraRevenue     int `json:"extra_revenue"`

  68. 	}

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

  70. 		tmpAgentRevenue := agentRevenue * v.AgentRevenueRate / 100

  71. 		tmpExtraRevenue := commissionRetention * v.ExtraRevenueRate / 100

  72. 		extraRevenue += tmpExtraRevenue

  73. 		agentRevenueFlows = append(agentRevenueFlows, struct {

  74. 			AgentId          int `json:"agent_id"`

  75. 			AgentRevenueRate int `json:"agent_revenue_rate"`

  76. 			ExtraRevenueRate int `json:"extra_revenue_rate"`

  77. 			AgentRevenue     int `json:"agent_revenue"`

  78. 			ExtraRevenue     int `json:"extra_revenue"`

  79. 		}{

  80. 			AgentId:          v.AgentId,

  81. 			AgentRevenueRate: v.AgentRevenueRate,

  82. 			ExtraRevenueRate: v.ExtraRevenueRate,

  83. 			AgentRevenue:     tmpAgentRevenue,

  84. 			ExtraRevenue:     tmpExtraRevenue,

  85. 		})

  86. 	}

  87. 

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

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

  90. 	defer session.Close()

  91. 	session.Begin()

  92. 

  93. 	now := time.Now()

  94. 	generateWxAdData = model.GenerateWxAdData{

  95. 		MediumId:                 originalWxAdData.MediumId,

  96. 		Uuid:                     originalWxAdData.Uuid,

  97. 		AppId:                    originalWxAdData.AppId,

  98. 		PlatformAppId:            originalWxAdData.PlatformAppId,

  99. 		OriginalDataId:           originalWxAdData.Id,

  100. 		SlotId:                   originalWxAdData.SlotId,

  101. 		AdSlot:                   originalWxAdData.AdSlot,

  102. 		Date:                     originalWxAdData.Date,

  103. 		ReqSuccCount:             originalWxAdData.ReqSuccCount,

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

  105. 		ExposureRate:             originalWxAdData.ExposureRate,

  106. 		ClickCount:               originalWxAdData.ClickCount,

  107. 		ClickRate:                originalWxAdData.ClickRate,

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

  109. 		PlatformRetention:        platformRetention,

  110. 		CommissionRetention:      commissionRetention,

  111. 		PriceAdjustmentRetention: priceAdjustmentRetention,

  112. 		MediaRevenue:             mediaRevenue,

  113. 		AgentRevenue:             agentRevenue,

  114. 		ExtraRevenue:             extraRevenue,

  115. 		AgreementSharing:         agreementSharing,

  116. 		AgreementSharingTotal:    agreementSharingTotal,

  117. 		PlatformRetentionRate:    mediumDivisionStrategy.PlatformRetentionRate,

  118. 		CommissionRetentionRate:  mediumDivisionStrategy.CommissionRetentionRate,

  119. 		MediaRevenueRate:         mediumDivisionStrategy.MediaRevenueRate,

  120. 		AgentRevenueRate:         mediumDivisionStrategy.AgentRevenueRate,

  121. 		ExtraRevenueRate:         mediumDivisionStrategy.ExtraRevenueRate,

  122. 		AgreementSharingRate:     mediumDivisionStrategy.AgreementSharingRate,

  123. 		IsGenerateReport:         0,

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

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

  126. 		Platform:                 originalWxAdData.Platform,

  127. 	}

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

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

  130. 	if err != nil {

  131. 		_ = session.Rollback()

  132. 		return

  133. 	}

  134. 

  135. 	var generateWxAdDataWithAgentFlows []*model.GenerateWxAdDataWithAgentFlow

  136. 	for _, v := range agentRevenueFlows {

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

  138. 			Uuid:             generateWxAdData.Uuid,

  139. 			AppId:            generateWxAdData.AppId,

  140. 			PlatformAppId:    generateWxAdData.PlatformAppId,

  141. 			AgentId:          v.AgentId,

  142. 			GenerateDataId:   generateWxAdData.Id,

  143. 			SlotId:           generateWxAdData.SlotId,

  144. 			AdSlot:           generateWxAdData.AdSlot,

  145. 			Date:             generateWxAdData.Date,

  146. 			AgentRevenue:     v.AgentRevenue,

  147. 			AgentRevenueRate: v.AgentRevenueRate,

  148. 			ExtraRevenue:     v.ExtraRevenue,

  149. 			ExtraRevenueRate: v.ExtraRevenueRate,

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

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

  152. 		})

  153. 	}

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

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

  156. 	if err != nil {

  157. 		_ = session.Rollback()

  158. 		return

  159. 	}

  160. 

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

  162. 	originalWxAdData.IsApply = 1

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

  164. 	if err != nil {

  165. 		_ = session.Rollback()

  166. 		return

  167. 	}

  168. 	return session.Commit(), generateWxAdData

  169. }

  170. 

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

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

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

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

  175. 	if err != nil {

  176. 		return

  177. 	}

  178. 	if originalWxAdData == nil {

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

  180. 		return

  181. 	}

  182. 

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

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

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

  186. 	if err != nil {

  187. 		return

  188. 	}

  189. 	if mediumDivisionStrategy == nil {

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

  191. 		return

  192. 	}

  193. 

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

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

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

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

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

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

  200. 	rateValue := decimal.NewFromInt(100)

  201. 	thousandRateValue := decimal.NewFromInt(1000)

  202. 

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

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

  205. 

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

  207. 

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

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

  210. 	return

  211. }

  212. 

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

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

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

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

  217. 	if err != nil {

  218. 		return

  219. 	}

  220. 	if generateWxAdData == nil {

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

  222. 		return

  223. 	}

  224. 

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

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

  227. 	if err != nil {

  228. 		return

  229. 	}

  230. 	if originalWxAdData == nil {

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

  232. 		return

  233. 	}

  234. 

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

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

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

  238. 	if err != nil {

  239. 		return

  240. 	}

  241. 

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

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

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

  245. 	if err != nil {

  246. 		return

  247. 	}

  248. 	if medium == nil {

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

  250. 	}

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

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

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

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

  255. 		if err1 != nil {

  256. 			return err1

  257. 		}

  258. 		if agent == nil {

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

  260. 		}

  261. 		agentMap[v.AgentId] = *agent

  262. 	}

  263. 

  264. 	// 4、生成数据

  265. 	now := time.Now()

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

  267. 	defer session.Close()

  268. 	session.Begin()

  269. 

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

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

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

  273. 	if err != nil {

  274. 		return

  275. 	}

  276. 	var basicIncomeBefore, mediumSettlementState, mediumSettlementPayState int

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

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

  279. 		mediumSettlementState = enum.MediumSettlementStateForCompleteSign

  280. 		mediumSettlementPayState = enum.MediumSettlementPayStateForPaymentAlready

  281. 	}

  282. 

  283. 	if mediumSettlement == nil {

  284. 		// 新增一条数据

  285. 		mediumSettlement = &model.MediumSettlement{

  286. 			Uuid:                     originalWxAdData.Uuid,

  287. 			MediumId:                 originalWxAdData.MediumId,

  288. 			AppId:                    originalWxAdData.AppId,

  289. 			PlatformAppId:            originalWxAdData.PlatformAppId,

  290. 			BusinessKind:             1,

  291. 			Kind:                     medium.SettlementType,

  292. 			BasicIncome:              generateWxAdData.MediaRevenue,

  293. 			OtherIncome:              0,

  294. 			OriginalIncome:           originalWxAdData.PublisherIncome,

  295. 			PlatformRetention:        generateWxAdData.PlatformRetention,

  296. 			CommissionRetention:      generateWxAdData.CommissionRetention,

  297. 			PriceAdjustmentRetention: generateWxAdData.PriceAdjustmentRetention,

  298. 			State:                    mediumSettlementState,

  299. 			PayState:                 mediumSettlementPayState,

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

  301. 			EndDate:                  "",

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

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

  304. 			SettleFile:               "",

  305. 		}

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

  307. 		if err != nil {

  308. 			_ = session.Rollback()

  309. 			return

  310. 		}

  311. 	} else {

  312. 		// 更新数据

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

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

  315. 			mediumSettlement.State = enum.MediumSettlementStateForAccountingInProgress

  316. 			// mediumSettlement.PayState = mediumSettlementPayState

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

  318. 			if err != nil {

  319. 				_ = session.Rollback()

  320. 				return

  321. 			}

  322. 

  323. 			// 新增一条数据

  324. 			mediumSettlement = &model.MediumSettlement{

  325. 				Uuid:                     originalWxAdData.Uuid,

  326. 				MediumId:                 originalWxAdData.MediumId,

  327. 				AppId:                    originalWxAdData.AppId,

  328. 				PlatformAppId:            originalWxAdData.PlatformAppId,

  329. 				BusinessKind:             1,

  330. 				Kind:                     medium.SettlementType,

  331. 				BasicIncome:              generateWxAdData.MediaRevenue,

  332. 				OtherIncome:              0,

  333. 				OriginalIncome:           originalWxAdData.PublisherIncome,

  334. 				PlatformRetention:        generateWxAdData.PlatformRetention,

  335. 				CommissionRetention:      generateWxAdData.CommissionRetention,

  336. 				PriceAdjustmentRetention: generateWxAdData.PriceAdjustmentRetention,

  337. 				State:                    mediumSettlementState,

  338. 				PayState:                 mediumSettlementPayState,

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

  340. 				EndDate:                  "",

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

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

  343. 				SettleFile:               "",

  344. 			}

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

  346. 			if err != nil {

  347. 				_ = session.Rollback()

  348. 				return

  349. 			}

  350. 		} else {

  351. 			basicIncomeBefore = mediumSettlement.BasicIncome

  352. 			mediumSettlement.BasicIncome += generateWxAdData.MediaRevenue

  353. 			mediumSettlement.OriginalIncome += originalWxAdData.PublisherIncome

  354. 			mediumSettlement.PlatformRetention += generateWxAdData.PlatformRetention

  355. 			mediumSettlement.CommissionRetention += generateWxAdData.CommissionRetention

  356. 			mediumSettlement.PriceAdjustmentRetention += generateWxAdData.PriceAdjustmentRetention

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

  358. 			if err != nil {

  359. 				_ = session.Rollback()

  360. 				return

  361. 			}

  362. 		}

  363. 	}

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

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

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

  367. 			Type:     md.FinMediumFlowDirectionExpenditure,

  368. 			Kind:     md.SettlementSubKindForMediumFlow,

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

  370. 			MediumId: originalWxAdData.MediumId,

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

  372. 			Memo:     md.SettlementSubTitleForMediumFlow,

  373. 		})

  374. 		if err != nil {

  375. 			_ = session.Rollback()

  376. 			return

  377. 		}

  378. 	}

  379. 

  380. 	// 4.2 新增 medium_settlement_with_flow 数据

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

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

  383. 		SettlementId:      mediumSettlement.Id,

  384. 		GenerateDataId:    generateWxAdData.Id,

  385. 		Amount:            generateWxAdData.MediaRevenue,

  386. 		BasicIncomeBefore: basicIncomeBefore,

  387. 		BasicIncomeAfter:  mediumSettlement.BasicIncome,

  388. 		OtherIncomeBefore: 0,

  389. 		OtherIncomeAfter:  0,

  390. 		Kind:              enum.MediumSettlementWithFlowKindForBasicIncome,

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

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

  393. 	})

  394. 	if err != nil {

  395. 		_ = session.Rollback()

  396. 		return

  397. 	}

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

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

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

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

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

  403. 		if err1 != nil {

  404. 			_ = session.Rollback()

  405. 			return err1

  406. 		}

  407. 

  408. 		var agentBasicIncomeBefore, agentOtherIncomeBefore, agentSettlementState, agentSettlementPayState int

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

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

  411. 			agentSettlementState = enum.AgentSettlementStateForCompleteSign

  412. 			agentSettlementPayState = enum.AgentSettlementPayStateForPaymentAlready

  413. 		}

  414. 		if agentSettlement == nil {

  415. 			// 新增一条数据

  416. 			agentSettlement = &model.AgentSettlement{

  417. 				Uuid:          v.Uuid,

  418. 				AgentId:       v.AgentId,

  419. 				MediumId:      originalWxAdData.MediumId,

  420. 				AppId:         v.AppId,

  421. 				PlatformAppId: v.PlatformAppId,

  422. 				BusinessKind:  1,

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

  424. 				BasicIncome:   v.AgentRevenue,

  425. 				OtherIncome:   v.ExtraRevenue,

  426. 				State:         agentSettlementState,

  427. 				PayState:      agentSettlementPayState,

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

  429. 				EndDate:       "",

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

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

  432. 			}

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

  434. 			if err != nil {

  435. 				_ = session.Rollback()

  436. 				return

  437. 			}

  438. 		} else {

  439. 			// 更新数据

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

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

  442. 				agentSettlement.State = enum.AgentSettlementStateForAccountingInProgress

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

  444. 				if err != nil {

  445. 					_ = session.Rollback()

  446. 					return

  447. 				}

  448. 

  449. 				// 新增一条数据

  450. 				agentSettlement = &model.AgentSettlement{

  451. 					Uuid:          v.Uuid,

  452. 					AgentId:       v.AgentId,

  453. 					MediumId:      originalWxAdData.MediumId,

  454. 					AppId:         v.AppId,

  455. 					PlatformAppId: v.PlatformAppId,

  456. 					BusinessKind:  1,

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

  458. 					BasicIncome:   v.AgentRevenue,

  459. 					OtherIncome:   v.ExtraRevenue,

  460. 					State:         agentSettlementState,

  461. 					PayState:      agentSettlementPayState,

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

  463. 					EndDate:       "",

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

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

  466. 				}

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

  468. 				if err != nil {

  469. 					_ = session.Rollback()

  470. 					return

  471. 				}

  472. 			} else {

  473. 				agentBasicIncomeBefore = agentSettlement.BasicIncome

  474. 				agentOtherIncomeBefore = agentSettlement.OtherIncome

  475. 				agentSettlement.BasicIncome += v.AgentRevenue

  476. 				agentSettlement.OtherIncome += v.ExtraRevenue

  477. 

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

  479. 				if err != nil {

  480. 					_ = session.Rollback()

  481. 					return

  482. 				}

  483. 			}

  484. 		}

  485. 

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

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

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

  489. 				Type:   md.FinAgentFlowDirectionExpenditure,

  490. 				Kind:   md.SettlementSubKindForAgentFlow,

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

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

  493. 				Memo:   md.SettlementSubTitleForMediumFlow,

  494. 			})

  495. 			if err != nil {

  496. 				_ = session.Rollback()

  497. 				return

  498. 			}

  499. 		}

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

  501. 			SettlementId:      mediumSettlement.Id,

  502. 			GenerateDataId:    generateWxAdData.Id,

  503. 			Amount:            v.AgentRevenue,

  504. 			BasicIncomeBefore: agentBasicIncomeBefore,

  505. 			BasicIncomeAfter:  agentSettlement.BasicIncome,

  506. 			OtherIncomeBefore: agentOtherIncomeBefore,

  507. 			OtherIncomeAfter:  agentSettlement.OtherIncome,

  508. 			Kind:              enum.AgentSettlementWithFlowKindForBasicIncome,

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

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

  511. 		})

  512. 		if err != nil {

  513. 			_ = session.Rollback()

  514. 			return

  515. 		}

  516. 	}

  517. 

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

  519. 	generateWxAdData.IsGenerateReport = 1

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

  521. 	if err != nil {

  522. 		_ = session.Rollback()

  523. 		return

  524. 	}

  525. 	return session.Commit()

  526. }