golang 长连接

· · 1884 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

参考:https://segmentfault.com/a/1190000017866100

package main

import (
    "fmt"
    "net/http"
    "time"
    "context"
    "sync"
)

var Status map[string](chan string) = make(map[string](chan string))
var Timeout = 1000
var Mutex sync.Mutex


func Get(w http.ResponseWriter, r *http.Request) {
    key, ok := r.URL.Query()["key"]
    fmt.Println(key[0])
    if !ok || len(key) < 1 {
        fmt.Println("get params failure")
        return
    }
    Mutex.Lock()
    Status[key[0]] = make(chan string)
    Mutex.Unlock()
    ctx, cancel := context.WithTimeout(context.Background(), time.Duration(Timeout)*time.Second)

    defer func(){
         Mutex.Lock()
         delete(Status, key[0])
         Mutex.Unlock()
         //close(Status[key[0]])
    }()

    defer cancel()
    select {
        case <-ctx.Done():  //timeout超时后,执行此case
            fmt.Fprintln(w, "Time out.")
        case result := <-Status[key[0]]:  //不用context,此处永久阻塞
            fmt.Fprintln(w, result)
    }

}

func Set(w http.ResponseWriter, r *http.Request) {
    fmt.Println(Status)
    value, ok := r.URL.Query()["value"]
    if !ok{
        fmt.Println("get params failure")
        return
    }

    keys, ok := r.URL.Query()["key"]
    fmt.Println(keys[0])
    if !ok{
        fmt.Println("get params failure")
        return
    }
    Mutex.Lock()
    Status[keys[0]] <- value[0]  //chan写key对应value, Get 接口对应pending阻塞将释放,api调用完成.
    Mutex.Unlock()
    defer func(){
        close(Status[keys[0]])
    }()

    fmt.Fprintln(w, "hello set")
}

func OldGet(w http.ResponseWriter, r *http.Request) {
    // Map是不能并发的写操作,但可以并发的读.
    // Status[key[0]] = make(chan string) 初始化通道是map写操作
    // result := <-Status[key[0]] 通道中取值也是map写操作,因通道中值取出后就无值,所以也是写操作
    // Status[key[0]] <- value向Status[key[0]] 通道内写值也map写操作
    // 因此三处操作的是一个map,所以要加同一把锁
    key, ok := r.URL.Query()["key"]
    fmt.Println(key[0])
    if !ok || len(key) < 1 {
        fmt.Println("get params failure")
        return
    }
    Status[key[0]] = make(chan string)
    ctx, cancel := context.WithTimeout(context.Background(), time.Duration(Timeout)*time.Second)
    defer cancel()
    select {
        case <-ctx.Done():  //timeout超时后,执行此case
            fmt.Fprintln(w, "Time out.")
        case result := <-Status[key[0]]:  //不用context,此处永久阻塞
            fmt.Fprintln(w, result)
    }

}




func main() {
    http.HandleFunc("/gets", Get)
    http.HandleFunc("/sets", Set)
    http.ListenAndServe(":8000", nil)
}


有疑问加站长微信联系(非本文作者)

本文来自:简书

感谢作者:

查看原文:golang 长连接

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

1884 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传