support the config of GRPC port (#7)

Author: WangXiangUSTC

grpc_server.go

@@ -2,6 +2,7 @@ package kubernetes
 
 import (
 	"context"
+	"fmt"
 	"net"
 	"time"
 
@@ -10,9 +11,9 @@ import (
 )
 
 // CreateGRPCServer ...
-func (k Kubernetes) CreateGRPCServer(port string) error {
+func (k Kubernetes) CreateGRPCServer() error {
 	log.Info("CreateGRPCServer")
-	grpcListener, err := net.Listen("tcp", port)
+	grpcListener, err := net.Listen("tcp", fmt.Sprintf(":%d", k.grpcPort))
 	if err != nil {
 		return err
 	}

kubernetes.go

@@ -59,6 +59,9 @@ type Kubernetes struct {
 	TransferTo       []string
 	Client           typev1.CoreV1Interface
 
+	// grpc port is the port used for request chaos request
+	grpcPort int
+
 	sync.RWMutex
 	chaosMap map[string]*pb.SetDNSChaosRequest
 	// namespace -> pod_name -> pod info

setup.go

@@ -43,8 +43,7 @@ func setup(c *caddy.Controller) error {
 		return plugin.Error(pluginName, err)
 	}
 
-	// TODO: support set port
-	err = k.CreateGRPCServer(":9288")
+	err = k.CreateGRPCServer()
 	if err != nil {
 		return plugin.Error(pluginName, err)
 	}
@@ -280,6 +279,18 @@ func ParseStanza(c *caddy.Controller) (*Kubernetes, error) {
 				continue
 			}
 			return nil, c.ArgErr()
+		case "grpcport":
+			args := c.RemainingArgs()
+			if len(args) == 1 {
+				port, err := strconv.Atoi(args[0])
+				if err != nil {
+					return nil, err
+				}
+				k8s.grpcPort = port
+			} else {
+				// use default port
+				k8s.grpcPort = 9288
+			}
 		default:
 			return nil, c.Errf("unknown property '%s'", c.Val())
 		}