分批获取所有家族成员

domain/model/groupPower_m/group_info.go

@@ -4,7 +4,6 @@ import (
 	"git.hilo.cn/hilo-common/domain"
 	"git.hilo.cn/hilo-common/resource/mysql"
 	"git.hilo.cn/hilo-common/utils"
-	"hilo-group/_const/enum/groupPower_e"
 )
 
 type GroupPowerInfo struct {
@@ -59,13 +58,19 @@ func GetGroupPowerUser(model *domain.Model, groupPowerId mysql.ID) ([]*GroupPowe
 	return m[groupPowerId], nil
 }
 
-// 获取所有家族的主
-func GetAllGroupPowerOwners(model *domain.Model) ([]mysql.ID, error) {
+// 分批获取所有家族成员
+func BatchGetAllGroupPowerUsers(model *domain.Model, lastId mysql.ID, limit int) ([]mysql.ID, mysql.ID, error) {
+	var res []GroupPowerUser
 	var userIds []mysql.ID
-	if err := model.DB().Model(GroupPowerUser{}).Select("user_id").
-		Where("role = ?", groupPower_e.GroupPowerUserRoleMgr).Find(&userIds).Error; err != nil {
-		model.Log.Errorf("GetAllGroupPowerOwners fail:%v", err)
-		return userIds, err
+	var nextLastId mysql.ID
+	if err := model.DB().Model(GroupPowerUser{}).
+		Where("id > ?", lastId).Order("id asc").Limit(limit).Find(&res).Error; err != nil {
+		model.Log.Errorf("BatchGetAllGroupPowerUsers fail:%v", err)
+		return userIds, 0, err
 	}
-	return userIds, nil
+	for _, v := range res {
+		userIds = append(userIds, v.UserId)
+		nextLastId = v.ID
+	}
+	return userIds, nextLastId, nil
 }

domain/service/event_s/event_init.go

@@ -195,23 +195,36 @@ func GroupPowerEvents() {
 
 	// 势力等级即将过期
 	group_power_ev.AddGroupPowerExpireNoticeAsync(func(model *domain.Model, e interface{}) error {
-		model.Log.Infof("msg AddGroupPowerExpireNoticeAsync")
-		powerOwners, err := groupPower_m.GetAllGroupPowerOwners(model)
-		if err != nil {
-			model.Log.Errorf("AddGroupPowerExpireNoticeAsync fail:%v", err)
-			return err
-		}
-		userMaps, err := user_m.GetUserMapByIds(model, powerOwners)
-		if err != nil {
-			return err
-		}
-		for uid, u := range userMaps {
-			if err := msg_m.NewUserRecord(model, uid, msg_e.GroupPowerExpireNotice, "", 0, "", "", "", "", "").Persistent(); err != nil {
-				model.Log.Errorf("msg AddGroupPowerExpireNoticeAsync NewUserRecord err:%v", err)
-				return nil
+		start := time.Now()
+		model.Log.Infof("msg AddGroupPowerExpireNoticeAsync,start:%v", start)
+		lastId := mysql.ID(0)
+		for {
+			powerUsers, nextLastId, err := groupPower_m.BatchGetAllGroupPowerUsers(model, lastId, 500)
+			if err != nil {
+				model.Log.Errorf("AddGroupPowerExpireNoticeAsync fail:%v", err)
+				return err
 			}
-			msg_m.SendEmasMsgAssistant(model, u.ExternalId, u.DeviceType)
+			if len(powerUsers) <= 0 {
+				break
+			}
+			userMaps, err := user_m.GetUserMapByIds(model, powerUsers)
+			if err != nil {
+				return err
+			}
+			for uid, u := range userMaps {
+				if err := msg_m.NewUserRecord(model, uid, msg_e.GroupPowerExpireNotice, "", 0, "", "", "", "", "").Persistent(); err != nil {
+					model.Log.Errorf("msg AddGroupPowerExpireNoticeAsync NewUserRecord err:%v", err)
+					return nil
+				}
+				_ = msg_m.SendEmasMsgAssistant(model, u.ExternalId, u.DeviceType)
+			}
+			if nextLastId <= 0 {
+				break
+			}
+			lastId = nextLastId
+			time.Sleep(time.Second) // 停顿一秒
 		}
+		model.Log.Infof("msg AddGroupPowerExpireNoticeAsync,end:%v", time.Now().Sub(start).Seconds())
 		return nil
 	})
 }

test/group_power_test.go

@@ -7,7 +7,7 @@ import (
 )
 
 func TestGetAllGroupPowerOwner(t *testing.T) {
-	owners, err := groupPower_m.GetAllGroupPowerOwners(domain.CreateModelNil())
+	owners, err := groupPower_m.BatchGetAllGroupPowerUsers(domain.CreateModelNil())
 	println(owners, err)
 }