Draft: big refactor
This commit is contained in:
50
utils/udp.go
Normal file
50
utils/udp.go
Normal file
@@ -0,0 +1,50 @@
|
||||
package workflows
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net"
|
||||
|
||||
"github.com/vmihailenco/msgpack/v5"
|
||||
)
|
||||
|
||||
type UDPPayload struct {
|
||||
MagicNumber int `msgpack:"magicNumber"`
|
||||
Data string `msgpack:"data"`
|
||||
}
|
||||
|
||||
const (
|
||||
BUFFER_SIZE int = 1024
|
||||
MAGIC_NUMBER int = 114514
|
||||
)
|
||||
|
||||
func ListenUDP(port string) (string, error) {
|
||||
addr, err := net.ResolveUDPAddr("udp", ":"+port)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to resolve address, %w", err)
|
||||
}
|
||||
|
||||
conn, err := net.ListenUDP("udp", addr)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to listen on addr '%v', %w", addr, err)
|
||||
}
|
||||
defer conn.Close()
|
||||
|
||||
for {
|
||||
buffer := make([]byte, BUFFER_SIZE)
|
||||
|
||||
n, clientAddr, err := conn.ReadFromUDP(buffer)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
|
||||
var payload UDPPayload
|
||||
err = msgpack.Unmarshal(buffer[:n], &payload)
|
||||
if err == nil && payload.MagicNumber == MAGIC_NUMBER {
|
||||
return payload.Data
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func SendPayload(endpoint, data string) error {
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user