go-fly/controller/tcp.go

package controller

import (
	"github.com/gin-gonic/gin"
	"log"
	"net"
)

var clientTcpList = make(map[string]net.Conn)

func NewTcpServer(tcpBaseServer string) {
	listener, err := net.Listen("tcp", tcpBaseServer)
	if err != nil {
		log.Println("Error listening", err.Error())
		return //终止程序
	}
	// 监听并接受来自客户端的连接
	for {
		conn, err := listener.Accept()
		if err != nil {
			log.Println("Error accepting", err.Error())
			return // 终止程序
		}
		var remoteIpAddress = conn.RemoteAddr()
		clientTcpList[remoteIpAddress.String()] = conn
		log.Println(remoteIpAddress, clientTcpList)
		//clientTcpList=append(clientTcpList,conn)
	}
}
func PushServerTcp(str []byte) {
	for ip, conn := range clientTcpList {
		line := append(str, []byte("\r\n")...)
		_, err := conn.Write(line)
		log.Println(ip, err)
		if err != nil {
			conn.Close()
			delete(clientTcpList, ip)
			//clientTcpList=append(clientTcpList[:index],clientTcpList[index+1:]...)
		}
	}
}
func DeleteOnlineTcp(c *gin.Context) {
	ip := c.Query("ip")
	for ipkey, conn := range clientTcpList {
		if ip == ipkey {
			conn.Close()
			delete(clientTcpList, ip)
		}
		if ip == "all" {
			conn.Close()
			delete(clientTcpList, ipkey)
		}
	}
	c.JSON(200, gin.H{
		"code": 200,
		"msg":  "ok",
	})
}