From 20098751ee3ec112e12990b96b03fb6c244774f0 Mon Sep 17 00:00:00 2001
From: suyl <770236076@qq.com>
Date: Wed, 16 Jun 2021 10:06:58 +0800
Subject: [PATCH] aa

---
 globals/globals.go |  2 +-
 main.go            | 30 ++++++++++++++++--------------
 2 files changed, 17 insertions(+), 15 deletions(-)

diff --git a/globals/globals.go b/globals/globals.go
index aaf8daaf7..60273c8bd 100644
--- a/globals/globals.go
+++ b/globals/globals.go
@@ -145,5 +145,5 @@ func IsMainProductEnv() bool {
 }
 
 func IsProductEnv() bool {
-	return IsMainProductEnv()
+	return beego.BConfig.RunMode == "print"
 }
diff --git a/main.go b/main.go
index ff6340a69..6b9f4d49a 100644
--- a/main.go
+++ b/main.go
@@ -146,20 +146,22 @@ func main() {
 				http.ListenAndServe("0.0.0.0:6060", nil)
 			}
 		}()
-		//l, err := net.Listen("tcp", ":8000")
-		//if err != nil {
-		//	fmt.Println("listen error:", err)
-		//	return
-		//}
-		//for {
-		//	c, err := l.Accept()
-		//	if err != nil {
-		//		fmt.Println("accept error:", err)
-		//		break
-		//	}
-		//	globals.SugarLogger.Debugf("tcp socket:", utils.Format4Output(c, true))
-		//	go handleConn(c)
-		//}
+		if globals.IsProductEnv() {
+			l, err := net.Listen("tcp", ":8000")
+			if err != nil {
+				fmt.Println("listen error:", err)
+				return
+			}
+			for {
+				c, err := l.Accept()
+				if err != nil {
+					fmt.Println("accept error:", err)
+					break
+				}
+				globals.SugarLogger.Debugf("tcp socket:", utils.Format4Output(c, true))
+				go handleConn(c)
+			}
+		}
 		beego.Run()
 	}
 }