53 lines
1.0 KiB
Go
53 lines
1.0 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
|
|
"google.golang.org/grpc"
|
|
|
|
flag "github.com/spf13/pflag"
|
|
"github.com/tblyler/sheepmq/sheepmq"
|
|
"github.com/tblyler/sheepmq/shepard"
|
|
)
|
|
|
|
func main() {
|
|
var port uint16
|
|
var listenAddr string
|
|
|
|
flag.StringVarP(&listenAddr, "addr", "h", "", "The address to listen on")
|
|
flag.Uint16VarP(&port, "port", "p", 0, "The port to bind")
|
|
|
|
flag.Parse()
|
|
|
|
listenAddr = fmt.Sprintf("%s:%d", listenAddr, port)
|
|
|
|
listener, err := net.Listen("tcp", listenAddr)
|
|
if err != nil {
|
|
fmt.Printf("Failed to listen on %s: %s", listenAddr, err)
|
|
return
|
|
}
|
|
|
|
defer listener.Close()
|
|
|
|
fmt.Println("Listening on", listener.Addr())
|
|
|
|
grpcServer := grpc.NewServer()
|
|
|
|
sheepmqServer, err := sheepmq.NewServer()
|
|
if err != nil {
|
|
fmt.Println("Failed to open sheepmq server:", err)
|
|
return
|
|
}
|
|
|
|
sheepmqGServer := sheepmq.NewGServer(sheepmqServer)
|
|
if err != nil {
|
|
fmt.Println("Failed to create sheepmq server:", err)
|
|
return
|
|
}
|
|
|
|
shepard.RegisterLoqServer(grpcServer, sheepmqServer)
|
|
|
|
grpcServer.Serve(listener)
|
|
}
|