You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.3 KiB
59 lines
1.3 KiB
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",
|
|
})
|
|
}
|