com.avanza.astrix.versioning.jackson2.JsonMessageMigrator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of astrix-versioning Show documentation
There is a newer version: 2.0.6
/*
 * Copyright 2014 Avanza Bank AB
 *
 * 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 com.avanza.astrix.versioning.jackson2;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import com.fasterxml.jackson.databind.node.ObjectNode;

/**
 * A message migrator is responsible for migration a message on an old version
 * to the current version, and also responsible for migration messages on the
 * current version to an older version. 
 * 
 * @author Elias Lindholm (elilin)
 *
 */
class JsonMessageMigrator {
	
	private Class type;
	private List> migrationsInOrder;
	private List> migrationsInReverseOrder;
	
	public JsonMessageMigrator(Class type,
							   List> migrations) {
		this.type = type;
		this.migrationsInOrder = new ArrayList<>(migrations);
		Collections.sort(this.migrationsInOrder);
		this.migrationsInReverseOrder = new ArrayList<>(migrationsInOrder);
		Collections.reverse(this.migrationsInReverseOrder);
		
	}

	public void upgrade(ObjectNode json, int fromVersion) {
		for (JsonMessageMigrationWithVersion migration : migrationsInOrder) {
			if (migration.getVersion() < fromVersion) {
				continue;
			}
			migration.upgrade(json);
		}
	}
	
	public void downgrade(ObjectNode json, int toVersion) {
		for (JsonMessageMigrationWithVersion migration : migrationsInReverseOrder) {
			if (!(migration.getVersion() >= toVersion)) {
				return;
			}
			migration.downgrade(json);
		}
	}

	public Class getJavaType() {
		return type;
	}
	
	static class JsonMessageMigrationWithVersion implements Comparable> {
		private int version;
		private AstrixJsonMessageMigration migration;

		public JsonMessageMigrationWithVersion(int version,
				AstrixJsonMessageMigration migration) {
			this.version = version;
			this.migration = migration;
		}
		
		public int getVersion() {
			return version;
		}
		
		void downgrade(ObjectNode json) {
			this.migration.downgrade(json);
		}
		
		void upgrade(ObjectNode json) {
			this.migration.upgrade(json);
		}

		@Override
		public int compareTo(JsonMessageMigrationWithVersion other) {
			return getVersion() - other.getVersion();
		}
		
	}
	
	static class Builder {
		private Class javaType;
		private List> migrations = new ArrayList<>();

		public Builder(Class javaType) {
			this.javaType = javaType;
		}
		
		public Builder addMigration(AstrixJsonMessageMigration migration, int fromVersion) {
			migrations.add(new JsonMessageMigrationWithVersion<>(fromVersion, migration));
			return this;
		}
		
		public JsonMessageMigrator build() {
			return new JsonMessageMigrator<>(javaType, migrations);
		}
		
	}
	
}