package znet

import (
	"fmt"
	"go-study/zinx/ziface"
	"net"
)

type Server struct {
	// Name 服务器名称
	Name string
	// IPVersion tcp4 or tcp6
	IPVersion string
	// IP 服务器监听的 IP
	IP string
	// Port 服务器监听的端口
	Port int
}

// NewServer 创建一个服务器句柄
func NewServer(name string) ziface.IServer {
	return &Server{
		Name:      name,
		IPVersion: "tcp4",
		IP:        "0.0.0.0",
		Port:      7777,
	}
}

// Start 启动服务器
func (s *Server) Start() {
	fmt.Printf("[START] Server listenner at IP: %s, Port %d, is starting\n", s.IP, s.Port)

	go func() {
		// 1. 获取一个 TCP 的 Addr
		addr, err := net.ResolveTCPAddr(s.IPVersion, fmt.Sprintf("%s:%d", s.IP, s.Port))
		if err != nil {
			fmt.Println("resolve tcp addr error:", err)
			return
		}

		// 2. 监听服务器地址
		listener, err := net.ListenTCP(s.IPVersion, addr)
		if err != nil {
			fmt.Println("listen", s.IPVersion, "err", err)
			return
		}

		fmt.Println("start Zinx server ", s.Name, " succ, now listening...")

		// 3. 阻塞等待客户端连接，处理客户端连接业务（读写）
		for {
			// 3.1 阻塞等待客户端连接，AcceptTCP 会阻塞
			conn, err := listener.AcceptTCP()
			if err != nil {
				fmt.Println("Accept err", err)
				continue
			}

			// 3.2 TODO 设置服务器最大连接数，如果超过最大连接数则关闭此新连接

			// 3.3 TODO 处理该新连接请求的业务方法，此时 handler 和 conn 应该是绑定的

			// 这里暂时做一个最大 512 字节的回显服务
			go func() {
				// 不断的循环，从客户端获取数据
				for {
					buf := make([]byte, 512)
					cnt, err := conn.Read(buf)
					if err != nil {
						fmt.Println("recv buf err", err)
						continue
					}

					// 回显
					if _, err := conn.Write(buf[:cnt]); err != nil {
						fmt.Println("write back buf err", err)
						continue
					}
				}
			}()
		}
	}()
}

// Stop 停止服务器
func (s *Server) Stop() {
	fmt.Println("[STOP] Zinx server, name ", s.Name)

	// TODO 将需要清理的连接信息或者其他信息一并停止或者清理
}

// Serve 运行服务器
func (s *Server) Serve() {
	s.Start()

	// TODO 做一些启动服务器之后的额外业务

	// 阻塞，否则主 Go 程退出，listener 会退出
	select {}
}