47 lines
974 B
Go
47 lines
974 B
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net"
|
|
"os"
|
|
|
|
"github.com/barakmich/go_raylet/ray_rpc"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/credentials/insecure"
|
|
"google.golang.org/grpc/grpclog"
|
|
)
|
|
|
|
var (
|
|
gRPCPort = flag.Int("grpc-port", 50000, "The gRPC server port")
|
|
httpPort = flag.Int("http-port", 8080, "The HTTP server port")
|
|
)
|
|
|
|
var log grpclog.LoggerV2
|
|
|
|
func init() {
|
|
log = grpclog.NewLoggerV2(os.Stdout, ioutil.Discard, ioutil.Discard)
|
|
grpclog.SetLoggerV2(log)
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
addr := fmt.Sprintf("0.0.0.0:%d", *gRPCPort)
|
|
lis, err := net.Listen("tcp", addr)
|
|
if err != nil {
|
|
log.Fatalln("Failed to listen:", err)
|
|
}
|
|
s := grpc.NewServer(
|
|
grpc.Creds(insecure.NewCredentials()),
|
|
)
|
|
server := NewMemRaylet()
|
|
ray_rpc.RegisterRayletDriverServer(s, server)
|
|
ray_rpc.RegisterRayletWorkerConnectionServer(s, server)
|
|
|
|
// Serve gRPC Server
|
|
log.Info("Serving gRPC on https://", addr)
|
|
go func() {
|
|
log.Fatal(s.Serve(lis))
|
|
}()
|
|
}
|