diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/DatabaseSchemaRepository.java b/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/DatabaseSchemaRepository.java index 12c980d2db9..4d44a45d679 100644 --- a/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/DatabaseSchemaRepository.java +++ b/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/DatabaseSchemaRepository.java @@ -32,7 +32,7 @@ import org.openmetadata.service.util.EntityUtil.Fields; import org.openmetadata.service.util.FullyQualifiedName; public class DatabaseSchemaRepository extends EntityRepository { - private static final String DATABASE_SCHEMA_UPDATE_FIELDS = "owner"; + private static final String DATABASE_SCHEMA_UPDATE_FIELDS = "owner,tags"; private static final String DATABASE_SCHEMA_PATCH_FIELDS = DATABASE_SCHEMA_UPDATE_FIELDS; public DatabaseSchemaRepository(CollectionDAO dao) { @@ -77,6 +77,8 @@ public class DatabaseSchemaRepository extends EntityRepository { addRelationship( database.getId(), schema.getId(), database.getType(), Entity.DATABASE_SCHEMA, Relationship.CONTAINS); storeOwner(schema, schema.getOwner()); + // Add tag to databaseSchema relationship + applyTags(schema); } private List getTables(DatabaseSchema schema) throws IOException { diff --git a/openmetadata-service/src/test/java/org/openmetadata/service/resources/databases/DatabaseSchemaResourceTest.java b/openmetadata-service/src/test/java/org/openmetadata/service/resources/databases/DatabaseSchemaResourceTest.java index ed017d4ff14..5d94d2c1a3a 100644 --- a/openmetadata-service/src/test/java/org/openmetadata/service/resources/databases/DatabaseSchemaResourceTest.java +++ b/openmetadata-service/src/test/java/org/openmetadata/service/resources/databases/DatabaseSchemaResourceTest.java @@ -78,7 +78,7 @@ class DatabaseSchemaResourceTest extends EntityResourceTest