Method handleUserValidationResult

internal/message/handler.go:263–314

(u *user.User, cost float64)

}

func (h *Handler) handleUserValidationResult(u *user.User, cost float64) error {
	err := h.uv.Validate(u, cost)

	if err != nil {
		telemetry.Incr("bricksllm.message.handler.handle_user_validation_result.validate_error", nil, 1)

		if _, ok := err.(expirationError); ok {
			telemetry.Incr("bricksllm.message.handler.handle_user_validation_result.expiraton_error", nil, 1)

			truePtr := true
			_, err = h.um.UpdateUser(u.Id, &user.UpdateUser{
				Revoked:       &truePtr,
				RevokedReason: key.RevokedReasonExpired,
			})

			if err != nil {
				telemetry.Incr("bricksllm.message.handler.handle_user_validation_result.update_user_error", nil, 1)
				return err
			}

			return nil
		}

		if _, ok := err.(rateLimitError); ok {
			telemetry.Incr("bricksllm.message.handler.handle_user_validation_result.rate_limit_error", nil, 1)

			err = h.uac.Set(u.Id, u.RateLimitUnit)
			if err != nil {
				telemetry.Incr("bricksllm.message.handler.handle_user_validation_result.set_rate_limit_error", nil, 1)
				return err
			}

			return nil
		}

		if _, ok := err.(costLimitError); ok {
			telemetry.Incr("bricksllm.message.handler.handle_user_validation_result.cost_limit_error", nil, 1)

			err = h.uac.Set(u.Id, u.CostLimitInUsdUnit)
			if err != nil {
				telemetry.Incr("bricksllm.message.handler.handle_user_validation_result.set_cost_limit_error", nil, 1)
				return err
			}

			return nil
		}

		return err
	}

	return nil
}

func (h *Handler) HandleEventWithRequestAndResponse(m Message) error {
	e, ok := m.Data.(*event.EventWithRequestAndContent)