Accept routing hints in comments (pg-sharding#317)

* Accept routing hints in comments
CBists · Nov 13, 2023 · 3847399 · 3847399
1 parent f2b1424
commit 3847399
Show file tree

Hide file tree

Showing 10 changed files with 122 additions and 90 deletions.
diff --git a/pkg/mock/conn/mock_instance.go b/pkg/mock/conn/mock_instance.go
diff --git a/pkg/mock/conn/raw_mock.go b/pkg/mock/conn/raw_mock.go
diff --git a/pkg/mock/pgx/mock_pgx_tx.go b/pkg/mock/pgx/mock_pgx_tx.go
diff --git a/pkg/mock/pgx/mock_pgxconn_iface.go b/pkg/mock/pgx/mock_pgxconn_iface.go
diff --git a/pkg/mock/shard/mock_shard.go b/pkg/mock/shard/mock_shard.go
diff --git a/router/client/client_test.go b/router/client/client_test.go
@@ -8,6 +8,7 @@ import (
 	"github.com/golang/mock/gomock"
 	"github.com/jackc/pgx/v5/pgproto3"
 	"github.com/pg-sharding/spqr/router/client"
+	"go.uber.org/mock/gomock"
 
 	"github.com/golang/mock/gomock"
 	"github.com/pg-sharding/spqr/pkg/conn"

diff --git a/router/frontend.go b/router/frontend.go
@@ -48,7 +48,6 @@ func deparseRouteHint(rst relay.RelayStateMgr, params map[string]string, dataspa
 		}
 
 		meta := qrouter.NewRoutingMetadataContext(krs, rls, dataspace, nil)
-
 		ds, err := rst.QueryRouter().DeparseKeyWithRangesInternal(context.TODO(), val, meta)
 		if err != nil {
 			return nil, err