golang练习之聊天室——丰富客户端功能

前面一篇设计的客户端太简陋了，只向服务端发送一个hello world就退出了。现在我们要丰富一下客户端的功能。让客户端不仅能向服务端发送数据，也能接收服务端发送过来的数据。

server代码：

package main

import (
	"github.com/apex/log"
	"net"
)

func handleOneClient(client net.Conn) {
	defer client.Close()
	buf := make([]byte, 1024)
	for {
		nbytes, err := client.Read(buf)

		if err != nil {
			return
		}
		log.Info(string(buf[:nbytes]))
		log.Info(client.RemoteAddr().String())
	}
	log.Info("client close.")
}

func main() {
	listener,err := net.Listen("tcp4", "127.0.0.1:54321")
	if err != nil {
		return
	}
	for {
		oneClient, err := listener.Accept()
		if err != nil {
			continue
		}

		go handleOneClient(oneClient)
	}
}

可以看到服务端代码只新增了打印客户端地址的功能。

client代码：

package main

import (
	"bufio"
	"github.com/apex/log"
	"net"
	"os"
	"strings"
)

func handleInput(conn net.Conn) {
	defer conn.Close()
	for {
		r := bufio.NewReader(os.Stdin)
		input, _, err := r.ReadLine()
		if err != nil {
			log.Error(err.Error())
			break
		}

		if strings.ToUpper(string(input)) == string("QUIT") {
			break
		}

		_, err = conn.Write([]byte(input))
		if err != nil {
			log.Error(err.Error())
			break
		}
	}
}
func main() {
	conn, err := net.Dial("tcp4", "127.0.0.1:54321")
	if err != nil {
		return
	}
	defer conn.Close()

	//处理往服务器发送数据的功能
	go handleInput(conn)

	buf := make([]byte, 1024)
	for {
		nbytes, err := conn.Read(buf)
		if err != nil {
			return
		}
		log.Info(string(buf[:nbytes]))
	}
}

客户端代码也很简单，在主线程里执行读取服务端数据功能，在goroutine里执行用户输入发送给服务端功能。

因为我很少写bufio.Reader（osStdin）,r.Readline()这种代码，这更像是C语言练习的时候用的scanf这种。所以稍微有些不熟悉。

 

（全文完）