Created
January 10, 2019 12:17
-
-
Save l0k18/23d32a8a6f16a8d411683ed705c8bd92 to your computer and use it in GitHub Desktop.
Super simple example UDP send and listen
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package main | |
import ( | |
"log" | |
"net" | |
"time" | |
) | |
type nodeCfg struct { | |
Listener string | |
Worker string | |
BufferSize int | |
} | |
type node struct { | |
cfg nodeCfg | |
connection *net.UDPConn | |
kill chan bool | |
} | |
const ( | |
uNet = "udp4" | |
) | |
var ( | |
_n = nodeCfg{ | |
Listener: "127.0.0.1:11011", | |
Worker: "127.0.0.1:11012", | |
BufferSize: 10240, | |
} | |
_w = nodeCfg{ | |
Listener: "127.0.0.1:11012", | |
Worker: "127.0.0.1:11011", | |
BufferSize: 10240, | |
} | |
) | |
func main() { | |
n := newNode(_n) | |
n.setupListener() | |
time.Sleep(time.Second * 1) | |
w := newNode(_w) | |
go n.readFromSocket() | |
for { | |
time.Sleep(time.Second) | |
go w.send([]byte("hello world")) | |
} | |
} | |
func newNode(nc nodeCfg) (n *node) { | |
n = &node{ | |
cfg: nc, | |
kill: make(chan bool), | |
} | |
return | |
} | |
func (n *node) setupListener() { | |
addr, err := net.ResolveUDPAddr(uNet, n.cfg.Listener) | |
check(err) | |
n.connection, err = net.ListenUDP(uNet, addr) | |
check(err) | |
} | |
func (n *node) readFromSocket() { | |
for { | |
var b = make([]byte, n.cfg.BufferSize) | |
count, addr, err := n.connection.ReadFromUDP(b[0:]) | |
check(err) | |
b = b[:count] | |
if count > 0 { | |
log.Print("'", string(b), "' <- ", addr) | |
select { | |
case <-n.kill: | |
break | |
default: | |
} | |
} | |
} | |
} | |
func (n *node) send(b []byte) { | |
addr, err := net.ResolveUDPAddr("udp4", n.cfg.Worker) | |
check(err) | |
conn, err := net.DialUDP(uNet, nil, addr) | |
check(err) | |
log.Print("'", string(b), "' -> ", addr) | |
_, err = conn.Write(b) | |
check(err) | |
} | |
func check(err error) { | |
if err != nil { | |
log.Print("ERR ", err) | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment