Use standard PORT for host port

This commit is contained in:
Marc Di Luzio 2020-06-13 11:41:24 +01:00
parent 3f879f9501
commit 914eef05c0
5 changed files with 39 additions and 24 deletions

View file

@ -2,10 +2,12 @@ package main
import (
"context"
"fmt"
"log"
"net"
"os"
"os/signal"
"strconv"
"sync"
"syscall"
@ -81,9 +83,13 @@ func (a *accountantServer) GetValue(_ context.Context, in *accounts.DataKey) (*a
// main
func main() {
// Verify the input
var address = os.Getenv("ROVE_ACCOUNTANT_GRPC")
if len(address) == 0 {
log.Fatal("No address set with $ROVE_ACCOUNTANT_GRPC")
var port = os.Getenv("PORT")
if len(port) == 0 {
log.Fatal("Must set $PORT")
}
iport, err := strconv.Atoi(port)
if err != nil {
log.Fatal("$PORT not valid int")
}
persistence.SetPath(data)
@ -95,7 +101,7 @@ func main() {
}
// Set up the RPC server and register
lis, err := net.Listen("tcp", address)
lis, err := net.Listen("tcp", fmt.Sprintf(":%d", iport))
if err != nil {
log.Fatalf("failed to listen: %v", err)
}
@ -114,7 +120,7 @@ func main() {
}()
// Serve the RPC server
log.Printf("Serving accountant on %s\n", address)
log.Printf("Serving accountant on %s\n", port)
if err := grpcServer.Serve(lis); err != nil {
log.Fatalf("failed to serve gRPC: %s", err)
}

View file

@ -6,6 +6,7 @@ import (
"log"
"net/http"
"os"
"strconv"
"github.com/grpc-ecosystem/grpc-gateway/runtime"
"google.golang.org/grpc"
@ -23,13 +24,17 @@ func main() {
log.Fatal("Must set $ROVE_GRPC")
}
var address = os.Getenv("ROVE_HTTP")
if len(address) == 0 {
log.Fatal("Must set $ROVE_HTTP")
var port = os.Getenv("PORT")
if len(port) == 0 {
log.Fatal("Must set $PORT")
}
iport, err := strconv.Atoi(port)
if err != nil {
log.Fatal("$PORT not valid int")
}
// Create a new mux and register it with the gRPC endpoint
fmt.Printf("Hosting reverse-proxy on %s for %s\n", address, endpoint)
fmt.Printf("Hosting reverse-proxy on %d for %s\n", iport, endpoint)
mux := runtime.NewServeMux()
opts := []grpc.DialOption{grpc.WithInsecure()}
if err := rove.RegisterRoveHandlerFromEndpoint(ctx, mux, endpoint, opts); err != nil {
@ -37,7 +42,7 @@ func main() {
}
// Start the HTTP server and proxy calls to gRPC endpoint when needed
if err := http.ListenAndServe(address, mux); err != nil {
if err := http.ListenAndServe(fmt.Sprintf(":%d", iport), mux); err != nil {
log.Fatal(err)
}
}

View file

@ -2,6 +2,7 @@ package main
import (
"flag"
"fmt"
"log"
"os"
"os/signal"
@ -31,11 +32,14 @@ func InnerMain() {
}
// Address to host the server on, automatically selected if empty
var address = os.Getenv("ROVE_GRPC")
if len(address) == 0 {
log.Fatalf("Must set $ROVE_GRPC")
var port = os.Getenv("PORT")
if len(port) == 0 {
log.Fatal("Must set $PORT")
}
iport, err := strconv.Atoi(port)
if err != nil {
log.Fatal("$PORT not valid int")
}
log.Printf("Initialising version %s...\n", version.Version)
// Set the persistence path
@ -53,7 +57,7 @@ func InnerMain() {
// Create the server data
s := internal.NewServer(
internal.OptionAddress(address),
internal.OptionAddress(fmt.Sprintf(":%d", iport)),
internal.OptionPersistentData(),
internal.OptionTick(tickRate))

View file

@ -17,8 +17,6 @@ import (
"google.golang.org/grpc"
)
var USAGE = ""
// Command usage
func Usage() {
fmt.Fprintf(os.Stderr, "Usage: %s COMMAND [OPTIONS]...\n", os.Args[0])
@ -36,6 +34,8 @@ func Usage() {
var home = os.Getenv("HOME")
var filepath = path.Join(home, ".local/share/rove.json")
const gRPCport = 9090
// General usage
var ver = flag.Bool("version", false, "Display version number")
var host = flag.String("host", "", "path to game host server")
@ -94,7 +94,7 @@ func InnerMain(command string) error {
}
// Set up the server
clientConn, err := grpc.Dial(config.Host, grpc.WithInsecure())
clientConn, err := grpc.Dial(fmt.Sprintf("%s:%d", config.Host, gRPCport), grpc.WithInsecure())
if err != nil {
return err
}