middleware/main.go at master · InoFlexin/middleware · GitHub

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
package main //simple server example

import (
	"fmt"
	"log"
	"net"
	"net/http"
	"sync"
	"time"

	"github.com/InoFlexin/serverbase/thttp"
	"github.com/InoFlexin/serverbase/base"
	"github.com/InoFlexin/serverbase/client"
)

//Override Server Message Type...
type (
	MyMessage       base.Message
	MyClientMessage base.Message
)

func (m MyMessage) OnMessageReceive(message *base.Message, client net.Conn) {
	fmt.Println("on message receive: "+message.Json+" action: %d", message.Action)

	base.Write(&base.Message{Json: "pong", Action: base.ON_MSG_RECEIVE}, client)
}

func (m MyMessage) OnConnect(message *base.Message, client net.Conn) {
	fmt.Println("on connect: "+message.Json+" action: %d", message.Action)
}

func (m MyMessage) OnClose(err error) {
	log.Println(err)
}

func (m MyClientMessage) OnMessageReceive(message *base.Message, server net.Conn) {
	fmt.Println("client on message receive: "+message.Json+" action: %d", message.Action)
}

func (m MyClientMessage) OnConnect(message *base.Message, server net.Conn) {
	fmt.Println("client on connect: "+message.Json+" action: %d", message.Action)
}

func (m MyClientMessage) OnClose(err error) {
	log.Println(err)
}

func main() {
	handler := thttp.HttpHandler{}


	wg := sync.WaitGroup{} //synchronized goroutine

	ev := MyMessage{}
	boot := base.Boot{Protocol: "tcp", Port: ":5092", ServerName: "testserver-35", Callback: ev, ReceiveSize: 1024, Complex: true}

	evm := MyClientMessage{}
	clientBoot := client.ClientBoot{Protocol: "tcp", HostAddr: "localhost", HostPort: ":5092", Callback: evm, BufferSize: 1024}

	wg.Add(1)
	go base.ServerStart(boot, &wg) //Server open
	wg.Wait()

	wg.Add(1)
	go client.ConnectServer(&clientBoot, &wg) //Client request connect to server
	wg.Wait()
	http.ListenAndServe(":5092",handler)

	/*
		Client test logics
	*/
	serverError := client.SendPing(time.Second * 2)

	if serverError != nil {
		log.Fatal(serverError)
	}
}