-
Notifications
You must be signed in to change notification settings - Fork 343
/
serve.go
95 lines (82 loc) · 2.74 KB
/
serve.go
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
package api
//go:generate swagger generate client -q -A lakefs -f ../../api/swagger.yml -P models.User -t gen
//go:generate swagger generate server -q -A lakefs -f ../../api/swagger.yml -P models.User -t gen --exclude-main
import (
"net/http"
"github.com/go-openapi/errors"
"github.com/go-openapi/loads"
"github.com/go-openapi/runtime"
"github.com/prometheus/client_golang/prometheus/promhttp"
"github.com/treeverse/lakefs/pkg/api/gen/restapi"
"github.com/treeverse/lakefs/pkg/api/gen/restapi/operations"
"github.com/treeverse/lakefs/pkg/auth"
"github.com/treeverse/lakefs/pkg/block"
"github.com/treeverse/lakefs/pkg/catalog"
"github.com/treeverse/lakefs/pkg/cloud"
"github.com/treeverse/lakefs/pkg/db"
"github.com/treeverse/lakefs/pkg/httputil"
"github.com/treeverse/lakefs/pkg/logging"
"github.com/treeverse/lakefs/pkg/stats"
)
const (
RequestIDHeaderName = "X-Request-ID"
LoggerServiceName = "rest_api"
)
func Serve(
catalog catalog.Interface,
authService auth.Service,
blockAdapter block.Adapter,
metadataManager auth.MetadataManager,
migrator db.Migrator,
collector stats.Collector,
cloudMetadataProvider cloud.MetadataProvider,
actions actionsHandler,
logger logging.Logger,
gatewayDomain string,
) http.Handler {
logger.Info("initialize OpenAPI server")
swaggerSpec, _ := loads.Analyzed(restapi.SwaggerJSON, "")
api := operations.NewLakefsAPI(swaggerSpec)
api.Logger = func(msg string, ctx ...interface{}) {
logging.Default().WithField("logger", "swagger").Debugf(msg, ctx)
}
api.BasicAuthAuth = NewBasicAuthHandler(authService)
api.JwtTokenAuth = NewJwtTokenAuthHandler(authService)
api.UseSwaggerUI()
api.ServeError = errors.ServeError
api.JSONConsumer = runtime.JSONConsumer()
api.MultipartformConsumer = runtime.DiscardConsumer
api.BinProducer = runtime.ByteStreamProducer()
api.JSONProducer = runtime.JSONProducer()
// bind our handlers to the server
controller := NewController(
catalog,
authService,
blockAdapter,
metadataManager,
migrator,
collector,
cloudMetadataProvider,
actions,
logger,
)
controller.Configure(api)
apiHandler := api.Serve(func(handler http.Handler) http.Handler {
// build handler for our REST API
return httputil.LoggingMiddleware(
RequestIDHeaderName,
logging.Fields{"service_name": LoggerServiceName},
promhttp.InstrumentHandlerCounter(requestCounter,
MetricsHandler(api.Context(),
NewCookieAPIHandler(handler))))
})
uiHandler := NewUIHandler(authService, gatewayDomain)
mux := http.NewServeMux()
mux.Handle("/_health", httputil.ServeHealth())
mux.Handle("/metrics", promhttp.Handler())
mux.Handle("/_pprof/", httputil.ServePPROF("/_pprof/"))
mux.Handle("/api/", apiHandler)
mux.Handle("/swagger.json", apiHandler)
mux.Handle("/", uiHandler)
return mux
}