go实现高效tcp服务器(适合短连接)类似fasthttp,长连接的每个连接分配一个协程(标准库中的http)

package main


import (
"fmt"
"net"
"os"
"time"
)


const (
MAX_CONN_NUM = 10
)


//服务处理函数,如果处理过程过于复杂,可以启动一定量的协程来做(协程池),

//处理完后再发回去通过通道,实现异步处理的方式

//echo server Goroutine
func EchoFunc(conn net.Conn) {
defer conn.Close()
buf := make([]byte, 1024)
for {
_, err := conn.Read(buf)
if err != nil {
//println("Error reading:", err.Error())
return
}
//send reply
_, err = conn.Write(buf)
if err != nil {
//println("Error send reply:", err.Error())
return
}
}
}


//initial listener and run
func main() {
listener, err := net.Listen("tcp", "0.0.0.0:8088")
if err != nil {
fmt.Println("error listening:", err.Error())
os.Exit(1)
}
defer listener.Close()


fmt.Printf("running ...\n")


var cur_conn_num int = 0
conn_chan := make(chan net.Conn)
ch_conn_change := make(chan int)


go func() {
for conn_change := range ch_conn_change {
cur_conn_num += conn_change
}
}()


go func() {
for _ = range time.Tick(1e8) {
fmt.Printf("cur conn num: %f\n", cur_conn_num)
}
}()


for i := 0; i < MAX_CONN_NUM; i++ {
go func() {
for conn := range conn_chan {
ch_conn_change <- 1
EchoFunc(conn)
ch_conn_change <- -1
}
}()
}


for {
conn, err := listener.Accept()
if err != nil {
println("Error accept:", err.Error())
return
}
conn_chan <- conn
}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值