Method List

module/ai-key/iml.go:277–364

(ctx context.Context, providerId string, keyword string, page, pageSize int, statuses []string)

}

func (i *imlKeyModule) List(ctx context.Context, providerId string, keyword string, page, pageSize int, statuses []string) ([]*ai_key_dto.Item, int64, error) {
	_, err := i.aiKeyService.DefaultKey(ctx, providerId)
	if err != nil {
		if !errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, 0, fmt.Errorf("get default key failed: %w", err)
		}
		info, err := i.providerService.Get(ctx, providerId)
		if err != nil {
			return nil, 0, fmt.Errorf("provider is unconfigued,id is %s", providerId)
		}
		err = i.aiKeyService.Create(ctx, &ai_key.Create{
			ID:         info.Id,
			Name:       info.Name,
			Config:     info.Config,
			Provider:   info.Id,
			Status:     ai_key_dto.KeyNormal.Int(),
			Priority:   1,
			ExpireTime: 0,
			UseToken:   0,
			Default:    true,
		})
		if err != nil {
			return nil, 0, fmt.Errorf("create default key failed: %w", err)
		}
	}
	w := map[string]interface{}{
		"provider": providerId,
	}
	hasExpired := true
	if len(statuses) > 0 {
		hasExpired = false
		statusConditions := make([]int, 0, len(statuses))
		for _, s := range statuses {
			status := ai_key_dto.KeyStatus(s)
			if status == ai_key_dto.KeyExpired {
				hasExpired = true
			}
			statusConditions = append(statusConditions, status.Int())
		}
		w["status"] = statusConditions
	}
	var list []*ai_key.Key
	var total int64
	if !hasExpired {
		if keyword != "" {
			list, err = i.aiKeyService.Search(ctx, keyword, w, "sort asc")
			if err != nil {
				return nil, 0, err
			}
			if len(list) == 0 {
				return nil, 0, nil
			}
			uuids := utils.SliceToSlice(list, func(key *ai_key.Key) string {
				return key.ID
			})
			w["uuid"] = uuids
		}
		list, total, err = i.aiKeyService.SearchUnExpiredByPage(ctx, w, page, pageSize, "sort asc")