分享
  1. 首页
  2. 文章

golang 长连接

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

参考: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

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

用户登录

没有账号?注册
或
(追記) (追記ここまで)

今日阅读排行

    加载中
(追記) (追記ここまで)

一周阅读排行

    加载中

关注我

  • 扫码关注领全套学习资料 关注微信公众号
  • 加入 QQ 群:
    • 192706294(已满)
    • 731990104(已满)
    • 798786647(已满)
    • 729884609(已满)
    • 977810755(已满)
    • 815126783(已满)
    • 812540095(已满)
    • 1006366459(已满)
    • 692541889

  • 关注微信公众号
  • 加入微信群:liuxiaoyan-s,备注入群
  • 也欢迎加入知识星球 Go粉丝们(免费)

给该专栏投稿 写篇新文章

每篇文章有总共有 5 次投稿机会

收入到我管理的专栏 新建专栏