Go实现简单的Socket服务端笔记(二)

第二步 循环读取数据

利用for循环,等待客户端发送数据

本文代码查看github:
https://github.com/zboyco/go-server/tree/step-2

修改上一步的代码,让服务端循环接收客户端发送的消息,主要使用for循环,直接上代码

package main

import (
	"fmt"
	"net"
)

func main() {

	//定义一个本机端口
	localAddress, _ := net.ResolveTCPAddr("tcp4", ":9043")

	//监听端口
	tcpListener, err := net.ListenTCP("tcp", localAddress)

	if err != nil {
		fmt.Println("监听出错, ", err)
		return
	}

	//程序返回后关闭socket
	defer tcpListener.Close()

	fmt.Println("等待客户连接...")

	//开始接收连接
	conn, err := tcpListener.Accept()

	if err != nil {
		fmt.Println("客户连接失败, ", err)
	}

	//获取连接地址
	remoteAddr := conn.RemoteAddr()

	fmt.Println("客户地址:", remoteAddr)

	//定义一个数据接收Buffer
	var buf [1024]byte

	for {
		fmt.Println("等待接收数据...")
		//读取数据
		n, err := conn.Read(buf[0:])

		if err != nil {
			fmt.Println("数据接收错误, ", err)
			return
		}

		//将bytes转为字符串
		result := string(buf[0:n])

		//输出结果
		fmt.Println("接收到数据:", result)
	}
}

上面的代码和第一步几乎一样,只是在最后接收用for套起来,实现循环接收

	//定义一个数据接收Buffer
	var buf [1024]byte

	for {
		fmt.Println("等待接收数据...")
		//读取数据
		n, err := conn.Read(buf[0:])

		if err != nil {
			fmt.Println("数据接收错误, ", err)
			return
		}

		//将bytes转为字符串
		result := string(buf[0:n])

		//输出结果
		fmt.Println("接收到数据:", result)
	}