From 244be863048ec10cec9f9c3a25c90ec6bdaf28ba Mon Sep 17 00:00:00 2001
From: lyyyuna <lyyyuna@outlook.com>
Date: Thu, 24 Jun 2021 15:33:49 +0800
Subject: [PATCH] goc server, non-block channel -> block

---
 pkg/build/tmpfolder.go    | 2 +-
 pkg/server/server.go      | 2 +-
 pkg/server/watchstream.go | 6 +-----
 3 files changed, 3 insertions(+), 7 deletions(-)

diff --git a/pkg/build/tmpfolder.go b/pkg/build/tmpfolder.go
index d81c415..1522835 100644
--- a/pkg/build/tmpfolder.go
+++ b/pkg/build/tmpfolder.go
@@ -69,7 +69,7 @@ func skipCopy(src string, info os.FileInfo) (bool, error) {
 func (b *Build) clean() {
 	if config.GocConfig.Debug != true {
 		if err := os.RemoveAll(config.GocConfig.TmpModProjectDir); err != nil {
-			log.Fatalf("fail to delete the temporary project: %v", config.GocConfig.TmpModProjectDir)
+			log.Fatalf("fail to delete the temporary project: %v", err)
 		}
 		log.Donef("delete the temporary project")
 	} else {
diff --git a/pkg/server/server.go b/pkg/server/server.go
index b43cfa0..cf796b1 100644
--- a/pkg/server/server.go
+++ b/pkg/server/server.go
@@ -55,7 +55,7 @@ func RunGocServerUntilExit(host string) {
 				return true
 			},
 		},
-		watchCh: make(chan []byte),
+		watchCh: make(chan []byte, 4096),
 	}
 
 	r := gin.Default()
diff --git a/pkg/server/watchstream.go b/pkg/server/watchstream.go
index 7d8a58b..02af6c7 100644
--- a/pkg/server/watchstream.go
+++ b/pkg/server/watchstream.go
@@ -74,11 +74,7 @@ func (gs *gocServer) serveWatchInternalStream(c *gin.Context) {
 			break
 		}
 		if mt == websocket.TextMessage {
-			// 非阻塞写
-			select {
-			case gs.watchCh <- message:
-			default:
-			}
+			gs.watchCh <- message
 		}
 	}
 }