分批获取userIds

domain/model/user_m/user.go

@@ -85,12 +85,28 @@ func GetUsersByIds(model *domain.Model, userIds []mysql.ID) ([]User, error) {
 	return users, nil
 }
 
+// 分批获取userIds
 func GetUserMapByIds(model *domain.Model, userIds []mysql.ID) (map[mysql.ID]User, error) {
-	rows, err := GetUsersByIds(model, userIds)
-	if err != nil {
-		return nil, err
-	}
 	result := make(map[mysql.ID]User, 0)
+	end := 500
+	if end > len(userIds) {
+		end = len(userIds)
+	}
+	var rows []User
+	start := 0
+	for end <= len(userIds) {
+		if end > len(userIds) {
+			end = len(userIds)
+		}
+		tmp, err := GetUsersByIds(model, userIds[start:end])
+		if err != nil {
+			return result, err
+		} else {
+			rows = append(rows, tmp...)
+		}
+		start += 500
+		end += 500
+	}
 	for _, i := range rows {
 		result[i.ID] = i
 	}

test/user_test.go

+package test
+
+import (
+	"git.hilo.cn/hilo-common/domain"
+	"hilo-group/domain/model/user_m"
+	"testing"
+)
+
+func TestGetUserMap(t *testing.T) {
+	var userIds []uint64
+	for i := 0; i < 10502; i++ {
+		userIds = append(userIds, 7642)
+	}
+	res, err := user_m.GetUserMapByIds(domain.CreateModelNil(), userIds)
+	t.Logf("%v-%v", res, err)
+}