精简后的项目：仅保留 api/ui 源码、文档、构建依赖

api/dataloader/userLoader.go

@@ -0,0 +1,71 @@
+package dataloader
+
+import (
+	"time"
+
+	"github.com/photoview/photoview/api/graphql/models"
+	"gorm.io/gorm"
+)
+
+func NewUserLoaderByToken(db *gorm.DB) *UserLoader {
+	return &UserLoader{
+		maxBatch: 100,
+		wait:     5 * time.Millisecond,
+		fetch: func(tokens []string) ([]*models.User, []error) {
+
+			var accessTokens []*models.AccessToken
+			err := db.Where("expire > ?", time.Now()).Where("value IN (?)", tokens).Find(&accessTokens).Error
+			if err != nil {
+				return nil, []error{err}
+			}
+
+			rows, err := db.Table("access_tokens").Select("distinct user_id").Where("expire > ?", time.Now()).Where("value IN (?)", tokens).Rows()
+			if err != nil {
+				return nil, []error{err}
+			}
+			userIDs := make([]int, 0)
+			for rows.Next() {
+				var id int
+				if err := db.ScanRows(rows, &id); err != nil {
+					return nil, []error{err}
+				}
+				userIDs = append(userIDs, id)
+			}
+			rows.Close()
+
+			var userMap map[int]*models.User
+			if len(userIDs) > 0 {
+
+				var users []*models.User
+				if err := db.Where("id IN (?)", userIDs).Find(&users).Error; err != nil {
+					return nil, []error{err}
+				}
+
+				userMap = make(map[int]*models.User, len(users))
+				for _, user := range users {
+					userMap[user.ID] = user
+				}
+			} else {
+				userMap = make(map[int]*models.User, 0)
+			}
+
+			tokenMap := make(map[string]*models.AccessToken, len(tokens))
+			for _, token := range accessTokens {
+				tokenMap[token.Value] = token
+			}
+
+			result := make([]*models.User, len(tokens))
+			for i, token := range tokens {
+				accessToken, tokenFound := tokenMap[token]
+				if tokenFound {
+					user, userFound := userMap[accessToken.UserID]
+					if userFound {
+						result[i] = user
+					}
+				}
+			}
+
+			return result, nil
+		},
+	}
+}