简单实现golang的socket编程(TCP)

 首先创建一个server.go作为服务端:

import (
	"bufio"
	"fmt"
	"net"
)

// TCP server端

func Process(conn net.Conn) {
	defer conn.Close() //关闭连接
	for {
		reader := bufio.NewReader(conn)
		var buf [1024]byte
		n, err := reader.Read(buf[:]) //读取数据
		if err != nil {
			fmt.Println("read failed, err:", err)
			break
		}
		str := string(buf[:n])
		fmt.Println("我收到了客户端发来的数据:", str)
		conn.Write([]byte("OK")) //发送数据
	}
}

func main() {
	listen, err := net.Listen("tcp", "127.0.0.1:20000")
	if err != nil {
		fmt.Println("listen failed, err:", err)
		return
	}
	for {
		conn, err := listen.Accept() //保持时刻建立连接
		if err != nil {
			fmt.Println("conn failed, err:", err)
			continue
		}
		go Process(conn)
	}
}

另一边创建client.go作为客户端:

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

func main() {
	conn, err := net.Dial("tcp", "127.0.0.1:20000") //连接
	if err != nil {
		fmt.Println("conn failed err:", err)
		return
	}
	defer conn.Close()
	inputReader := bufio.NewReader(os.Stdin) //获取终端输入
	for {
		input, _ := inputReader.ReadString('\n') //读取用户输入
		inputInfo := strings.Trim(input, "\r\n")
		if strings.ToUpper(inputInfo) == "Q" {
			return
		}
		_, err := conn.Write([]byte(inputInfo)) //写入数据
		if err != nil {
			return
		}
		var buf [1024]byte
		n, err := conn.Read(buf[:]) //读出数据
		if err != nil {
			fmt.Println("recv failed, err:", err)
			return
		}
		fmt.Println("服务端收到了你的数据并发送了", string(buf[:n]))
	}
}

分别启动server和client,在client的终端上面输入内容,按回车发送内容。服务端接收到客户端发送的内容并返回ok,提示已经接收到数据。

客户端终端:

 服务端终端:简单实现golang的socket编程(TCP)_第1张图片

 

你可能感兴趣的:(golang,tcp/ip,开发语言)