mirror of
https://github.com/linka-cloud/grpc.git
synced 2024-11-05 02:36:24 +00:00
27 lines
656 B
Go
27 lines
656 B
Go
package auth
|
|
|
|
import (
|
|
"context"
|
|
|
|
grpc_auth "github.com/grpc-ecosystem/go-grpc-middleware/auth"
|
|
|
|
"go.linka.cloud/grpc/interceptors"
|
|
"go.linka.cloud/grpc/interceptors/metadata"
|
|
)
|
|
|
|
type TokenValidator func(ctx context.Context, token string) (context.Context, error)
|
|
|
|
func makeTokenAuthFunc(v TokenValidator) grpc_auth.AuthFunc {
|
|
return func(ctx context.Context) (context.Context, error) {
|
|
a, err := grpc_auth.AuthFromMD(ctx, "bearer")
|
|
if err != nil {
|
|
return ctx, err
|
|
}
|
|
return v(ctx, a)
|
|
}
|
|
}
|
|
|
|
func NewBearerClientInterceptors(token string) interceptors.ClientInterceptors {
|
|
return metadata.NewInterceptors("authorization", "bearer "+token)
|
|
}
|