li.strolch.persistence.api.UpdateActivitiesCommand Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of agent Show documentation
Strolch Agent which is the runtime for Strolch
The newest version!
/*
 * Copyright 2013 Robert von Burg 
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package li.strolch.persistence.api;

import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.List;

import li.strolch.agent.api.ActivityMap;
import li.strolch.exception.StrolchException;
import li.strolch.model.activity.Activity;
import li.strolch.service.api.Command;
import li.strolch.utils.dbc.DBC;

/**
 * @author Robert von Burg 
 */
public class UpdateActivitiesCommand extends Command {

	private List activities = new ArrayList<>();
	private final List replaced = new ArrayList<>();

	public UpdateActivitiesCommand(StrolchTransaction tx) {
		super(tx);
	}

	/**
	 * @param activities
	 * 		the activities to set for updating
	 */
	public void setActivities(List activities) {
		this.activities = activities;
	}

	/**
	 * @param activity
	 * 		the activity to add for updating
	 */
	public void addActivity(Activity activity) {
		this.activities.add(activity);
	}

	@Override
	public void validate() {
		DBC.PRE.assertNotNull("Activities may not be null!", this.activities);
	}

	@Override
	public void doCommand() {

		this.activities.forEach(activity -> tx().lock(activity));

		ActivityMap activityMap = tx().getActivityMap();

		this.activities.forEach(activity -> {
			Activity replaced = activityMap.getBy(tx(), activity.getType(), activity.getId());
			if (this.replaced == null) {
				String msg = "The Activity {0} can not be updated as it does not exist!!";
				msg = MessageFormat.format(msg, activity.getLocator());
				throw new StrolchException(msg);
			}

			activityMap.update(tx(), activity);
			this.replaced.add(replaced);
		});
	}

	@Override
	public void undo() {
		if (tx().isRollingBack() && !this.replaced.isEmpty()) {
			this.replaced.forEach(activity -> {
				if (tx().isVersioningEnabled())
					tx().getActivityMap().undoVersion(tx(), activity);
				else
					tx().getActivityMap().update(tx(), activity);
			});
		}
	}
}