41 lines
798 B
Go
41 lines
798 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"net"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"github.com/hashicorp/nomad/demo/grpc-checks/example"
|
||
|
|
"google.golang.org/grpc"
|
||
|
|
ghc "google.golang.org/grpc/health/grpc_health_v1"
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
|
||
|
|
port := os.Getenv("GRPC_HC_PORT")
|
||
|
|
if port == "" {
|
||
|
|
port = "3333"
|
||
|
|
}
|
||
|
|
address := fmt.Sprintf(":%s", port)
|
||
|
|
|
||
|
|
log.Printf("creating tcp listener on %s", address)
|
||
|
|
listener, err := net.Listen("tcp", address)
|
||
|
|
if err != nil {
|
||
|
|
log.Printf("unable to create listener: %v", err)
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
|
||
|
|
log.Printf("creating grpc server")
|
||
|
|
grpcServer := grpc.NewServer()
|
||
|
|
|
||
|
|
log.Printf("registering health server")
|
||
|
|
ghc.RegisterHealthServer(grpcServer, example.New())
|
||
|
|
|
||
|
|
log.Printf("listening ...")
|
||
|
|
if err := grpcServer.Serve(listener); err != nil {
|
||
|
|
log.Printf("unable to listen: %v", err)
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
}
|