diff --git a/main.go b/main.go index 59a5857..80166ba 100644 --- a/main.go +++ b/main.go @@ -14,6 +14,7 @@ import ( type server struct { pb.UnimplementedControlServiceServer + wgClient *wgctrl.Client } func (s *server) Ping(ctx context.Context, req *pb.PingRequest) (*pb.PingResponse, error) { @@ -30,7 +31,8 @@ func main() { } grpcServer := grpc.NewServer() - pb.RegisterControlServiceServer(grpcServer, &server{}) + srv := &server{wgClient: client} + pb.RegisterControlServiceServer(grpcServer, srv) log.Println("gRPC server running on :50051") if err := grpcServer.Serve(lis); err != nil {