Add default_route_behaviour setting (#145)

To set up failed-to-match logic in router
pg-sharding · Jun 2, 2023 · c154e38 · c154e38
1 parent 23fd2a9
commit c154e38
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/pkg/config/router.go b/pkg/config/router.go
@@ -55,7 +55,8 @@ type Router struct {
 }
 
 type QRouter struct {
-	MulticastUnroutableInsertStatement bool `json:"multicast_unroutable_insert_statement" toml:"multicast_unroutable_insert_statement" yaml:"multicast_unroutable_insert_statement"`
+	MulticastUnroutableInsertStatement bool   `json:"multicast_unroutable_insert_statement" toml:"multicast_unroutable_insert_statement" yaml:"multicast_unroutable_insert_statement"`
+	DefaultRouteBehaviour              string `json:"default_route_behaviour" toml:"default_route_behaviour" yaml:"default_route_behaviour"`
 }
 
 type BackendRule struct {

diff --git a/router/pkg/qrouter/proxy_routing.go b/router/pkg/qrouter/proxy_routing.go
@@ -670,7 +670,12 @@ func (qr *ProxyQrouter) Route(ctx context.Context, parsedStmt *pgquery.ParseResu
 	}
 
 	if route == nil {
-		return MultiMatchState{}, nil
+		switch qr.cfg.DefaultRouteBehaviour {
+		case "BLOCK":
+			return SkipRoutingState{}, fmt.Errorf("failed to match query to any sharding rule")
+		default:
+			return MultiMatchState{}, nil
+		}
 	}
 
 	spqrlog.Logger.Printf(spqrlog.DEBUG1, "parsed shard route %+v", route)