Spaces:
Running
Running
File size: 2,914 Bytes
b110593 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
// _ _
// __ _____ __ ___ ___ __ _| |_ ___
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
// \ V V / __/ (_| |\ V /| | (_| | || __/
// \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
//
// Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
//
// CONTACT: [email protected]
//
package rest
import (
middleware "github.com/go-openapi/runtime/middleware"
"github.com/go-openapi/strfmt"
"github.com/sirupsen/logrus"
"github.com/weaviate/weaviate/adapters/handlers/rest/operations"
"github.com/weaviate/weaviate/adapters/handlers/rest/operations/classifications"
"github.com/weaviate/weaviate/entities/models"
"github.com/weaviate/weaviate/usecases/classification"
"github.com/weaviate/weaviate/usecases/monitoring"
)
func setupClassificationHandlers(api *operations.WeaviateAPI,
classifier *classification.Classifier, metrics *monitoring.PrometheusMetrics, logger logrus.FieldLogger,
) {
metricRequestsTotal := newClassificationRequestsTotal(metrics, logger)
api.ClassificationsClassificationsGetHandler = classifications.ClassificationsGetHandlerFunc(
func(params classifications.ClassificationsGetParams, principal *models.Principal) middleware.Responder {
res, err := classifier.Get(params.HTTPRequest.Context(), principal, strfmt.UUID(params.ID))
if err != nil {
metricRequestsTotal.logError("", err)
return classifications.NewClassificationsGetInternalServerError().WithPayload(errPayloadFromSingleErr(err))
}
if res == nil {
metricRequestsTotal.logUserError("")
return classifications.NewClassificationsGetNotFound()
}
metricRequestsTotal.logOk("")
return classifications.NewClassificationsGetOK().WithPayload(res)
},
)
api.ClassificationsClassificationsPostHandler = classifications.ClassificationsPostHandlerFunc(
func(params classifications.ClassificationsPostParams, principal *models.Principal) middleware.Responder {
res, err := classifier.Schedule(params.HTTPRequest.Context(), principal, *params.Params)
if err != nil {
metricRequestsTotal.logUserError("")
return classifications.NewClassificationsPostBadRequest().WithPayload(errPayloadFromSingleErr(err))
}
metricRequestsTotal.logOk("")
return classifications.NewClassificationsPostCreated().WithPayload(res)
},
)
}
type classificationRequestsTotal struct {
*restApiRequestsTotalImpl
}
func newClassificationRequestsTotal(metrics *monitoring.PrometheusMetrics, logger logrus.FieldLogger) restApiRequestsTotal {
return &classificationRequestsTotal{
restApiRequestsTotalImpl: &restApiRequestsTotalImpl{newRequestsTotalMetric(metrics, "rest"), "rest", "classification", logger},
}
}
func (e *classificationRequestsTotal) logError(className string, err error) {
switch err.(type) {
default:
e.logServerError(className, err)
}
}
|