tux/trok
1
0
Fork 0
mirror of https://github.com/tuxdotrs/trok.git synced 2025-07-06 20:06:34 +05:30
Code Issues Packages Projects Releases Wiki Activity

feat: setup addr arg

Browse Source
This commit is contained in:
tux
2025-02-27 22:17:13 +05:30
parent e64319616f
commit b5756cf429
4 changed files with 21 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
internal/server/tcp.go
View File

@ -5,7 +5,6 @@ Copyright © 2024 tux <0xtux@pm.me>
package server
import (
"fmt"
"net"
)
@ -14,13 +13,8 @@ type TCPServer struct {
listener net.Listener
}
func (s *TCPServer) Init(port uint16, title string) error {
address := ":"
if port > 0 {
address = fmt.Sprintf(":%d", port)
}
ln, err := net.Listen("tcp", address)
func (s *TCPServer) Init(addr, title string) error {
ln, err := net.Listen("tcp", addr)
if err != nil {
return err
}
@ -44,6 +38,14 @@ func (s *TCPServer) Stop() error {
return s.listener.Close()
}
func (s *TCPServer) Addr() string {
return s.listener.Addr().String()
}
func (c *TCPServer) Host() string {
return c.listener.Addr().(*net.TCPAddr).IP.String()
}
func (s *TCPServer) Port() uint16 {
return uint16(s.listener.Addr().(*net.TCPAddr).Port)
}