Mirror/mochi
1
0
Fork 0
mirror of https://github.com/sot-tech/mochi.git synced 2026-05-13 01:08:36 -07:00
Code Issues Packages Projects Releases Wiki Activity

tracker: Pass listen address into tracker.Server.Serve()

Browse Source
This commit is contained in:
Justin Li
2015-02-21 13:16:21 -05:00
parent f98c675bc7
commit 669128c83a
4 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
udp/udp.go
View File

@@ -25,13 +25,13 @@ type Server struct {
done bool
}
func (s *Server) serve() error {
listenAddr, err := net.ResolveUDPAddr("udp", s.config.UDPListenAddr)
func (s *Server) serve(listenAddr string) error {
udpAddr, err := net.ResolveUDPAddr("udp", listenAddr)
if err != nil {
return err
}
sock, err := net.ListenUDP("udp", listenAddr)
sock, err := net.ListenUDP("udp", udpAddr)
defer sock.Close()
if err != nil {
return err
@@ -75,8 +75,8 @@ func (s *Server) serve() error {
}
// Serve runs a UDP server, blocking until the server has shut down.
func (s *Server) Serve() {
glog.V(0).Info("Starting UDP on ", s.config.UDPListenAddr)
func (s *Server) Serve(addr string) {
glog.V(0).Info("Starting UDP on ", addr)
go func() {
// Generate a new IV every hour.
@@ -85,7 +85,7 @@ func (s *Server) Serve() {
}
}()
if err := s.serve(); err != nil {
if err := s.serve(addr); err != nil {
glog.Errorf("Failed to run UDP server: %s", err.Error())
} else {
glog.Info("UDP server shut down cleanly")