Skip to content

Commit 4f57727

Browse files
Updating methods for get() and batchGet() APIs to filter by deleted_ts (#544)
1 parent cd549bb commit 4f57727

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

dao-impl/ebean-dao/src/main/java/com/linkedin/metadata/dao/utils/SQLStatementUtils.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ public class SQLStatementUtils {
118118

119119
private static final String SQL_GET_ALL_COLUMNS =
120120
"SELECT COLUMN_NAME FROM information_schema.COLUMNS WHERE TABLE_SCHEMA = database() AND TABLE_NAME = '%s'";
121-
121+
122122
private static final String SQL_URN_EXIST_TEMPLATE = "SELECT urn FROM %s WHERE urn = '%s' AND deleted_ts IS NULL";
123123

124124
private static final String INSERT_LOCAL_RELATIONSHIP = "INSERT INTO %s (metadata, source, destination, source_type, "
@@ -209,6 +209,8 @@ public static <ASPECT extends RecordTemplate> String createAspectReadSql(@Nonnul
209209
stringBuilder.append(String.format(sqlTemplate, columnName, tableName, columnName));
210210
stringBuilder.append(urnList);
211211
stringBuilder.append(RIGHT_PARENTHESIS);
212+
stringBuilder.append(" AND ");
213+
stringBuilder.append(DELETED_TS_IS_NULL_CHECK);
212214
return stringBuilder.toString();
213215
}
214216

dao-impl/ebean-dao/src/test/java/com/linkedin/metadata/dao/utils/SQLStatementUtilsTest.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,14 +102,16 @@ public void testCreateAspectReadSql() {
102102
"SELECT urn, a_aspectfoo, lastmodifiedon, lastmodifiedby "
103103
+ "FROM metadata_entity_foo "
104104
+ "WHERE JSON_EXTRACT(a_aspectfoo, '$.gma_deleted') IS NULL "
105-
+ "AND urn IN ('urn:li:foo:1', 'urn:li:foo:2')";
105+
+ "AND urn IN ('urn:li:foo:1', 'urn:li:foo:2') "
106+
+ "AND deleted_ts IS NULL";
106107
assertEquals(SQLStatementUtils.createAspectReadSql(AspectFoo.class, set, false, false), expectedSql);
107108

108109
//test when includedSoftDeleted is true
109110
expectedSql =
110111
"SELECT urn, a_aspectfoo, lastmodifiedon, lastmodifiedby "
111112
+ "FROM metadata_entity_foo "
112-
+ "WHERE urn IN ('urn:li:foo:1', 'urn:li:foo:2')";
113+
+ "WHERE urn IN ('urn:li:foo:1', 'urn:li:foo:2') "
114+
+ "AND deleted_ts IS NULL";
113115
assertEquals(SQLStatementUtils.createAspectReadSql(AspectFoo.class, set, true, false), expectedSql);
114116
}
115117

0 commit comments

Comments
 (0)