-
Notifications
You must be signed in to change notification settings - Fork 0
/
entry.go
73 lines (69 loc) · 1.63 KB
/
entry.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package connection
import (
"context"
"github.com/mailru/easygo/netpoll"
"lib/logger"
"net"
"server/routine"
"server/store"
"time"
)
func Init() error {
initEpoller()
return nil
}
func Handle(netConn net.Conn) error {
conn := &storeConn{netConn: netConn}
if isNetDegrade() {
logger.Info("[Handle] net is degrade, use go-routine.")
ctx, _ := context.WithTimeout(context.Background(), time.Second)
err := routine.Start(ctx, func(t *routine.Task) (err error) {
degradeProcess(conn)
return nil
})
if err != nil {
logger.Error("[Handle] routine start error.err:%s", err.Error())
return err
}
return nil
}
conn.epollFd, _ = netpoll.HandleReadOnce(conn.netConn)
handleFunc := func(ctx context.Context) error {
if deadLine, ok := ctx.Deadline(); ok {
conn.SetReadDeadline(deadLine)
conn.SetWriteDeadline(deadLine)
}
msg, err := conn.Read()
if err != nil {
conn.Close()
logger.Info("[Handle] rend conn msg error. err:%s", err.Error())
return err
}
result, err := store.Execute(msg)
if err != nil {
logger.Error("[Handle] store execute error. err:%s", err.Error())
return err
}
conn.Write(result)
return nil
}
return epollStart(conn, time.Second, handleFunc)
}
func degradeProcess(conn *storeConn) {
for {
msg, err := conn.Read()
if err != nil {
conn.Close()
logger.Info("[degradeProcess] rend conn msg error. err:%s", err.Error())
break
}
result, err := store.Execute(msg)
if err != nil {
conn.Close()
logger.Error("[degradeProcess] store execute error. err:%s", err.Error())
break
}
conn.SetWriteDeadline(time.Now().Add(time.Second))
conn.Write(result)
}
}