注册HTTP URL处理程序的程序切片

Golang不知道大家是否熟悉？今天我将给大家介绍《注册HTTP URL处理程序的程序切片》，这篇文章主要会讲到等等知识点，如果你在看完本篇文章后，有更好的建议或者发现哪里有问题，希望大家都能积极评论指出，谢谢！希望我们能一起加油进步！

为了将消息从 goroutine 广播到多个 http url 处理程序，我尝试注册这些 http url 处理程序，并在 main.go 中使用以下代码：

type websockethandler func(http.responsewriter, *http.request)

type threadsafeslice struct {
    sync.mutex
    handlers []*websockethandler
}

var sliceofhandlers threadsafeslice

func (slice *threadsafeslice) push(handle *websockethandler) { //register

    slice.lock()
    defer slice.unlock()

    slice.handlers = append(slice.handlers, handle)
}

其中forwardmsgtoclient()是需要注册的http url处理程序，

broadcastmessage() goroutine 可以将消息广播到多个 forwardmsgtoclient() 处理程序，代码如下：

func main() {

    go broadcastmessage()
    http.handlefunc("/websocket", forwardmsgtoclient)
    http.listenandserve(":3000", nil)

}

func forwardMsgToClient(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    for {
         // Forward message to the client upon receiving a msg from publisher
    }
}

以上代码均在main.go中

但是问题是，在 rw、e := l.accept() 在 ../go/src/net/http 中调用之后，为各个客户端生成 goroutine forwardmsgtoclient() /server.go。

注册(push()) http url处理函数(forwardmsgtoclient())的原因是为了让broadcastmessage() goroutine知道为所有http url处理程序创建的通道数量，并且取消注册http url处理函数时删除通道(forwardmsgtoclient())。

有点紧张，如果我们需要修改 /go/src/net/http/server.go 来实现这个目标

如何在sliceofhandlers.handlers中注册(push())http url处理函数forwardmsgtoclient()？

解决方案

要将消息广播到所有连接的 websocket 客户端，请执行以下操作：

• 在升级时将连接添加到集合。
• 连接关闭时从集合中删除连接。
• 通过迭代集合进行广播。

一个简单的方法是：

type clients struct {
    sync.mutex
    m map[*websocket.conn]struct{}
}

var clients = clients{m: map[*websocket.conn]struct{}{}}

func (cs *clients) add(c *websocket.conn) {
    cs.lock()
    cs.m[c] = struct{}{}
    cs.unlock()
}

func (cs *clients) remove(c *websocket.conn) {
    cs.lock()
    delete(cs.m, c)
    cs.unlock()
}

func (cs *clients) broadcast(message []byte) {
    cs.lock()
    defer cs.unlock()
    for c, _ := range m {
       c.writemessage(websocket.textmessage, message)
    }
}

处理程序从集合中添加和删除连接，如下所示：

func forwardMsgToClient(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        // handle error
    }
    defer c.Close()
    clients.add(c)
    defer clients.remove(c)

    // Read the connection as required by the package.
    for {
        if _, _, err := c.NextReader(); err != nil {
            break
        }
    }
}

要向所有连接的客户端发送消息，请调用 clients.broadcast(message)。

由于以下几个原因，这种简单的方法尚未准备好用于生产：它不处理从 writemessage 返回的错误，广播可能会在卡住的客户端上阻塞。

有关更强大的解决方案，请参阅 Gorilla chat example hub。集线器在广播者和连接之间插入一个通道，从而允许集线器无阻塞地进行广播。问题中的 go broadcastmessage() 对应于 gorilla 示例中的 go hub.run() 。问题中的 forwardmsgtoclient 处理程序将创建一个 *client 并在升级时将其发送到集线器 register 通道，并在断开连接时将该 *client 发送到集线器 unregister 通道。 *client 有一个泵送到连接的通道。

本篇关于《注册HTTP URL处理程序的程序切片》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！