add more options for grpc

listener/grpc/metadata.go

@@ -1,6 +1,8 @@
 package grpc
 
 import (
+	"time"
+
 	mdata "github.com/go-gost/core/metadata"
 	mdutil "github.com/go-gost/core/metadata/util"
 )
@@ -10,24 +12,44 @@ const (
 )
 
 type metadata struct {
-	backlog  int
-	insecure bool
-	path     string
+	backlog                      int
+	insecure                     bool
+	path                         string
+	keepalive                    bool
+	keepaliveMinTime             time.Duration
+	keepaliveTime                time.Duration
+	keepaliveTimeout             time.Duration
+	keepalivePermitWithoutStream bool
+	keepaliveMaxConnectionIdle   time.Duration
 }
 
 func (l *grpcListener) parseMetadata(md mdata.Metadata) (err error) {
-	const (
-		backlog  = "backlog"
-		insecure = "grpcInsecure"
-		path     = "path"
-	)
-
-	l.md.backlog = mdutil.GetInt(md, backlog)
+	l.md.backlog = mdutil.GetInt(md, "grpc.backlog", "backlog")
 	if l.md.backlog <= 0 {
 		l.md.backlog = defaultBacklog
 	}
 
-	l.md.insecure = mdutil.GetBool(md, insecure)
-	l.md.path = mdutil.GetString(md, path)
+	l.md.insecure = mdutil.GetBool(md, "grpc.insecure", "grpcInsecure", "insecure")
+	l.md.path = mdutil.GetString(md, "grpc.path", "path")
+
+	l.md.keepalive = mdutil.GetBool(md, "grpc.keepalive", "keepalive", "keepAlive")
+	if l.md.keepalive {
+		l.md.keepaliveMinTime = mdutil.GetDuration(md, "grpc.keepalive.minTime", "keepalive.minTime")
+		if l.md.keepaliveMinTime <= 0 {
+			l.md.keepaliveMinTime = 30 * time.Second
+		}
+		l.md.keepaliveTime = mdutil.GetDuration(md, "grpc.keepalive.time", "keepalive.time")
+		if l.md.keepaliveTime <= 0 {
+			l.md.keepaliveTime = 60 * time.Second
+		}
+		l.md.keepaliveTimeout = mdutil.GetDuration(md, "grpc.keepalive.timeout", "keepalive.timeout")
+		if l.md.keepaliveTimeout <= 0 {
+			l.md.keepaliveTimeout = 30 * time.Second
+		}
+
+		l.md.keepalivePermitWithoutStream = mdutil.GetBool(md, "grpc.keepalive.permitWithoutStream", "keepalive.permitWithoutStream")
+		l.md.keepaliveMaxConnectionIdle = mdutil.GetDuration(md, "grpc.keepalive.maxConnectionIdle", "keepalive.maxConnectionIdle")
+	}
+
 	return
 }