首页 > 技术文章 > GoTcp通信系统

snyn 2022-05-10 14:33 原文

client

package main

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

func main() {

	conn, err := net.Dial("tcp", "127.0.0.1:8888")
	if err != nil {
		fmt.Println("client dial err=", err)
		return
	}

	reader := bufio.NewReader(os.Stdin)
	go func() {
		//创建切片
		buf := make([]byte, 1024)

		//1 等待客户端通过conn发送信息
		//2 如果没有writer发送就一直阻塞在这
		n, err := conn.Read(buf)
		if err != nil {
			fmt.Println("服务器read err=", err) //出错退出
			return
		}
		//3. 显示读取内容到终端

		fmt.Print(string(buf[:n]))



	}()
	for {



		//从终端读取数据(会以文件的形式)
		line, err := reader.ReadString('\n')
		if err != nil {
			fmt.Println("readstring err=", err)
		}
		//如多用户输入exit就退出
		line = strings.Trim(line, "\r\n")
		if line == "exit" {
			fmt.Println("客户端退出")
			return

		}
		//将line发给服务器
		n, err := conn.Write([]byte(line + "\n"))
		if err != nil {
			fmt.Println("conn.Write err=", err)
		}
		fmt.Printf("发送了%d个字节\n",n)
	}
}

server

package main

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

func process(conn net.Conn) {

	//循环接受客户端发送数据,不让协程退出
	defer conn.Close() //关闭conn连接
	go func() {
		for {

			//创建切片
			buf := make([]byte, 1024)

			//1 等待客户端通过conn发送信息
			//2 如果没有writer发送就一直阻塞
			n, err := conn.Read(buf)
			if err != nil {
				fmt.Println("服务器read err=", err) //出错退出
				return
			}
			//3. 显示读取内容到终端
			fmt.Print(string(buf[:n]))
		}
	}()
	for {
		//发送消息给客户端
		reader := bufio.NewReader(os.Stdin)
		//从终端读取数据(会以文件的形式)
		line, err := reader.ReadString('\n')
		if err != nil {
			fmt.Println("readstring err=", err)
		}
		//如多用户输入exit就退出
		line = strings.Trim(line, "\r\n")
		if line == "exit" {
			fmt.Println("客户端退出")
			return

		}
		//将line发给服务器

		_, err = conn.Write([]byte(line + "\n"))
		if err != nil {
			fmt.Println("conn.Write err=", err)
		}
	}


}

func main() {

	fmt.Println("服务器开始监听------")
	listen, err := net.Listen("tcp", "0.0.0.0:8888")
	if err != nil {
		fmt.Println("listen err=", err)
		return
	}

	defer listen.Close() //延时关闭listen

	for {

		fmt.Println("-----等待客户连接----")
		conn, err := listen.Accept()
		if err != nil {
			fmt.Println("Accept err=", err)
		} else {
			fmt.Printf("Accept success con类型%T,客户ip%v\n", conn, conn.RemoteAddr().String())
		}

		//这里要起一个协程
		go process(conn)
	}
}

推荐阅读