ProblemStatusCodeDataOnDemand.java

package gov.samhsa.consent2share.domain.reference;

import java.security.SecureRandom;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Random;

import javax.validation.ConstraintViolation;
import javax.validation.ConstraintViolationException;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Configurable;
import org.springframework.data.domain.PageRequest;
import org.springframework.stereotype.Component;

@Component
@Configurable
public class ProblemStatusCodeDataOnDemand {

	private Random rnd = new SecureRandom();

	private List<ProblemStatusCode> data;

	@Autowired
	ProblemStatusCodeRepository problemStatusCodeRepository;

	public ProblemStatusCode getNewTransientProblemStatusCode(int index) {
		ProblemStatusCode obj = new ProblemStatusCode();
		setCode(obj, index);
		setCodeSystem(obj, index);
		setCodeSystemName(obj, index);
		setDisplayName(obj, index);
		setOriginalText(obj, index);
		return obj;
	}

	public void setCode(ProblemStatusCode obj, int index) {
		String code = "code_" + index;
		if (code.length() > 250) {
			code = code.substring(0, 250);
		}
		obj.setCode(code);
	}

	public void setCodeSystem(ProblemStatusCode obj, int index) {
		String codeSystem = "codeSystem_" + index;
		if (codeSystem.length() > 250) {
			codeSystem = codeSystem.substring(0, 250);
		}
		obj.setCodeSystem(codeSystem);
	}

	public void setCodeSystemName(ProblemStatusCode obj, int index) {
		String codeSystemName = "codeSystemName_" + index;
		if (codeSystemName.length() > 250) {
			codeSystemName = codeSystemName.substring(0, 250);
		}
		obj.setCodeSystemName(codeSystemName);
	}

	public void setDisplayName(ProblemStatusCode obj, int index) {
		String displayName = "displayName_" + index;
		if (displayName.length() > 250) {
			displayName = displayName.substring(0, 250);
		}
		obj.setDisplayName(displayName);
	}

	public void setOriginalText(ProblemStatusCode obj, int index) {
		String originalText = "originalText_" + index;
		if (originalText.length() > 250) {
			originalText = originalText.substring(0, 250);
		}
		obj.setOriginalText(originalText);
	}

	public ProblemStatusCode getSpecificProblemStatusCode(int index) {
		init();
		if (index < 0) {
			index = 0;
		}
		if (index > (data.size() - 1)) {
			index = data.size() - 1;
		}
		ProblemStatusCode obj = data.get(index);
		Long id = obj.getId();
		return problemStatusCodeRepository.findOne(id);
	}

	public ProblemStatusCode getRandomProblemStatusCode() {
		init();
		ProblemStatusCode obj = data.get(rnd.nextInt(data.size()));
		Long id = obj.getId();
		return problemStatusCodeRepository.findOne(id);
	}

	public boolean modifyProblemStatusCode(ProblemStatusCode obj) {
		return false;
	}

	public void init() {

		int pageNumber = 0;
		int pageSize = 10;
		data = problemStatusCodeRepository.findAll(
				new PageRequest(pageNumber, pageSize)).getContent();
		if (data == null) {
			throw new IllegalStateException(
					"Find entries implementation for 'ProblemStatusCode' illegally returned null");
		}
		if (!data.isEmpty()) {
			return;
		}

		data = new ArrayList<ProblemStatusCode>();
		for (int i = 0; i < 10; i++) {
			ProblemStatusCode obj = getNewTransientProblemStatusCode(i);
			try {
				problemStatusCodeRepository.save(obj);
			} catch (ConstraintViolationException e) {
				StringBuilder msg = new StringBuilder();
				for (Iterator<ConstraintViolation<?>> iter = e
						.getConstraintViolations().iterator(); iter.hasNext();) {
					ConstraintViolation<?> cv = iter.next();
					msg.append("[").append(cv.getConstraintDescriptor())
							.append(":").append(cv.getMessage()).append("=")
							.append(cv.getInvalidValue()).append("]");
				}
				throw new RuntimeException(msg.toString(), e);
			}
			problemStatusCodeRepository.flush();
			data.add(obj);
		}
	}
}