42 lines
1.2 KiB
Go
42 lines
1.2 KiB
Go
|
package xrpc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
loggerpb "github.com/xtls/xray-core/app/log/command"
|
||
|
|
handlerpb "github.com/xtls/xray-core/app/proxyman/command"
|
||
|
|
statspb "github.com/xtls/xray-core/app/stats/command"
|
||
|
|
"google.golang.org/grpc"
|
||
|
|
"google.golang.org/grpc/credentials/insecure"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Clients groups the gRPC stubs that the samples rely on.
|
||
|
|
type Clients struct {
|
||
|
|
Connection *grpc.ClientConn
|
||
|
|
Handler handlerpb.HandlerServiceClient
|
||
|
|
Logger loggerpb.LoggerServiceClient
|
||
|
|
Stats statspb.StatsServiceClient
|
||
|
|
}
|
||
|
|
|
||
|
|
// New establishes an insecure (plaintext) connection against a running Xray API endpoint.
|
||
|
|
func New(ctx context.Context, addr string, port uint16, dialOpts ...grpc.DialOption) (*Clients, error) {
|
||
|
|
target := fmt.Sprintf("%s:%d", addr, port)
|
||
|
|
opts := []grpc.DialOption{
|
||
|
|
grpc.WithTransportCredentials(insecure.NewCredentials()),
|
||
|
|
}
|
||
|
|
opts = append(opts, dialOpts...)
|
||
|
|
|
||
|
|
conn, err := grpc.DialContext(ctx, target, opts...)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return &Clients{
|
||
|
|
Connection: conn,
|
||
|
|
Handler: handlerpb.NewHandlerServiceClient(conn),
|
||
|
|
Logger: loggerpb.NewLoggerServiceClient(conn),
|
||
|
|
Stats: statspb.NewStatsServiceClient(conn),
|
||
|
|
}, nil
|
||
|
|
}
|