Go的tcp长连接

Go的tcp长连接

为订阅打基础

server

package main

import (
	"fmt"
	"log"
	"net"
	"strconv"
	"time"
)

type Manger struct {
	Kv map[int]*net.TCPConn
	fl int
}

func (m *Manger)run()  {
	var dataNow int

	go func() {
		for {
			dataNow++
			time.Sleep(time.Second * 2)
		}
	}()

	for {
		for k, v := range m.Kv {
			if dataNow % 10 != k {
				fmt.Println("此时结果: ", dataNow % 10)
				continue
			}

			_, err := v.Write([]byte(strconv.Itoa(dataNow)))
			if err != nil {
				log.Println(err)
			}

			time.Sleep(time.Millisecond * 300)
		}
	}
}

func main() {
	m := &Manger{Kv:make(map[int]*net.TCPConn)}

	go m.run()

	localAddress, _ := net.ResolveTCPAddr("tcp4", "0.0.0.0:3333") //定义一个本机IP和端口。
	tcpListener, err := net.ListenTCP("tcp", localAddress)       //在刚定义好的地址上进监听请求。
	if err != nil {
		fmt.Println("监听出错：", err)
		return
	}
	defer func() { //担心return之前忘记关闭连接，因此在defer中先约定好关它。
		tcpListener.Close()
	}()


	fmt.Println("ok")
	for {
		conn, err := tcpListener.AcceptTCP() //接受连接。
		if err != nil {
			fmt.Println("接受连接失败：")
			return
		}
		fmt.Println("收到链接")
		err = conn.SetKeepAlive(true)
		if err != nil {
			log.Println(err)
		}

		m.Kv[m.fl] = conn
		fmt.Println("收到订阅: ", m.fl)
		m.fl++
	}
}

func re(conn *net.TCPConn)  {
	for {
		data := make([]byte, 1024 * 1024 * 10)
		read, err := conn.Read(data)
		if err != nil {
			//log.Println()
		}

		fmt.Println(string(data[:read]))
	}

}

func w(conn *net.TCPConn)  {
	var i int
	i=1
	for {
		_, err := conn.Write([]byte(strconv.Itoa(i)))
		if err != nil {
			log.Println(err)
		}
		i++

		time.Sleep(time.Second * time.Duration(i))
	}
}

client

package main

import (
	"fmt"
	"log"
	"net"
)

func main() {
	localAddress, _ := net.ResolveTCPAddr("tcp4", "127.0.0.1:3333") //定义一个本机IP和端口。
	dial, err := net.DialTCP("tcp", nil,localAddress)
	if err != nil {
		log.Println(err)
		return
	}
	dial.SetKeepAlive(true)

	for {
		data := make([]byte, 1024)
		read, err := dial.Read(data)
		if err != nil {
			log.Println(err)
			continue
		}

		fmt.Println(string(data[:read]))
	}
}