From 2c0657fe68042b62600faa9dd764a183beef826f Mon Sep 17 00:00:00 2001
From: Parth Panchal <83201188+parthp2107@users.noreply.github.com>
Date: Tue, 20 Sep 2022 17:38:03 +0530
Subject: [PATCH] Fixed#7251: POST API calls to /usage endpoint are
 inconsistent (#7587)

* Fixed#7251: POST API calls to /usage endpoint are inconsistent

* Fixed#7251: POST API calls to /usage endpoint are inconsistent
---
 .../org/openmetadata/service/jdbi3/UsageRepository.java   | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

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 =