Update

Author: devlights

examples/singleapp/socket/tcp_basic/Taskfile.yml

@@ -0,0 +1,26 @@
+version: '3'
+
+tasks:
+  build:
+    cmds:
+      - (cd client; go build)
+      - (cd server; go build)
+  run:
+    cmds:
+      - task: build
+      - task: start-server
+      - task: run-client
+  start-server:
+    cmds:
+      - (cd server; ./server)
+  run-client:
+    cmds:
+      - (cd client; ./client)
+  kill-server:
+    cmds:
+      - pkill server
+  clean:
+    cmds:
+      - (cd client; go clean)
+      - (cd server; go clean)
+      - find . -type f -name "*.PID" -exec rm -rf {} \;

examples/singleapp/socket/tcp_basic/client/main.go

@@ -11,6 +11,7 @@
 package main
 
 import (
+	"bytes"
 	"errors"
 	"io"
 	"log"
@@ -54,29 +55,56 @@ func main() {
 
 	_, err = conn.Write(message)
 	if err != nil {
-		appLog.Printf("error at conn.Write (%v)", err)
+		errLog.Printf("error at conn.Write (%v)", err)
+		return
+	}
+
+	err = conn.CloseWrite()
+	if err != nil {
+		errLog.Printf("error at conn.Write (%v)", err)
 		return
 	}
 
 	//
 	// 受信
 	//
 	var (
-		buf       = make([]byte, 4096)
+		buf       = new(bytes.Buffer)
 		bytesRead int
 	)
 
-	bytesRead, err = conn.Read(buf)
-	switch {
-	case err != nil && !errors.Is(err, io.EOF):
-		appLog.Printf("error at conn.Read (%v)", err)
-		return
-	case bytesRead == 0:
-		appLog.Printf("closed by remote")
-		return
-	default:
-		appLog.Printf("%d bytes recv", bytesRead)
+	for {
+		var (
+			chunk     = make([]byte, 1)
+			chunkSize int
+		)
+
+		chunkSize, err = conn.Read(chunk)
+		appLog.Printf("chunk recv: %dbyte(s)\terr:%v", chunkSize, err)
+
+		if err != nil {
+			if errors.Is(err, io.EOF) {
+				bytesRead += chunkSize
+				buf.Write(chunk)
+				break
+			}
+
+			errLog.Printf("error at conn.Read (%v)", err)
+			return
+		}
+
+		if chunkSize == 0 {
+			appLog.Printf("closed by remote")
+			return
+		}
+
+		bytesRead += chunkSize
+		buf.Write(chunk)
 	}
+	appLog.Printf("%d bytes recv", bytesRead)
 
-	appLog.Printf("recv: %s\n", buf[:bytesRead])
+	var (
+		data = buf.Bytes()[:bytesRead]
+	)
+	appLog.Printf("recv: %s\n", data)
 }

examples/singleapp/socket/tcp_basic/server/main.go

@@ -91,36 +91,63 @@ func main() {
 			// 受信
 			//
 			var (
-				chunk     = make([]byte, 4096)
+				buf       = new(bytes.Buffer)
 				bytesRead int
 			)
 
-			bytesRead, err = conn.Read(chunk)
-			switch {
-			case err != nil && !errors.Is(err, io.EOF):
-				appLog.Printf("error at conn.Read (%v)", err)
-				return
-			case bytesRead == 0:
-				appLog.Printf("closed by remote")
-				return
-			default:
-				appLog.Printf("%d bytes recv", bytesRead)
+			for {
+				var (
+					chunk     = make([]byte, 1)
+					chunkSize int
+				)
+
+				chunkSize, err = conn.Read(chunk)
+				appLog.Printf("chunk recv: %dbyte(s)\terr:%v", chunkSize, err)
+
+				if err != nil {
+					if errors.Is(err, io.EOF) {
+						bytesRead += chunkSize
+						buf.Write(chunk)
+						break
+					}
+
+					appLog.Printf("error at conn.Read (%v)", err)
+					return
+				}
+
+				if chunkSize == 0 {
+					appLog.Printf("closed by remote")
+					return
+				}
+
+				bytesRead += chunkSize
+				buf.Write(chunk)
 			}
+			appLog.Printf("%d bytes recv", bytesRead)
 
-			appLog.Printf("recv: %s\n", chunk[:bytesRead])
+			var (
+				data = buf.Bytes()[:bytesRead]
+			)
+			appLog.Printf("recv: %s\n", data)
 
 			//
 			// 送信
 			//
 			var (
-				message = bytes.ToUpper(chunk[:bytesRead])
+				message = bytes.ToUpper(data)
 			)
 
 			_, err = conn.Write(message)
 			if err != nil {
 				errLog.Printf("error at client writeto (%v)", err)
 				return
 			}
+
+			err = conn.CloseWrite()
+			if err != nil {
+				errLog.Printf("error at conn closewrite (%v)", err)
+				return
+			}
 		}()
 	}
 }