diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/UsageRepository.java b/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/UsageRepository.java index b9fda6369d9..1d9eb076f8b 100644 --- a/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/UsageRepository.java +++ b/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/UsageRepository.java @@ -134,8 +134,12 @@ public class UsageRepository { // Insert usage record insertToUsageRepository(method, entityId, entityType, usage); Table updated = Entity.getEntity(Entity.TABLE, UUID.fromString(entityId), fields, Include.ALL); - insertToUsageRepository(method, table.getDatabaseSchema().getId().toString(), Entity.DATABASE_SCHEMA, usage); - insertToUsageRepository(method, table.getDatabase().getId().toString(), Entity.DATABASE, usage); + dao.usageDAO() + .insertOrUpdateCount( + usage.getDate(), table.getDatabaseSchema().getId().toString(), Entity.DATABASE_SCHEMA, usage.getCount()); + dao.usageDAO() + .insertOrUpdateCount( + usage.getDate(), table.getDatabase().getId().toString(), Entity.DATABASE, usage.getCount()); dao.usageDAO().computePercentile(entityType, usage.getDate()); ChangeDescription change =