2021-01-10 15:31:37 +08:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"flag"
|
|
|
|
"net/http"
|
|
|
|
|
|
|
|
"github.com/golang/glog"
|
|
|
|
"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
|
|
|
|
"google.golang.org/grpc"
|
|
|
|
|
2021-01-16 16:28:32 +08:00
|
|
|
gw "github.com/esinio/geco/gen/proto/echo/v1" // Update
|
2021-01-10 15:31:37 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
|
|
|
// command-line options:
|
|
|
|
// gRPC server endpoint
|
|
|
|
grpcServerEndpoint = flag.String("grpc-server-endpoint", "localhost:9090", "gRPC server endpoint")
|
|
|
|
)
|
|
|
|
|
|
|
|
func run() error {
|
|
|
|
ctx := context.Background()
|
|
|
|
ctx, cancel := context.WithCancel(ctx)
|
|
|
|
defer cancel()
|
|
|
|
|
|
|
|
// Register gRPC server endpoint
|
|
|
|
// Note: Make sure the gRPC server is running properly and accessible
|
|
|
|
mux := runtime.NewServeMux()
|
|
|
|
opts := []grpc.DialOption{grpc.WithInsecure()}
|
|
|
|
err := gw.RegisterEchoServiceHandlerFromEndpoint(ctx, mux, *grpcServerEndpoint, opts)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
// Start HTTP server (and proxy calls to gRPC server endpoint)
|
|
|
|
return http.ListenAndServe(":8081", mux)
|
|
|
|
}
|
|
|
|
|
|
|
|
func main() {
|
|
|
|
flag.Parse()
|
|
|
|
defer glog.Flush()
|
|
|
|
|
|
|
|
if err := run(); err != nil {
|
|
|
|
glog.Fatal(err)
|
|
|
|
}
|
|
|
|
}
|