refactor: better project structure (#35)
This project is starting to grow, and I've seen flaws in the initial project structure design I think I'll restructure the entire project directory for better scalability and also cleaner
This commit is contained in:
@@ -0,0 +1,61 @@
|
||||
package ipc
|
||||
|
||||
import (
|
||||
"net"
|
||||
)
|
||||
|
||||
type State struct {
|
||||
Query string
|
||||
}
|
||||
|
||||
type Server struct {
|
||||
conn *net.UDPConn
|
||||
StateChan chan State
|
||||
ListenPort int
|
||||
}
|
||||
|
||||
func NewServer() (*Server, error) {
|
||||
addr := &net.UDPAddr{
|
||||
IP: net.ParseIP("127.0.0.1"),
|
||||
Port: 0, // OS assigns random port
|
||||
}
|
||||
conn, err := net.ListenUDP("udp", addr)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
localAddr, ok := conn.LocalAddr().(*net.UDPAddr)
|
||||
if !ok {
|
||||
return nil, net.InvalidAddrError("not a UDP address")
|
||||
}
|
||||
|
||||
s := &Server{
|
||||
conn: conn,
|
||||
StateChan: make(chan State, 100),
|
||||
ListenPort: localAddr.Port,
|
||||
}
|
||||
|
||||
return s, nil
|
||||
}
|
||||
|
||||
func (s *Server) Start() {
|
||||
buf := make([]byte, 4096)
|
||||
for {
|
||||
n, _, err := s.conn.ReadFromUDP(buf)
|
||||
if err != nil {
|
||||
break
|
||||
}
|
||||
|
||||
query := string(buf[:n])
|
||||
// send non-blocking so we don't choke the UDP read loop
|
||||
select {
|
||||
case s.StateChan <- State{Query: query}:
|
||||
default:
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Server) Close() {
|
||||
_ = s.conn.Close()
|
||||
close(s.StateChan)
|
||||
}
|
||||
Reference in New Issue
Block a user