refactor: add context propagation to connection and event systems

client/client.go

@@ -1,6 +1,7 @@
 package client
 
 import (
+	"context"
 	"net"
 
 	"github.com/admin-else/strom/proto"
@@ -22,13 +23,17 @@ func Connect(addr string) (ret *proto.Conn, err error) {
 	return
 }
 
+func ConnectVersionLess(addr string) (ret *proto.Conn, err error) {
+	return ConnectVersionLessCtx(context.Background(), addr)
+}
+
 // ConnectVersionLess connects to the given address without setting the version you should set it
 // when using this function.
 // You must close the connection when you're done with it
-func ConnectVersionLess(addr string) (ret *proto.Conn, err error) {
-	ret = proto.NewConn()
+func ConnectVersionLessCtx(ctx context.Context, addr string) (ret *proto.Conn, err error) {
+	ret = proto.NewConnCtx(ctx)
 	ret.Actor = proto_base.Client
-	ret.Conn, err = net.Dial("tcp", addr)
+	ret.Conn, err = (&net.Dialer{}).DialContext(ctx, "tcp", addr)
 	if err != nil {
 		return
 	}

client/status.go

@@ -1,6 +1,7 @@
 package client
 
 import (
+	"context"
 	"encoding/json"
 	"time"
 
@@ -39,7 +40,7 @@ func (s *StatusClient) OnPong(p *v1_21_8.StatusToClientPacketPing) (err error) {
 // StatusRaw returns a StatusClient that is not connected to a server.
 // So ignore the resource leak warning. And maybe attach a warn ignore comment.
 // It does not resolve SRV records.
-func StatusRaw(addr string) (s *StatusClient, err error) {
+func StatusRaw(ctx context.Context, addr string) (s *StatusClient, err error) {
 	c, err := ConnectVersionLess(addr)
 	if err != nil {
 		return
@@ -72,7 +73,7 @@ func StatusRaw(addr string) (s *StatusClient, err error) {
 
 // StatusNoDns is like Status but does not resolve SRV records.
 func StatusNoDns(addr string) (status server.StatusResponse, err error) {
-	s, err := StatusRaw(addr)
+	s, err := StatusRaw(context.Background(), addr)
 	if err != nil {
 		return
 	}