Protei/PromucFlow_constructor
2
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Added a migration to old rest api datasources which adds the new mandatory key isSendSessionEnabled with value N (#2388)

Browse Source
This commit is contained in:
Trisha Anand 2020-12-29 16:29:28 +05:30 committed by GitHub
parent 4c4d0b72f9
commit 735a4398b2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 49 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
app/server/appsmith-server/src/main/java/com/appsmith/server/migrations/DatabaseChangelog.java
View File

@ -2,7 +2,9 @@ package com.appsmith.server.migrations;
import com.appsmith.external.models.BaseDomain;
import com.appsmith.external.models.DBAuth;
import com.appsmith.external.models.DatasourceConfiguration;
import com.appsmith.external.models.Policy;
import com.appsmith.external.models.Property;
import com.appsmith.server.acl.AppsmithRole;
import com.appsmith.server.constants.FieldName;
import com.appsmith.server.domains.Action;
@ -1516,4 +1518,51 @@ public class DatabaseChangelog {
);
}
}
@ChangeSet(order = "047", id = "add-isSendSessionEnabled-key-for-datasources", author = "")
public void addIsSendSessionEnabledPropertyInDatasources(MongoTemplate mongoTemplate) {
String keyName = "isSendSessionEnabled";
Plugin restApiPlugin = mongoTemplate.findOne(
query(where("packageName").is("restapi-plugin")),
Plugin.class
);
final org.springframework.data.mongodb.core.query.Query query = query(where("pluginId").is(restApiPlugin.getId()));
for (Datasource datasource : mongoTemplate.find(query, Datasource.class)) {
// Find if the datasource should be updated with the new key
Boolean updateRequired = false;
if (datasource.getDatasourceConfiguration() == null) {
updateRequired = true;
datasource.setDatasourceConfiguration(new DatasourceConfiguration());
datasource.getDatasourceConfiguration().setProperties(new ArrayList<>());
} else if (datasource.getDatasourceConfiguration().getProperties() == null) {
updateRequired = true;
datasource.getDatasourceConfiguration().setProperties(new ArrayList<>());
} else {
List<Property> properties = datasource.getDatasourceConfiguration().getProperties();
Optional<Property> isSendSessionEnabledOptional = properties
.stream()
.filter(property -> property.getKey().equals(keyName))
.findFirst();
if (!isSendSessionEnabledOptional.isPresent()) {
updateRequired = true;
}
}
// If the property does not exist, add the same.
if (updateRequired) {
Property newProperty = new Property();
newProperty.setKey(keyName);
newProperty.setValue("N");
datasource.getDatasourceConfiguration().getProperties().add(newProperty);
mongoTemplate.save(datasource);
}
}
}
}