From ce4b0fb6fb69ffc61bc4fb21b508a6a23b811d55 Mon Sep 17 00:00:00 2001
From: atanarjuat <atanarjuat@example.org>
Date: Mon, 23 May 2022 09:18:50 +0000
Subject: add kcp

---
 server/main.go | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'server/main.go')

diff --git a/server/main.go b/server/main.go
index 3ade4dc..f8b7d85 100644
--- a/server/main.go
+++ b/server/main.go
@@ -97,8 +97,13 @@ func main() {
 
 	logger.Printf("DEBUG: %v", listenConfig)
 
-	// TODO: pass kcp mode
-	ln, err := listenConfig.Listen(ctx, "tcp", addr)
+	var network string
+	if os.Getenv("KCP") == "1" {
+		network = "kcp"
+	} else {
+		network = "tcp"
+	}
+	ln, err := listenConfig.Listen(ctx, network, addr)
 
 	if err != nil {
 		logger.Fatalf("error binding to %s: %v", addr, err)
-- 
cgit v1.2.3