yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2026-01-06 04:26:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

MINOR - Clean Automator owner migration (#17294)

Browse Source 
* MINOR - Clean Automator owner migration

* MINOR - Clean Automator owner migration
This commit is contained in:
Pere Miquel Brull 2024-08-06 09:02:09 +02:00 committed by GitHub
parent 7dd65fc7e0
commit 744ca6227c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
openmetadata-service/src/main/java/org/openmetadata/service/migration/utils/v150/MigrationUtil.java
View File

@ -37,6 +37,7 @@ public class MigrationUtil {
private static final String QUERY_AUTOMATOR =
"SELECT json FROM ingestion_pipeline_entity where appType = 'Automator'";
private static final String ADD_OWNER_ACTION = "AddOwnerAction";
private static final String OWNER_KEY = "owner";
/**
* We need to update the `AddOwnerAction` action in the automator to have a list of owners
@ -63,14 +64,17 @@ public class MigrationUtil {
actions.forEach(
action -> {
JsonObject actionObj = (JsonObject) action;
if (ADD_OWNER_ACTION.equals(actionObj.getString("type"))) {
JsonObject owner = actionObj.getJsonObject("owner");
// Only process the automations with AddOwnerAction that still have the
// `owner` key present
if (ADD_OWNER_ACTION.equals(actionObj.getString("type"))
&& actionObj.containsKey(OWNER_KEY)) {
JsonObject owner = actionObj.getJsonObject(OWNER_KEY);
JsonArrayBuilder owners = Json.createArrayBuilder();
owners.add(owner);
actionObj =
Json.createObjectBuilder(actionObj)
.add("owners", owners)
.remove("owner")
.remove(OWNER_KEY)
.build();
}
updatedActions.add(actionObj);