-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
86 lines (68 loc) · 1.72 KB
/
main.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
74
75
76
77
78
79
80
81
82
83
84
85
86
package main
import (
"fmt"
"log"
"net"
"os"
"os/signal"
"runtime"
"syscall"
"time"
"github.com/bannerchi/dorylus/models"
"github.com/bannerchi/dorylus/tcp"
Config "github.com/bannerchi/dorylus/util/config"
Filter "github.com/bannerchi/dorylus/util/filter"
)
type Callback struct{}
func (this *Callback) OnConnect(c *tcp.Conn) bool {
addr := c.GetRawConn().RemoteAddr()
c.PutExtraData(addr)
fmt.Println("OnConnect:", addr)
return true
}
func (this *Callback) OnMessage(c *tcp.Conn, p tcp.Packet) bool {
//check type
echoPacket := p.(*tcp.EchoPacket)
var resvMsg = new(tcp.EchoPacket)
req := string(echoPacket.GetBody())
resv := Filter.ResponsFilter(req)
resvMsg = tcp.NewEchoPacket(resv, false)
c.AsyncWritePacket(tcp.NewEchoPacket(resvMsg.Serialize(), true), time.Second)
return true
}
func (this *Callback) OnClose(c *tcp.Conn) {
fmt.Println("OnClose:", c.GetExtraData())
}
func main() {
conf := Config.GetConfig()
tcpPort := conf.String("tcp.port")
// init models
models.Init()
//set cpus for max
runtime.GOMAXPROCS(runtime.NumCPU())
// creates a tcp listener
tcpAddr, err := net.ResolveTCPAddr("tcp4", tcpPort)
checkError(err)
listener, err := net.ListenTCP("tcp", tcpAddr)
checkError(err)
// creates a server
config := &tcp.Config{
PacketSendChanLimit: 20,
PacketReceiveChanLimit: 20,
}
srv := tcp.NewServer(config, &Callback{}, &tcp.EchoProtocol{})
// starts service
go srv.Start(listener, time.Second)
fmt.Println("listening:", listener.Addr())
// catchs system signal
chSig := make(chan os.Signal)
signal.Notify(chSig, syscall.SIGINT, syscall.SIGTERM)
fmt.Println("Signal: ", <-chSig)
// stops service
srv.Stop()
}
func checkError(err error) {
if err != nil {
log.Fatal(err)
}
}