falloutBot/go-mc
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #65 from DeadlySurgeon/master

Browse Source 
Adds `Close()` function to RCONClientConn and RCONServerConn
This commit is contained in:
Tnze
2020-07-13 09:51:48 +08:00
committed by GitHub
parent ee23be105f 2159d1c959
commit 6d526bc7e7
2 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
net/rcon.go
View File

@ -177,12 +177,14 @@ func (r *RCONConn) RespCmd(resp string) error {
type RCONClientConn interface { type RCONClientConn interface {
Cmd(cmd string) error Cmd(cmd string) error
Resp() (resp string, err error) Resp() (resp string, err error)
Close() error
} }
type RCONServerConn interface { type RCONServerConn interface {
AcceptLogin(password string) error AcceptLogin(password string) error
AcceptCmd() (cmd string, err error) AcceptCmd() (cmd string, err error)
RespCmd(resp string) error RespCmd(resp string) error
Close() error
} }
func ListenRCON(addr string) (*RCONListener, error) { func ListenRCON(addr string) (*RCONListener, error) {

3
net/rcon_test.go
View File

@ -17,6 +17,8 @@ func server(t *testing.T, prepare chan<- int) {
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
defer l.Close()
prepare <- 1 prepare <- 1
for { for {
@ -55,6 +57,7 @@ func client(t *testing.T) {
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
defer conn.Close()
err = conn.Cmd("TEST COMMAND") err = conn.Cmd("TEST COMMAND")
if err != nil { if err != nil {