golang长连接

1. type Longsocket struct {
2. Ws *websocket.Conn
3. writeCh chan []byte
4. readCh chan []byte
5. ShakeHand bool
6. Url string
7. Protocol string
8. Origin string
9. BufferSize int
10. Status int
11. mu sync.Mutex
12. }

1. //call func with a gorouting, it will send shake hands message to service to make sure self is ok
2. //if you want to send message call func 'Write', and the case writeCh will be vaild
3. func (l *Longsocket) WriteLoop() {
4. defer func() {
5. if err := recover(); err != nil {
6. //fmt.Println("writeloop", err)
7. }
8. }()
9. for {
10. errCount := 0
11. if l.Status != STATUS_CONNECT {
12. break
13. }
14. select {
15. case <-time.After(time.Second * time.Duration(SHAKE_HANDS_FREQUENCY)):
16. if l.ShakeHand {
17. _, err := l.Ws.Write([]byte(SHAKE_HANDS_MSG))
18. if err != nil {
19. errCount++
20. }
21. }
22. case msg := <-l.writeCh:
23. _, err := l.Ws.Write(msg)
24. if err != nil {
25. errCount++
26. }
27. }
28. if errCount != 0 {
29. break
30. }
31. }
32. l.Close()
33. }

1. //read message form socket and write them to readCh
2. func (l *Longsocket) ReadLoop() {
3. defer func() {
4. if err := recover(); err != nil {
5. //fmt.Println("readloop", err)
6. }
7. }()
8. for {
9. if l.Status != STATUS_CONNECT {
10. break
11. }
12. buf := make([]byte, l.BufferSize)
13. n, err := l.Ws.Read(buf)
14. if err != nil {
15. break
16. }
17. if n > 0 {
18. l.readCh <- buf[0:n]
19. }
20. }
21. l.Close()
22. }

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

• 请尽量让自己的回复能够对别人有帮助
• 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
• 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
• 图片支持拖拽、截图粘贴等方式上传