mirror of
https://github.com/linka-cloud/grpc.git
synced 2024-11-22 02:46:25 +00:00
96 lines
3.2 KiB
Go
96 lines
3.2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"time"
|
|
|
|
logging2 "github.com/grpc-ecosystem/go-grpc-middleware/v2/interceptors/logging"
|
|
"go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp"
|
|
"go.opentelemetry.io/otel/trace"
|
|
|
|
"go.linka.cloud/grpc-tookit/example/pb"
|
|
"go.linka.cloud/grpc-toolkit/interceptors/auth"
|
|
"go.linka.cloud/grpc-toolkit/interceptors/ban"
|
|
"go.linka.cloud/grpc-toolkit/interceptors/defaulter"
|
|
"go.linka.cloud/grpc-toolkit/interceptors/iface"
|
|
"go.linka.cloud/grpc-toolkit/interceptors/logging"
|
|
metrics2 "go.linka.cloud/grpc-toolkit/interceptors/metrics"
|
|
"go.linka.cloud/grpc-toolkit/interceptors/tracing"
|
|
validation2 "go.linka.cloud/grpc-toolkit/interceptors/validation"
|
|
"go.linka.cloud/grpc-toolkit/logger"
|
|
"go.linka.cloud/grpc-toolkit/service"
|
|
)
|
|
|
|
func newService(ctx context.Context, opts ...service.Option) (service.Service, error) {
|
|
log := logger.C(ctx)
|
|
metrics := metrics2.NewInterceptors(metrics2.WithExemplarFromContext(metrics2.DefaultExemplarFromCtx))
|
|
|
|
address := "0.0.0.0:9991"
|
|
|
|
var svc service.Service
|
|
opts = append(opts,
|
|
service.WithContext(ctx),
|
|
service.WithAddress(address),
|
|
// service.WithRegistry(mdns.NewRegistry()),
|
|
service.WithReflection(true),
|
|
service.WithoutCmux(),
|
|
service.WithGateway(pb.RegisterGreeterHandler),
|
|
service.WithGatewayPrefix("/rest"),
|
|
service.WithGRPCWeb(true),
|
|
service.WithGRPCWebPrefix("/grpc"),
|
|
service.WithMiddlewares(otelhttp.NewMiddleware("hello"), httpLogger),
|
|
service.WithInterceptors(
|
|
tracing.NewInterceptors(),
|
|
metrics,
|
|
logging.New(ctx, logging2.WithFieldsFromContext(func(ctx context.Context) logging2.Fields {
|
|
if span := trace.SpanContextFromContext(ctx); span.IsSampled() {
|
|
return logging2.Fields{"traceID", span.TraceID().String()}
|
|
}
|
|
return nil
|
|
})),
|
|
),
|
|
service.WithServerInterceptors(
|
|
ban.NewInterceptors(ban.WithDefaultJailDuration(time.Second), ban.WithDefaultCallback(func(action ban.Action, actor string, rule *ban.Rule) error {
|
|
log.WithFields("action", action, "actor", actor, "rule", rule.Name).Info("ban callback")
|
|
return nil
|
|
})),
|
|
auth.NewServerInterceptors(auth.WithBasicValidators(func(ctx context.Context, user, password string) (context.Context, error) {
|
|
if !auth.Equals(user, "admin") || !auth.Equals(password, "admin") {
|
|
return ctx, fmt.Errorf("invalid user or password")
|
|
}
|
|
log.Infof("request authenticated")
|
|
return ctx, nil
|
|
})),
|
|
),
|
|
service.WithInterceptors(
|
|
defaulter.NewInterceptors(),
|
|
validation2.NewInterceptors(true),
|
|
),
|
|
// enable server interface interceptor
|
|
service.WithServerInterceptors(iface.New()),
|
|
)
|
|
svc, err := service.New(opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
pb.RegisterGreeterServer(svc, &GreeterHandler{})
|
|
metrics.Register(svc)
|
|
return svc, nil
|
|
}
|
|
|
|
func httpLogger(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
|
|
start := time.Now()
|
|
log := logger.From(request.Context()).WithFields(
|
|
"method", request.Method,
|
|
"host", request.Host,
|
|
"path", request.URL.Path,
|
|
"remoteAddress", request.RemoteAddr,
|
|
)
|
|
next.ServeHTTP(writer, request)
|
|
log.WithField("duration", time.Since(start)).Info()
|
|
})
|
|
}
|