101 lines
2.3 KiB
Go
101 lines
2.3 KiB
Go
package api
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"net/http"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
|
||
"server/internal/grpc"
|
||
)
|
||
|
||
type YMTUsersAPI struct {
|
||
grpcClient *grpc.UserClient
|
||
}
|
||
|
||
func YMTUsersHandler(grpcAddr string) http.Handler {
|
||
var api *YMTUsersAPI
|
||
if grpcAddr != "" {
|
||
client, err := grpc.NewUserClient(grpcAddr)
|
||
if err != nil {
|
||
// 如果 gRPC 连接失败，记录错误但继续运行（降级处理）
|
||
// 在实际调用时会返回错误
|
||
} else {
|
||
api = &YMTUsersAPI{grpcClient: client}
|
||
}
|
||
}
|
||
if api == nil {
|
||
// 如果没有 gRPC 客户端，创建一个空的 API（会返回错误）
|
||
api = &YMTUsersAPI{}
|
||
}
|
||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
p := strings.TrimPrefix(r.URL.Path, "/api/ymt/users")
|
||
if r.Method == http.MethodGet && p == "" {
|
||
api.list(w, r)
|
||
return
|
||
}
|
||
w.WriteHeader(http.StatusNotFound)
|
||
})
|
||
}
|
||
|
||
func (a *YMTUsersAPI) list(w http.ResponseWriter, r *http.Request) {
|
||
if a.grpcClient == nil {
|
||
fail(w, r, http.StatusInternalServerError, "gRPC client not initialized")
|
||
return
|
||
}
|
||
|
||
limitStr := r.URL.Query().Get("limit")
|
||
limit := 2000
|
||
if limitStr != "" {
|
||
if n, err := strconv.Atoi(limitStr); err == nil && n > 0 && n <= 10000 {
|
||
limit = n
|
||
}
|
||
}
|
||
|
||
// 忽略 q 参数，始终返回所有数据（不做过滤）
|
||
// 获取搜索关键词（如果有）
|
||
// keyword := r.URL.Query().Get("q")
|
||
keyword := "" // 始终传空字符串，返回所有数据
|
||
|
||
// 创建带超时的 context
|
||
ctx, cancel := context.WithTimeout(r.Context(), 10*time.Second)
|
||
defer cancel()
|
||
|
||
// 调用 gRPC 服务
|
||
resp, err := a.grpcClient.SimpleListAllUser(ctx, keyword)
|
||
if err != nil {
|
||
fail(w, r, http.StatusInternalServerError, fmt.Sprintf("gRPC call failed: %v", err))
|
||
return
|
||
}
|
||
|
||
// 转换响应格式
|
||
out := []map[string]interface{}{}
|
||
for i, user := range resp.GetList() {
|
||
if i >= limit {
|
||
break
|
||
}
|
||
if user == nil {
|
||
continue
|
||
}
|
||
// 构建显示名称：realname（id）或 username（id）
|
||
displayName := user.GetRealname()
|
||
if displayName == "" {
|
||
displayName = user.GetUsername()
|
||
}
|
||
userId := user.GetId()
|
||
if displayName == "" {
|
||
displayName = strconv.FormatInt(int64(userId), 10)
|
||
}
|
||
display := fmt.Sprintf("%s（%d）", displayName, userId)
|
||
|
||
out = append(out, map[string]interface{}{
|
||
"id": userId,
|
||
"name": display,
|
||
})
|
||
}
|
||
|
||
ok(w, r, out)
|
||
}
|