【Go】简单版并发服务器

TCP 的C/S架构

Server

package main

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

func dealConn(conn net.Conn) {
	defer conn.Close() //此函数结束时，关闭连接套接字
	//conn.RemoteAddr().String()：连接客服端的网络地址
	ipAddr := conn.RemoteAddr().String()
	fmt.Println(ipAddr, "连接成功")
	buf := make([]byte, 1024) //缓冲区，用于接收客户端发送的数据
	for {
		//阻塞等待用户发送的数据
		n, err := conn.Read(buf) //n 代码接收数据的长度
		if err != nil {
			fmt.Println(err)
			return
		}
		//切片截取，只截取有效数据
		result := buf[:n]
		fmt.Printf("接收到数据来自[%s]==>[%d]:%s\n", ipAddr, n,
			string(result))
		if "exit" == string(result) { //如果对方发送"exit"，退出此链接
			fmt.Println(ipAddr, "退出连接")
			return
		}
		//把接收到的数据转换为大写，再给客户端发送
		conn.Write([]byte(strings.ToUpper(string(result))))
	}
}

func main() {
	//监听
	listenner, err := net.Listen("tcp", "127.0.0.1:8000")
	if err != nil {
		log.Fatal(err) //log.Fatal()会产生 panic
	}
	defer listenner.Close()

	for {
		conn, err := listenner.Accept() //阻塞等待客户端连接
		if err != nil {
			log.Println(err)
			continue
		}
		go dealConn(conn)
	}
}

client

package main

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

func main() {
	//客户端主动连接服务器
	conn, err := net.Dial("tcp", "127.0.0.1:8000")
	if err != nil {
		log.Fatal(err) //log.Fatal()会产生 panic
		return
	}
	defer conn.Close()

	//发送数据
	buf := make([]byte, 1024) //缓冲区
	for {
		fmt.Printf("请输入发送的内容：")
		//fmt.Scan(&buf)
		scanner := bufio.NewScanner(os.Stdin)
		scanner.Scan()
		buf = []byte(scanner.Text())
		fmt.Printf("发送的内容：%s\n", string(buf))
		//发送数据
		conn.Write(buf)
		//阻塞等待服务器回复的数据
		n, err := conn.Read(buf) //n 代码接收数据的长度
		if err != nil {
			fmt.Println(err)
			return
		}
		//切片截取，只截取有效数据
		result := buf[:n]
		fmt.Printf("接收到数据[%d]:%s\n", n, string(result))
	}
}