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

将创建socket的方法放入单独的包中

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

修改以下项目结构,将之前的代码放到server包中,main.go中直接调用就行了

项目结构如下:

|-go-server
|-server
	|-socket.go

socket.go 中增加一个结构体,增加一个工厂函数用来返回一个新的Server.

代码如下:

socket.go

package server

import (
	"fmt"
	"net"
)

//服务结构
type Server struct {
	ip   string
	port int
}

//新建一个服务
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)
		return
	}

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

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

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

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

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

func handleClient(conn net.Conn) {
	//获取连接地址
	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)
	}
}

main.go

package main

import (
	"fmt"

	"go-server/server"
)

func main() {

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

	mainServer.Start()
}