38 lines
852 B
Go
38 lines
852 B
Go
package admin_agent
|
||
|
||
import (
|
||
"context"
|
||
|
||
"qnc-server/app/main/api/internal/svc"
|
||
|
||
"github.com/Masterminds/squirrel"
|
||
)
|
||
|
||
// batchAgentCodesByIds 按代理主键批量查询 agent_code，用于列表软关联展示。
|
||
func batchAgentCodesByIds(ctx context.Context, svcCtx *svc.ServiceContext, ids []string) map[string]int64 {
|
||
out := make(map[string]int64)
|
||
uniq := make([]string, 0, len(ids))
|
||
seen := make(map[string]struct{}, len(ids))
|
||
for _, id := range ids {
|
||
if id == "" {
|
||
continue
|
||
}
|
||
if _, ok := seen[id]; ok {
|
||
continue
|
||
}
|
||
seen[id] = struct{}{}
|
||
uniq = append(uniq, id)
|
||
}
|
||
if len(uniq) == 0 {
|
||
return out
|
||
}
|
||
agents, err := svcCtx.AgentModel.FindAll(ctx, svcCtx.AgentModel.SelectBuilder().Where(squirrel.Eq{"id": uniq}), "")
|
||
if err != nil {
|
||
return out
|
||
}
|
||
for _, a := range agents {
|
||
out[a.Id] = a.AgentCode
|
||
}
|
||
return out
|
||
}
|