CustomMigrationCallbackExample.java

package com.emc.storageos.db.client.upgrade.callbacks;

import java.net.URI;
import java.util.Iterator;
import java.util.List;

import com.emc.storageos.db.client.DbClient;
import com.emc.storageos.db.client.model.Host;
import com.emc.storageos.db.client.upgrade.BaseCustomMigrationCallback;
import com.emc.storageos.svcs.errorhandling.resources.MigrationCallbackException;
/**
 * Migration callback example.
 * 
 */

public class CustomMigrationCallbackExample extends BaseCustomMigrationCallback {
	private static final int MIN_LABEL_LENGTH = 2;

	@Override
	public void process() throws MigrationCallbackException {
		this.validateHostLabelLength();
	}
	
	private void validateHostLabelLength() throws MigrationCallbackException {
		try {
			DbClient dbClient = this.getDbClient();
			List<URI> hostIds = dbClient.queryByType(Host.class, true);
			Iterator<Host> hosts = dbClient.queryIterativeObjects(Host.class, hostIds, true);
			while (hosts.hasNext()) {
				Host host = hosts.next();
				if (host.getLabel()!=null && host.getLabel().length()<MIN_LABEL_LENGTH) {
					String errorMsg = String.format("%s failed: invalid label length %s(%s)", this.getName(), Host.class.getSimpleName(), host.getId().toString());
					throw new MigrationCallbackException(errorMsg, new IllegalStateException());
				}
			}

		} catch (Exception e) {
			String errorMsg = String.format("%s encounter unexpected error %s", this.getName(), e.getMessage());
			throw new MigrationCallbackException(errorMsg, e);
		}
	}
}