mirror of
https://github.com/linka-cloud/grpc.git
synced 2024-11-05 02:36:24 +00:00
25 lines
613 B
Go
25 lines
613 B
Go
package recovery
|
|
|
|
import (
|
|
grpc_recovery "github.com/grpc-ecosystem/go-grpc-middleware/recovery"
|
|
"google.golang.org/grpc"
|
|
|
|
"go.linka.cloud/grpc-toolkit/interceptors"
|
|
)
|
|
|
|
func NewInterceptors(opts ...grpc_recovery.Option) interceptors.ServerInterceptors {
|
|
return &recovery{opts: opts}
|
|
}
|
|
|
|
type recovery struct {
|
|
opts []grpc_recovery.Option
|
|
}
|
|
|
|
func (i *recovery) UnaryServerInterceptor() grpc.UnaryServerInterceptor {
|
|
return grpc_recovery.UnaryServerInterceptor(i.opts...)
|
|
}
|
|
|
|
func (i *recovery) StreamServerInterceptor() grpc.StreamServerInterceptor {
|
|
return grpc_recovery.StreamServerInterceptor(i.opts...)
|
|
}
|