-
Notifications
You must be signed in to change notification settings - Fork 24
/
provide.go
74 lines (61 loc) · 2.31 KB
/
provide.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
package check
import (
"fmt"
"github.com/prometheus/client_golang/prometheus"
"go.uber.org/fx"
"google.golang.org/grpc"
"google.golang.org/grpc/health"
"google.golang.org/grpc/health/grpc_health_v1"
flowcontrolv1 "github.com/fluxninja/aperture/api/gen/proto/go/aperture/flowcontrol/check/v1"
"github.com/fluxninja/aperture/pkg/log"
"github.com/fluxninja/aperture/pkg/policies/flowcontrol/iface"
"github.com/fluxninja/aperture/pkg/policies/flowcontrol/servicegetter"
)
// Module is a set of default providers for flowcontrol components
//
// Note that the handler needs to be Registered for flowcontrol to be available
// externally.
func Module() fx.Option {
return fx.Options(
fx.Provide(
ProvideMetrics,
ProvideHandler,
),
fx.Invoke(Register),
)
}
// ConstructorIn holds parameters for ProvideHandler.
type ConstructorIn struct {
fx.In
ServiceGetter servicegetter.ServiceGetter
Metrics Metrics
EngineAPI iface.Engine
}
// ProvideHandler provides a Flow Control Handler.
func ProvideHandler(
in ConstructorIn,
) (flowcontrolv1.FlowControlServiceServer, HandlerWithValues, error) {
h := NewHandler(in.ServiceGetter, in.Metrics, in.EngineAPI)
// Note: Returning the same handler twice as different interfaces – once as
// a handler to be registered on grpc server and once for consumption by
// authz
return h, h, nil
}
// ProvideDummyHandler provides an empty Flow Control Handler.
var ProvideDummyHandler = fx.Annotate(NewHandler, fx.As(new(HandlerWithValues)))
// ProvideMetrics provides flowcontrol metrics that hook to prometheus registry.
func ProvideMetrics(promRegistry *prometheus.Registry) (Metrics, error) {
metrics, err := NewPrometheusMetrics(promRegistry)
if err != nil {
return nil, fmt.Errorf("failed creating Prometheus collector: %v", err)
}
return metrics, nil
}
// ProvideNopMetrics provides disabled flowcontrol metrics.
func ProvideNopMetrics() Metrics { return NopMetrics{} }
// Register registers flowcontrol service on a grpc server.
func Register(server *grpc.Server, handler flowcontrolv1.FlowControlServiceServer, healthsrv *health.Server) {
flowcontrolv1.RegisterFlowControlServiceServer(server, handler)
healthsrv.SetServingStatus("aperture.flowcontrol.v1.FlowControlService", grpc_health_v1.HealthCheckResponse_SERVING)
log.Info().Msg("flowcontrol handler registered")
}