mirror of
https://github.com/SagerNet/sing-box.git
synced 2024-12-05 04:46:40 +00:00
52 lines
1.3 KiB
Go
52 lines
1.3 KiB
Go
|
package v2raygrpc
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
|
||
|
"google.golang.org/grpc"
|
||
|
)
|
||
|
|
||
|
type GunService interface {
|
||
|
Context() context.Context
|
||
|
Send(*Hunk) error
|
||
|
Recv() (*Hunk, error)
|
||
|
}
|
||
|
|
||
|
func ServerDesc(name string) grpc.ServiceDesc {
|
||
|
return grpc.ServiceDesc{
|
||
|
ServiceName: name,
|
||
|
HandlerType: (*GunServiceServer)(nil),
|
||
|
Methods: []grpc.MethodDesc{},
|
||
|
Streams: []grpc.StreamDesc{
|
||
|
{
|
||
|
StreamName: "Tun",
|
||
|
Handler: _GunService_Tun_Handler,
|
||
|
ServerStreams: true,
|
||
|
ClientStreams: true,
|
||
|
},
|
||
|
},
|
||
|
Metadata: "gun.proto",
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (c *gunServiceClient) TunCustomName(ctx context.Context, name string, opts ...grpc.CallOption) (GunService_TunClient, error) {
|
||
|
stream, err := c.cc.NewStream(ctx, &ServerDesc(name).Streams[0], "/"+name+"/Tun", opts...)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
x := &gunServiceTunClient{stream}
|
||
|
return x, nil
|
||
|
}
|
||
|
|
||
|
var _ GunServiceCustomNameClient = (*gunServiceClient)(nil)
|
||
|
|
||
|
type GunServiceCustomNameClient interface {
|
||
|
TunCustomName(ctx context.Context, name string, opts ...grpc.CallOption) (GunService_TunClient, error)
|
||
|
Tun(ctx context.Context, opts ...grpc.CallOption) (GunService_TunClient, error)
|
||
|
}
|
||
|
|
||
|
func RegisterGunServiceCustomNameServer(s *grpc.Server, srv GunServiceServer, name string) {
|
||
|
desc := ServerDesc(name)
|
||
|
s.RegisterService(&desc, srv)
|
||
|
}
|