-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
98 lines (89 loc) · 2.06 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
87
88
89
90
91
92
93
94
95
96
97
98
/*
____ ___ _____ ___________
\ \/ / / \\__ ___/
\ / / \ / \ | |
/ \ / Y \| |
/___/\ \\____|__ /|____|
\_/ \/
createTime:2022/9/11
createUser:Administrator
*/
package main
import (
"fmt"
"log"
"net"
"test/customer_pool"
"time"
)
type PoolTest struct {
Conn *net.UDPConn
}
var myPool *customer_pool.MyConnPool
func init() {
myPool = customer_pool.NewMyConnPool(3, 1, func() (interface{}, error) {
return connectUdp()
})
if myPool == nil {
log.Panicln("NewMyConnPool error")
return
}
log.Println("myPool == ", myPool)
}
func main() {
for i := 0; i < 10; i++ {
msg := fmt.Sprintf("send udp data is %d", i)
go SendMsg(msg)
}
time.Sleep(10 * time.Second)
}
// 初始化链接类
func connectUdp() (*PoolTest, error) {
// 创建一个 udp 句柄
log.Println(">>>>> 创建一个 udp 句柄 ... ")
// 连接服务器
conn, err := net.DialUDP("udp", nil, &net.UDPAddr{
IP: net.IPv4(127, 0, 0, 1),
Port: 9998,
})
if err != nil {
log.Println("Connect to udp server failed,err:", err)
return nil, err
}
log.Printf("<<<<<< new udp connect %+v", conn)
return &PoolTest{Conn: conn}, nil
}
func SendMsg(msg string) {
var client *PoolTest
// 从连接池中获取一个实例
obj, err := myPool.GetObject()
if err != nil {
log.Printf("GetObject error : %+v", err)
return
}
client = obj.(*PoolTest)
// 调用需要的方法
log.Println(msg)
client.SendMsg([]byte(msg))
// 交还连接池
myPool.ReturnObject(client)
}
func (this *PoolTest) SendMsg(data []byte) {
_, err := this.Conn.Write(data)
if err != nil {
log.Printf("write udp error : %+v", err)
return
}
//读取回信
result := make([]byte, 1024)
//if err := this.Conn.SetReadDeadline(time.Now().Add(1 * time.Second)); err != nil {
// log.Printf("SetReadDeadlineerror: %+v", err)
// return
//}
n, remoteAddr, err := this.Conn.ReadFromUDP(result)
if err != nil {
log.Printf("read udp server msg error [data:%s] :%+v", string(data), err)
return
}
log.Printf("Recived msg from %s, data:%s \n", remoteAddr, string(result[:n]))
}