better

lib/bouncer/bouncers/v2/bouncer.go

@@ -16,7 +16,7 @@ func clientFail(ctx *backends.Context, client fed.ReadWriter, err perror.Error)
 	_ = client.WritePacket(ctx.Packet)
 }
 
-func Bounce(client, server fed.ReadWriter, initialPacket fed.Packet) (clientError error, serverError error) {
+func Bounce(client, server fed.ReadWriter, initialPacket fed.Packet) (packet fed.Packet, clientError error, serverError error) {
 	ctx := backends.Context{
 		Server: server,
 		Packet: initialPacket,
@@ -31,5 +31,7 @@ func Bounce(client, server fed.ReadWriter, initialPacket fed.Packet) (clientErro
 		clientFail(&ctx, client, perror.Wrap(serverError))
 	}
 
+	packet = ctx.Packet
+
 	return
 }

lib/gat/pool/pool.go

@@ -364,7 +364,7 @@ func (T *Pool) serve(client *Client, initialized bool) error {
 			err, serverErr = Pair(T.options, client, server)
 		}
 		if err == nil && serverErr == nil {
-			err, serverErr = bouncers.Bounce(client.GetReadWriter(), server.GetReadWriter(), packet)
+			packet, err, serverErr = bouncers.Bounce(client.GetReadWriter(), server.GetReadWriter(), packet)
 		}
 		if serverErr != nil {
 			return serverErr

lib/gsql/query_test.go

@@ -58,7 +58,7 @@ func TestQuery(t *testing.T) {
 	if err != nil {
 		t.Error(err)
 	}
-	clientErr, serverErr := bouncers.Bounce(client, server, initial)
+	_, clientErr, serverErr := bouncers.Bounce(client, server, initial)
 	if clientErr != nil {
 		t.Error(clientErr)
 	}

test/runner.go

@@ -108,7 +108,7 @@ func (T *Runner) runModeL1(dialer dialer.Dialer, client *gsql.Client) error {
 			return err
 		}
 
-		clientErr, serverErr := bouncers.Bounce(client, server, p)
+		_, clientErr, serverErr := bouncers.Bounce(client, server, p)
 		if clientErr != nil {
 			return clientErr
 		}