step.migration.tasks.MigrateFunctions Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of step-controller-initialization Show documentation
There is a newer version: 3.16.3
Show newest version
package step.migration.tasks;

import java.util.concurrent.atomic.AtomicInteger;

import org.bson.Document;

import com.mongodb.Block;
import com.mongodb.client.result.UpdateResult;

import step.core.GlobalContext;
import step.core.Version;
import step.migration.MigrationTask;

/**
 * This task migrates the artefact of type 'CallFunction' that have the attribute 'function' declared as string instead of DynamicValue
 * do this only when migrating from 3.4.0 to 3.5.0 or higher
 *
 */
public class MigrateFunctions extends MigrationTask {

	public MigrateFunctions() {
		super(new Version(3,5,0));
	}

	@Override
	public void runUpgradeScript() {
		migrateCallFunction(context);
		migrateGeneralScriptFunction(context);
		migrateGeneralScriptFunctions(context);
	}
	
	private void migrateGeneralScriptFunction(GlobalContext context) {
		logger.info("Searching for keywords of type 'Script' to be migrated...");
		
		com.mongodb.client.MongoCollection functions = context.getMongoClientSession().getMongoDatabase().getCollection("functions");
		
		Document filter = new Document("type", "step.plugins.functions.types.GeneralScriptFunction");
		Document replacement = new Document("$set", new Document("type", "step.plugins.java.GeneralScriptFunction"));
		UpdateResult result = functions.updateMany(filter, replacement);
		
		logger.info("Migrated "+result.getModifiedCount()+" artefacts of type 'step.plugins.functions.types.GeneralScriptFunction'");
	}
	
	private void migrateGeneralScriptFunctions(GlobalContext context) {
		logger.info("Searching for functions of type 'step.plugins.functions.types.GeneralScriptFunction' to be migrated...");
		com.mongodb.client.MongoCollection functions = context.getMongoClientSession().getMongoDatabase().getCollection("functions");
		
		AtomicInteger i = new AtomicInteger();
		Document filterCallFunction = new Document("type", "step.plugins.functions.types.GeneralScriptFunction");
		functions.find(filterCallFunction).forEach(new Block() {

			@Override
			public void apply(Document t) {
				t.replace("type", "step.plugins.java.GeneralScriptFunction");
				Document filter = new Document("_id", t.get("_id"));
				functions.replaceOne(filter, t);
				i.incrementAndGet();
			}
		});
		
		logger.info("Migrated "+i.get()+" functions of type 'step.plugins.functions.types.GeneralScriptFunction'");
	}
	
	// This function migrates the artefact of type 'CallFunction' that have the attribute 'function' declared as string instead of DynamicValue
	// TODO do this only when migrating from 3.4.0 to 3.5.0 or higher
	private void migrateCallFunction(GlobalContext context) {
		logger.info("Searching for artefacts of type 'CallFunction' to be migrated...");
		com.mongodb.client.MongoCollection artefacts = context.getMongoClientSession().getMongoDatabase().getCollection("artefacts");
		
		AtomicInteger i = new AtomicInteger();
		Document filterCallFunction = new Document("_class", "CallFunction");
		artefacts.find(filterCallFunction).forEach(new Block() {

			@Override
			public void apply(Document t) {
				if(t.containsKey("function")) {
					try {
						i.incrementAndGet();
						String function = t.getString("function");
						Document d = new Document();
						d.append("dynamic", false);
						d.append("value", function);
						t.replace("function", d);
						
						Document filter = new Document("_id", t.get("_id"));
						
						artefacts.replaceOne(filter, t);
						logger.info("Migrating "+function+" to "+d.toJson());
					} catch(ClassCastException e) {
						// ignore
					}
				}
			}
		});
		
		logger.info("Migrated "+i.get()+" artefacts of type 'CallFunction'");
	}
	
	@Override
	public void runDowngradeScript() {
		// TODO Auto-generated method stub
		
	}

}