fix: db connector type fix (#3560)

TBD54566975 · Nov 28, 2024 · 6d25f64 · 6d25f64
1 parent 66c3518
commit 6d25f64
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/internal/schema/database.go b/internal/schema/database.go
@@ -187,6 +187,12 @@ func DatabaseConnectorFromProto(s *schemapb.DatabaseConnector) DatabaseConnector
 	switch s := s.Value.(type) {
 	case *schemapb.DatabaseConnector_DsnDatabaseConnector:
 		return &DSNDatabaseConnector{DSN: s.DsnDatabaseConnector.Dsn}
+	case *schemapb.DatabaseConnector_AwsiamAuthDatabaseConnector:
+		return &AWSIAMAuthDatabaseConnector{
+			Username: s.AwsiamAuthDatabaseConnector.Username,
+			Endpoint: s.AwsiamAuthDatabaseConnector.Endpoint,
+			Database: s.AwsiamAuthDatabaseConnector.Database,
+		}
 	default:
 		panic(fmt.Sprintf("unknown database connector type: %T", s))
 	}