yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-09-27 18:05:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: handle postgres migration error (#13844)

Browse Source 
* fix: handle postgres migration error

* fix: removed MySQL comment
This commit is contained in:
Teddy 2023-11-03 15:03:03 +01:00 committed by GitHub
parent 9d58b56a1c
commit 9d238d142a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
openmetadata-service/src/main/java/org/openmetadata/service/migration/utils/v120/MigrationUtil.java
View File

@ -13,7 +13,9 @@ import org.openmetadata.service.Entity;
import org.openmetadata.service.jdbi3.CollectionDAO; import org.openmetadata.service.jdbi3.CollectionDAO;
import org.openmetadata.service.jdbi3.GlossaryTermRepository; import org.openmetadata.service.jdbi3.GlossaryTermRepository;
import org.openmetadata.service.jdbi3.QueryRepository; import org.openmetadata.service.jdbi3.QueryRepository;
import org.openmetadata.service.resources.databases.DatasourceConfig;
import org.openmetadata.service.util.JsonUtils; import org.openmetadata.service.util.JsonUtils;
import org.postgresql.util.PGobject;
@Slf4j @Slf4j
public class MigrationUtil { public class MigrationUtil {
@ -136,7 +138,15 @@ public class MigrationUtil {
.mapToMap() .mapToMap()
.forEach( .forEach(
row -> { row -> {
GlossaryTerm term = JsonUtils.readValue((String) row.get("json"), GlossaryTerm.class); String jsonRow;
if (Boolean.TRUE.equals(DatasourceConfig.getInstance().isMySQL())) {
jsonRow = (String) row.get("json");
} else {
// Postgres stores JSON as a JSONB, so we can't just cast it as a string
PGobject pgObject = (PGobject) row.get("json");
jsonRow = pgObject.getValue();
}
GlossaryTerm term = JsonUtils.readValue(jsonRow, GlossaryTerm.class);
if (term.getStatus() == GlossaryTerm.Status.DRAFT) { if (term.getStatus() == GlossaryTerm.Status.DRAFT) {
term.setStatus(GlossaryTerm.Status.APPROVED); term.setStatus(GlossaryTerm.Status.APPROVED);
collectionDAO.glossaryTermDAO().update(term); collectionDAO.glossaryTermDAO().update(term);