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

增加AppSession结构体

OnMessage返回AppSession结构体,提供Send方法,服务器可以主动向客户端发送数据

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

新建一个AppSession结构体,将客户端会话放在AppSession中,方便扩展和管理

1. 在server目录中增加client.go文件
   代码如下:

package server

import "net"

//客户端结构体
type AppSession struct {
	conn net.Conn //socket连接
}

func (client *AppSession) Send(buf []byte) {
	client.conn.Write(buf)
}

2. 修改socket.go

代码如下:

package server

import (
	"fmt"
	"net"
)

//服务结构
type Server struct {
	ip        string
	port      int
	OnError   func(error)
	OnMessage func(*AppSession, []byte)
}

//新建一个服务
func New(ip string, port int) *Server {
	return &Server{
		ip:   ip,
		port: port,
	}
}

//开始监听
func (server *Server) Start() {
	//定义一个本机端口
	localAddress, _ := net.ResolveTCPAddr("tcp4", fmt.Sprintf("%s:%d", server.ip, server.port))

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

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

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

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

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

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

		//启用goroutine处理
		go handleClient(server, conn)
	}
}

//读取数据
func handleClient(server *Server, conn net.Conn) {
	//获取连接地址
	remoteAddr := conn.RemoteAddr()

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

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

	for {
		fmt.Println("等待接收数据...")
		//读取数据,io.Reader 需要传入一个byte切片
		n, err := conn.Read(buf[0:])

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

		if server.OnMessage == nil {
			fmt.Println("错误,未找到数据处理方法!")
			continue
		}
		server.OnMessage(&AppSession{
			conn: conn,
		}, buf[0:n])
	}
}

主要修改了handleClient最后调用onmessage方法,传入了一个AppSession的指针

3. 修改main.go
   修改onMessage参数类型为AppSession

package main

import (
	"fmt"
	"go-server/server"
)

func main() {
	fmt.Println("hello golang!!!")

	mainServer := server.New("127.0.0.1", 9043)

	mainServer.OnMessage = onMessage

	mainServer.OnError = onError

	mainServer.Start()
}

//接收数据方法
func onMessage(client *server.AppSession, bytes []byte) {
	//将bytes转为字符串
	result := string(bytes)

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

	client.Send([]byte("Got!"))
}

//接收错误方法
func onError(err error) {
	//输出结果
	fmt.Println("错误: ", err)
}