socket-go

go使用socket通过tcp流通信

1. buf缓冲区有多大，读多大，读到没为止
2. buf缓冲区小了会多次读取
3. TCP存在粘包问题，可以通过应用层解决（HTTP/自定义协议）

示例代码

server端：

package main

import (
	"fmt"
	"log"
	"net"
)

/*
	socket变成是对tcp通讯过程的封装，unix server端网络编程过程为：
	Server -> Bind -> Listen -> Accept，
	go中直接使用Listen + Accept
*/

func main() {
	ServerSocket()
}

func ServerSocket() {
	server, err := net.Listen("tcp", ":8087")

	if err != nil {
		fmt.Println("开启socket服务失败")
	} else {
		fmt.Println("正在开启Server ...")
	}

	for {
		// 接受连接，会阻塞
		conn, err := server.Accept()
		if err != nil {
			fmt.Println("连接出错")
		} else {
			// 接受来自客户端的连接请求，一个连接建立一个conn，服务器可能耗尽资源
			go connHandler(conn)
		}
	}
}

func connHandler(c net.Conn) {
	if c == nil {
		log.Panic("无效的socket连接")
	}

	buf := make([]byte, 1024)
	for {
		cnt, err := c.Read(buf)
		if cnt == 0 || err != nil {
			c.Close()
			break
		}

		fCommand := string(buf[0:cnt])

		fmt.Printf("客户端 %s 传输 -> %s\n", c.RemoteAddr(), fCommand)

		switch fCommand {
		case "ping":
			c.Write([]byte("pong\n"))
		case "hello":
			c.Write([]byte("world\n"))
		default:
			c.Write([]byte(fCommand + "\n"))
		}
	}
}

client端：

package main

import (
	"bufio"
	"fmt"
	"net"
	"os"
	"strings"
)

func main() {
	ClientSocket()
}

func ClientSocket() {
	conn, err := net.Dial("tcp", "127.0.0.1:8087")
	if err != nil {
		fmt.Println("客户端建立连接失败")
		return
	}
	cConnHandler(conn)
}

func cConnHandler(c net.Conn) {
	reader := bufio.NewReader(os.Stdin)
	buf := make([]byte, 1024)

	fmt.Println("请输出请求数据...")

	for {
		input, _ := reader.ReadString('\n')
		input = strings.TrimSpace(input)
		c.Write([]byte(input))
		cnt, err := c.Read(buf)

		if err != nil {
			fmt.Printf("读取数据失败 %s \n", err.Error())
			continue
		}

		fmt.Println("服务端回复:" + string(buf[0:cnt]))
	}
}