Method UpdateKeyStatus

module/ai-key/iml.go:366–418

(ctx context.Context, providerId string, id string, enable bool)

}

func (i *imlKeyModule) UpdateKeyStatus(ctx context.Context, providerId string, id string, enable bool) error {
	_, err := i.providerService.Get(ctx, providerId)
	if err != nil {
		return fmt.Errorf("provider not found: %w", err)
	}
	info, err := i.aiKeyService.Get(ctx, id)
	if err != nil {
		return fmt.Errorf("key not found: %w", err)
	}
	return i.transaction.Transaction(ctx, func(ctx context.Context) error {
		if !enable {
			status := ai_key_dto.KeyDisable.Int()
			err = i.aiKeyService.Save(ctx, id, &ai_key.Edit{
				Status: &status,
			})
			if err != nil {
				return err
			}
			releases := []*gateway.DynamicRelease{{
				BasicItem: &gateway.BasicItem{
					ID:       fmt.Sprintf("%s-%s", providerId, id),
					Resource: "ai-key",
				},
				Attr: nil,
			}}
			return i.syncGateway(ctx, cluster.DefaultClusterID, releases, false)
		}
		if info.Status == ai_key_dto.KeyDisable.Int() || info.Status == ai_key_dto.KeyExceed.Int() {
			// 超额 或 停用状态，可启用
			if info.ExpireTime > 0 && time.Unix(int64(info.ExpireTime), 0).Before(time.Now()) {
				// 如果过期时间未更改，且已过期，则设置为过期状态
				status := ai_key_dto.KeyExpired.Int()
				return i.aiKeyService.Save(ctx, id, &ai_key.Edit{
					Status: &status,
				})
			}
			status := ai_key_dto.KeyNormal.Int()
			err = i.aiKeyService.Save(ctx, id, &ai_key.Edit{
				Status: &status,
			})
			if err != nil {
				return err
			}
			info, err = i.aiKeyService.Get(ctx, id)
			if err != nil {
				return err
			}
			releases := []*gateway.DynamicRelease{newKey(info)}
			return i.syncGateway(ctx, cluster.DefaultClusterID, releases, true)
		}
		return nil
	})
}

func (i *imlKeyModule) Sort(ctx context.Context, providerId string, input *ai_key_dto.Sort) error {
	_, err := i.providerService.Get(ctx, providerId)