From aaa219a7ce91ce3e4bfaae4fb636580f804f7f03 Mon Sep 17 00:00:00 2001
From: liwei1dao <liwei1dao@gmail.com>
Date: Fri, 4 Nov 2022 18:46:27 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E6=94=AF=E4=BB=98=E7=B3=BB?=
 =?UTF-8?q?=E7=BB=9F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 modules/pay/modelPay.go |  9 +++++++--
 modules/pay/module.go   | 12 ++++++++++--
 services/worker/main.go |  2 ++
 3 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/modules/pay/modelPay.go b/modules/pay/modelPay.go
index 1508ba074..36e5b2f48 100644
--- a/modules/pay/modelPay.go
+++ b/modules/pay/modelPay.go
@@ -41,6 +41,11 @@ func (this *modelPayComp) queryUserPayData(uId string) (result []*pb.DBPayOrder,
 	return
 }
 
-func (this *modelPayComp) addUserPayOrder() {
-
+//添加用户订单数据
+func (this *modelPayComp) addUserPayOrder(order *pb.DBPayOrder) (err error) {
+	if err = this.AddList(order.Uid, order.Orderid, order); err != nil {
+		this.module.Errorf("err:%v", err)
+		return
+	}
+	return
 }
diff --git a/modules/pay/module.go b/modules/pay/module.go
index 565c344ac..f600786e2 100644
--- a/modules/pay/module.go
+++ b/modules/pay/module.go
@@ -9,6 +9,7 @@ import (
 	"go_dreamfactory/modules"
 	"go_dreamfactory/pb"
 	cfg "go_dreamfactory/sys/configure/structs"
+	"time"
 )
 
 /*
@@ -98,13 +99,20 @@ func (this *Pay) Rpc_ModulePayDelivery(ctx context.Context, args *pb.PayDelivery
 			return
 		}
 	}
-	
+	if err = this.modelPay.addUserPayOrder(&pb.DBPayOrder{
+		Orderid:   args.Orderid,
+		Uid:       args.Uid,
+		Productid: args.Productid,
+		Ctime:     time.Now().Unix(),
+	}); err != nil {
+		reply.Code = pb.ErrorCode_DBError
+		return
+	}
 	switch conf.RechargeType {
 	case 1:
 		break
 	case 2:
 		break
 	}
-
 	return
 }
diff --git a/services/worker/main.go b/services/worker/main.go
index edb2f46a7..3b03123c8 100644
--- a/services/worker/main.go
+++ b/services/worker/main.go
@@ -24,6 +24,7 @@ import (
 	"go_dreamfactory/modules/moonfantasy"
 	"go_dreamfactory/modules/notify"
 	"go_dreamfactory/modules/pagoda"
+	"go_dreamfactory/modules/pay"
 	"go_dreamfactory/modules/rtask"
 	"go_dreamfactory/modules/shop"
 	"go_dreamfactory/modules/smithy"
@@ -96,6 +97,7 @@ func main() {
 		troll.NewModule(),
 		sociaty.NewModule(),
 		horoscope.NewModule(),
+		pay.NewModule(),
 	)
 }