go, websocket的线程池编程

go 服务器demo代码如下:

package main

import (
    "fmt"
    "log"
    "net/http"
    "sync"

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
    CheckOrigin: func(r *http.Request) bool {
        // 允许来自任何源的连接,或者根据你的需求设置
        return true
    },
}

const (
    MaxConnections = 1000                //最大并发连接数;
)

var (
    g_semaphore = make(chan struct{}, MaxConnections) //信号量;
    g_waitGroup sync.WaitGroup                        //计数器;
)

func newConn(conn *websocket.Conn) {

    defer g_waitGroup.Done()         //计数器-1;
    defer func() { <-g_semaphore }() //释放一个信号量;

    //
    defer conn.Close()

    for {
        // 读取客户端发来的消息
        _, message, err := conn.ReadMessage()
        if err != nil {
            log.Println("read err: ", err)
            break
        }
        fmt.Printf("received: %s\n", message)

        // 发送消息给客户端
        // err = conn.WriteMessage(websocket.TextMessage, 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

风尘无名

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值