added database type into resolver's

Author: hasirciogluhq

.DS_Store

@@ -13,7 +13,7 @@ type RoutingMetadata map[string]string
 // BackendResolver defines how to find a backend address based on metadata.
 // It is purely a lookup mechanism and knows nothing about the network.
 type BackendResolver interface {
-	Resolve(ctx context.Context, metadata RoutingMetadata) (string, error)
+	Resolve(ctx context.Context, metadata RoutingMetadata, databaseType DatabaseType) (string, error)
 }
 
 // ConnectionHandler defines the interface for handling a client connection.
@@ -41,3 +41,11 @@ type TLSProvider interface {
 	GetCertificate(ctx context.Context) (*tls.Certificate, error)
 	Store(ctx context.Context, certPEM, keyPEM []byte) error
 }
+
+type DatabaseType string
+
+const (
+	DatabaseTypePostgresql DatabaseType = "postgresql"
+	DatabaseTypeMysql      DatabaseType = "mysql"
+	DatabaseTypeScylla     DatabaseType = "scylla"
+)

cmd/proxy/internal/core/types.go

@@ -30,7 +30,7 @@ func NewK8sResolver(clientset *kubernetes.Clientset) *K8sResolver {
 	}
 }
 
-func (r *K8sResolver) Resolve(ctx context.Context, metadata core.RoutingMetadata) (string, error) {
+func (r *K8sResolver) Resolve(ctx context.Context, metadata core.RoutingMetadata, databaseType core.DatabaseType) (string, error) {
 	deploymentID, ok := metadata["deployment_id"]
 	if !ok {
 		return "", fmt.Errorf("metadata missing 'deployment_id' (check connection string format: user.deployment_id[.pool])")
@@ -49,6 +49,10 @@ func (r *K8sResolver) Resolve(ctx context.Context, metadata core.RoutingMetadata
 			continue
 		}
 
+		if labels["xdatabase-proxy-database-type"] != string(databaseType) {
+			continue
+		}
+
 		if labels["xdatabase-proxy-deployment-id"] == deploymentID &&
 			labels["xdatabase-proxy-pooled"] == pooled {
 

cmd/proxy/internal/discovery/kubernetes/resolver.go

@@ -37,7 +37,7 @@ func NewResolver(mappingStr string) (*Resolver, error) {
 	return &Resolver{backends: backends}, nil
 }
 
-func (r *Resolver) Resolve(ctx context.Context, metadata core.RoutingMetadata) (string, error) {
+func (r *Resolver) Resolve(ctx context.Context, metadata core.RoutingMetadata, databaseType core.DatabaseType) (string, error) {
 	deploymentID, ok := metadata["deployment_id"]
 	if !ok {
 		return "", fmt.Errorf("metadata missing 'deployment_id'")

cmd/proxy/internal/discovery/memory/resolver.go

@@ -76,7 +76,7 @@ func (p *PostgresProxy) HandleConnection(clientConn net.Conn) {
 	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
 	defer cancel()
 
-	backendAddr, err := p.Resolver.Resolve(ctx, metadata)
+	backendAddr, err := p.Resolver.Resolve(ctx, metadata, core.DatabaseTypeMysql)
 	if err != nil {
 		logger.Error("Resolution failed", "error", err, "remote_addr", clientConn.RemoteAddr())
 		_ = p.sendErrorResponse(clientConn, &ErrorResponse{