💬 Raw methods accepts []any

Author: No3371

conn.go

@@ -174,7 +174,11 @@ func NewConnAuth(remote *net.TCPAddr, authProvider AuthProvider) (*Conn, error)
 	return conn, nil
 }
 
-func (c *Conn) BuildSingleActionPacketRaw(segs []string) (raw string, err error) {
+// Allows building a packet easily like:
+//     c.BuildSingleActionPacketRaw("SET", "X", 100)
+//
+// The arguments accept any type. The arguments are formatted internally with %v so most basic types should be supported.
+func (c *Conn) BuildSingleActionPacketRaw(segs []any) (raw string, err error) {
 	c.strBuilder.Reset()
 	_, err = fmt.Fprint(c.strBuilder, "*1\n")
 	if err != nil {
@@ -189,14 +193,15 @@ func (c *Conn) BuildSingleActionPacketRaw(segs []string) (raw string, err error)
 	return c.strBuilder.String(), nil
 }
 
-func (c *Conn) appendSingleActionRaw(segs []string) (err error) {
+func (c *Conn) appendSingleActionRaw(segs []any) (err error) {
 	_, err = fmt.Fprintf(c.strBuilder, "~%d\n", len(segs))
 	if err != nil {
 		return err
 	}
 
 	for _, s := range segs {
-		_, err = fmt.Fprintf(c.strBuilder, "%d\n%s\n", len(s), s)
+		str := fmt.Sprintf("%v", s)
+		_, err = fmt.Fprintf(c.strBuilder, "%d\n%s\n", len(str), str)
 		if err != nil {
 			return err
 		}

conn_actions.go

@@ -295,7 +295,11 @@ func (c *Conn) Exec(ctx context.Context, packet *QueryPacket) ([]response.Respon
 	return rp.resps, nil
 }
 
-func (c *Conn) ExecSingleActionPacketRaw(segments ...string) (response.ResponseEntry, error) {
+// Allows executing a packet easily like:
+//     c.ExecSingleActionPacketRaw("SET", "X", 100)
+//
+// The arguments accept any type. The arguments are formatted internally with %v so most basic types should be supported.
+func (c *Conn) ExecSingleActionPacketRaw(segments ...any) (response.ResponseEntry, error) {
 	raw, err := c.BuildSingleActionPacketRaw(segments)
 	if err != nil {
 		return response.EmptyResponseEntry, err

conn_actions_test.go

@@ -16,13 +16,13 @@ func TestConn_ExecSingleActionPacketRaw(t *testing.T) {
 
 	tests := []struct {
 		name    string
-		args    []string
+		args    []any
 		want    any
 		wantErr bool
 	}{
-		{ "SetX", []string { "SET", "X", "100" }, protocol.RespOkay, false },
-		{ "SysInfoProtocol", []string { "SYS", "INFO", "PROTOCOL" }, skytable.ProtoVer, false },
-		{ "DelX", []string { "DEL", "X" }, uint64(1), false },
+		{ "SetX", []any { "SET", "X", 100 }, protocol.RespOkay, false },
+		{ "SysInfoProtocol", []any { "SYS", "INFO", "PROTOCOL" }, skytable.ProtoVer, false },
+		{ "DelX", []any { "DEL", "X" }, uint64(1), false },
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {

conn_test.go

@@ -10,12 +10,13 @@ func TestConn_BuildSingleActionPacketRaw(t *testing.T) {
 
 	tests := []struct {
 		name    string
-		args    []string
+		args    []any
 		wantRaw string
 		wantErr bool
 	}{
-		{ "GetX", []string { "GET", "X" }, "*1\n~2\n3\nGET\n1\nX\n", false },
-		{ "SysInfoProtocol", []string { "SYS", "INFO", "PROTOCOL" }, "*1\n~3\n3\nSYS\n4\nINFO\n8\nPROTOCOL\n", false },
+		{ "SetX", []any { "SET", "X", 100 }, "*1\n~3\n3\nSET\n1\nX\n3\n100\n", false },
+		{ "GetX", []any { "GET", "X" }, "*1\n~2\n3\nGET\n1\nX\n", false },
+		{ "SysInfoProtocol", []any { "SYS", "INFO", "PROTOCOL" }, "*1\n~3\n3\nSYS\n4\nINFO\n8\nPROTOCOL\n", false },
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {

connpool.go

@@ -41,12 +41,14 @@ func NewConnPool(remote *net.TCPAddr, opts ConnPoolOptions) *ConnPool {
 		opts.Cap = int64(runtime.NumCPU()) * 2
 	}
 
-	return &ConnPool{
+	cp := &ConnPool{
 		opened:       0,
 		available:    make(chan *Conn, opts.Cap),
 		remote:       remote,
 		opts: opts,
 	}
+
+	return cp
 }
 
 func (c *ConnPool) OpenedConns() int64 {

connpool_actions.go

@@ -158,7 +158,7 @@ func (c *ConnPool) Exec(ctx context.Context, packet *QueryPacket) ([]response.Re
 	return conn.Exec(ctx, packet)
 }
 
-func (c *ConnPool) ExecSingleActionPacketRaw(segments ...string) (response.ResponseEntry, error) {
+func (c *ConnPool) ExecSingleActionPacketRaw(segments ...any) (response.ResponseEntry, error) {
 	conn, err := c.popConn(false)
 	if err != nil {
 		return response.EmptyResponseEntry, fmt.Errorf("*ConnPool.ExecSingleActionPacketRaw(): %w", err)