/
gudp_server.go
98 lines (86 loc) · 2.35 KB
/
gudp_server.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
// Copyright 2017 gf Author(https://github.com/gogf/gf). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://github.com/gogf/gf.
package gudp
import (
"errors"
"net"
"github.com/gogf/gf/container/gmap"
"github.com/gogf/gf/os/glog"
"github.com/gogf/gf/util/gconv"
)
const (
gDEFAULT_SERVER = "default"
)
// Server is the UDP server.
type Server struct {
conn *Conn // UDP server connection object.
address string // UDP server listening address.
handler func(*Conn) // Handler for UDP connection.
}
var (
// serverMapping is used for instance name to its UDP server mappings.
serverMapping = gmap.NewStrAnyMap(true)
)
// GetServer creates and returns a UDP server instance with given name.
func GetServer(name ...interface{}) *Server {
serverName := gDEFAULT_SERVER
if len(name) > 0 && name[0] != "" {
serverName = gconv.String(name[0])
}
if s := serverMapping.Get(serverName); s != nil {
return s.(*Server)
}
s := NewServer("", nil)
serverMapping.Set(serverName, s)
return s
}
// NewServer creates and returns a UDP server.
// The optional parameter <name> is used to specify its name, which can be used for
// GetServer function to retrieve its instance.
func NewServer(address string, handler func(*Conn), name ...string) *Server {
s := &Server{
address: address,
handler: handler,
}
if len(name) > 0 && name[0] != "" {
serverMapping.Set(name[0], s)
}
return s
}
// SetAddress sets the server address for UDP server.
func (s *Server) SetAddress(address string) {
s.address = address
}
// SetHandler sets the connection handler for UDP server.
func (s *Server) SetHandler(handler func(*Conn)) {
s.handler = handler
}
// Close closes the connection.
// It will make server shutdowns immediately.
func (s *Server) Close() error {
return s.conn.Close()
}
// Run starts listening UDP connection.
func (s *Server) Run() error {
if s.handler == nil {
err := errors.New("start running failed: socket handler not defined")
glog.Error(err)
return err
}
addr, err := net.ResolveUDPAddr("udp", s.address)
if err != nil {
glog.Error(err)
return err
}
conn, err := net.ListenUDP("udp", addr)
if err != nil {
glog.Error(err)
return err
}
s.conn = NewConnByNetConn(conn)
s.handler(s.conn)
return nil
}