linka-cloud/grpc-toolkit
2
0
Fork 0
mirror of https://github.com/linka-cloud/grpc.git synced 2025-06-22 01:02:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

interceptors: add ban

Browse Source 
health: set services serving on start and not available on close

Signed-off-by: Adphi <philippe.adrien.nousse@gmail.com>
This commit is contained in:
Adphi
2022-09-27 17:06:18 +02:00
parent 9bf4e691ce
commit c7096975b1
13 changed files with 350 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
example/example.go
View File

@ -22,6 +22,7 @@ import (
"go.linka.cloud/grpc/client"
"go.linka.cloud/grpc/interceptors/auth"
"go.linka.cloud/grpc/interceptors/ban"
"go.linka.cloud/grpc/interceptors/defaulter"
"go.linka.cloud/grpc/interceptors/iface"
metrics2 "go.linka.cloud/grpc/interceptors/metrics"
@ -137,6 +138,7 @@ func run(opts ...service.Option) {
service.WithMiddlewares(httpLogger),
service.WithInterceptors(metrics),
service.WithServerInterceptors(
ban.NewInterceptors(),
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")
@ -167,7 +169,7 @@ func run(opts ...service.Option) {
}
}()
<-ready
s, err := client.New(
copts := []client.Option{
// client.WithName(name),
// client.WithVersion(version),
client.WithAddress("localhost:9991"),
@ -177,14 +179,28 @@ func run(opts ...service.Option) {
logger.From(ctx).WithFields("party", "client", "method", method).Info(req)
return invoker(ctx, method, req, reply, cc, opts...)
}),
client.WithInterceptors(auth.NewBasicAuthClientIntereptors("admin", "admin")),
)
}
s, err := client.New(copts...)
if err != nil {
log.Fatal(err)
}
g := NewGreeterClient(s)
h := grpc_health_v1.NewHealthClient(s)
hres, err := h.Check(ctx, &grpc_health_v1.HealthCheckRequest{})
for i := 0; i < 4; i++ {
_, err := h.Check(ctx, &grpc_health_v1.HealthCheckRequest{})
if err != nil {
log.Error(err)
} else {
log.Fatalf("expected error")
}
}
s, err = client.New(append(copts, client.WithInterceptors(auth.NewBasicAuthClientIntereptors("admin", "admin")))...)
if err != nil {
log.Fatal(err)
}
g = NewGreeterClient(s)
h = grpc_health_v1.NewHealthClient(s)
hres, err := h.Check(ctx, &grpc_health_v1.HealthCheckRequest{Service: "helloworld.Greeter"})
if err != nil {
log.Fatal(err)
}