From 41e61dd3a5c465d6c37bc836659e367b2d5d2abe Mon Sep 17 00:00:00 2001
From: meixiongfeng <766881921@qq.com>
Date: Wed, 4 Jan 2023 10:12:43 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B7=A8=E6=9C=8D=E8=B0=83=E7=94=A8=E8=8E=B7?=
 =?UTF-8?q?=E5=8F=96=E7=89=B9=E6=9D=83=E6=8E=A5=E5=8F=A3=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 modules/privilege/module.go | 36 +++++++++++++++++++++++++++---------
 1 file changed, 27 insertions(+), 9 deletions(-)

diff --git a/modules/privilege/module.go b/modules/privilege/module.go
index 2a95e15de..f693a0dfe 100644
--- a/modules/privilege/module.go
+++ b/modules/privilege/module.go
@@ -9,6 +9,7 @@ import (
 	"go_dreamfactory/pb"
 	"go_dreamfactory/sys/configure"
 	"go_dreamfactory/utils"
+	"time"
 
 	"go.mongodb.org/mongo-driver/bson/primitive"
 )
@@ -410,17 +411,34 @@ func (this *Privilege) SendDailyPrivilegeMail(session comm.IUserSession, cId []i
 	}
 }
 func (this *Privilege) GetCountByPrivilegeId(uid string, pId int32) (count int32) {
-	vip, err := this.modelVip.getVipList(uid)
-	if err != nil {
-		return
-	}
-	if v, ok := vip.Privilege[pId]; ok {
-		data := this.configure.GetPrivilegeByType(pId)
-		for _, v1 := range v.PrivilegeID {
-			if c, ok1 := data[v1]; ok1 {
-				count += c
+	if this.IsCross() { // 跨服情况
+		if model, err := this.GetDBModuleByUid(uid, comm.TableVip, time.Hour); err == nil {
+			vip := &pb.DBVip{}
+			if err = model.Get(uid, vip); err == nil {
+				if v, ok := vip.Privilege[pId]; ok {
+					data := this.configure.GetPrivilegeByType(pId)
+					for _, v1 := range v.PrivilegeID {
+						if c, ok1 := data[v1]; ok1 {
+							count += c
+						}
+					}
+				}
+			}
+		}
+	} else {
+		vip, err := this.modelVip.getVipList(uid)
+		if err != nil {
+			return
+		}
+		if v, ok := vip.Privilege[pId]; ok {
+			data := this.configure.GetPrivilegeByType(pId)
+			for _, v1 := range v.PrivilegeID {
+				if c, ok1 := data[v1]; ok1 {
+					count += c
+				}
 			}
 		}
 	}
+
 	return
 }