diff --git a/openbas-api/src/main/java/io/openbas/migration/V3_43__Payloads_default_values.java b/openbas-api/src/main/java/io/openbas/migration/V3_43__Payloads_default_values.java new file mode 100644 index 0000000000..ceb82c50ef --- /dev/null +++ b/openbas-api/src/main/java/io/openbas/migration/V3_43__Payloads_default_values.java @@ -0,0 +1,24 @@ +package io.openbas.migration; + +import org.flywaydb.core.api.migration.BaseJavaMigration; +import org.flywaydb.core.api.migration.Context; +import org.springframework.stereotype.Component; + +import java.sql.Connection; +import java.sql.Statement; + +@Component +public class V3_43__Payloads_default_values extends BaseJavaMigration { + + @Override + public void migrate(Context context) throws Exception { + Connection connection = context.getConnection(); + Statement select = connection.createStatement(); + + select.executeUpdate("UPDATE payloads SET payload_source = 'MANUAL' WHERE payload_source IS NULL;"); + select.executeUpdate("UPDATE payloads SET payload_status = 'UNVERIFIED' WHERE payload_source IS NULL;"); + + select.execute("ALTER TABLE payloads ALTER COLUMN payload_source SET DEFAULT 'MANUAL';"); + select.execute("ALTER TABLE payloads ALTER COLUMN payload_status SET DEFAULT 'UNVERIFIED';"); + } +}