From 5421f9d925a15eddfcaf356839ee1a180b36689a Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 25 八月 2023 14:01:35 +0800
Subject: [PATCH] Merge branch 'hotfix'

---
 model/serviceCollectionPlan.go |   11 +++++++++++
 1 files changed, 11 insertions(+), 0 deletions(-)

diff --git a/model/serviceCollectionPlan.go b/model/serviceCollectionPlan.go
index cd72dee..3ffcf2e 100644
--- a/model/serviceCollectionPlan.go
+++ b/model/serviceCollectionPlan.go
@@ -17,6 +17,7 @@
 		SourceType       constvar.CollectionSourceType `gorm:"column:source_type;type:tinyint;not null;default 0;comment:婧愬崟绫诲瀷锛�1閿�鍞槑缁�2鏈嶅姟鍚堝悓3閿�鍞彂绁級" json:"sourceType"`       // 婧愬崟绫诲瀷锛�1閿�鍞槑缁�2鏈嶅姟鍚堝悓3閿�鍞彂绁級
 		SourceId         int                           `gorm:"column:source_id;type:int;not null;default 0;comment:婧愬崟id" json:"sourceId"`                                // 婧愬崟id
 		PrincipalId      int                           `gorm:"column:principal_id;type:int;not null;default 0;comment:鏀舵璐熻矗浜篒D" json:"principalId"`                       // 鏀舵璐熻矗浜篒D
+		Principal        User                          `gorm:"foreignKey:PrincipalId" json:"principal"`                                                                  // 鏀舵璐熻矗浜篒D
 		Term             int                           `gorm:"column:term;type:tinyint;not null;default 0;comment:鏈熸" json:"term"`                                       // 鏈熸
 		Percent          decimal.Decimal               `gorm:"column:percent;type:decimal(5,2);not null;default 0.00;comment:鏀舵姣斾緥" gorm:"" json:"percent"`               // 姣斾緥
 		Amount           decimal.Decimal               `gorm:"column:amount;type:decimal(12,2);not null;default '0.00';comment:閲戦" gorm:"" json:"amount"`                // 閲戦
@@ -62,6 +63,11 @@
 		db = db.Where("source_id = ?", slf.SourceId)
 	}
 
+	if slf.SourceType != 0 {
+		db = db.Where("source_type = ?", slf.SourceType)
+	}
+
+	db = db.Preload("Client").Preload("Principal")
 	return db
 }
 
@@ -107,6 +113,11 @@
 	return slf
 }
 
+func (slf *ServiceCollectionPlanSearch) SetSourceType(sourceType constvar.CollectionSourceType) *ServiceCollectionPlanSearch {
+	slf.SourceType = sourceType
+	return slf
+}
+
 func (slf *ServiceCollectionPlanSearch) First() (*ServiceCollectionPlan, error) {
 	var db = slf.build()
 	var record = new(ServiceCollectionPlan)

--
Gitblit v1.8.0