D-Net

package eu.dnetlib.data.mdstore.modular.inspector;

import static eu.dnetlib.miscutils.collections.MappedCollection.listMap;

import java.util.List;

import javax.annotation.Resource;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Controller;

import org.springframework.ui.Model;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RequestMethod;

import org.springframework.web.bind.annotation.RequestParam;

import com.google.common.collect.Iterables;

import com.google.common.collect.Lists;

import eu.dnetlib.data.mdstore.MDStoreServiceException;

import eu.dnetlib.data.mdstore.modular.MDStoreFeeder;

import eu.dnetlib.data.mdstore.modular.connector.MDStore;

import eu.dnetlib.data.mdstore.modular.connector.MDStoreDao;

import eu.dnetlib.data.mdstore.modular.connector.MDStoreManagerInfo;

import eu.dnetlib.data.mdstore.modular.connector.MDStoreTransactionManager;

import eu.dnetlib.data.mdstore.modular.mongodb.MongoMDStore;

import eu.dnetlib.enabling.inspector.AbstractInspectorController;

import eu.dnetlib.enabling.resultset.ResultSetListener;

import eu.dnetlib.miscutils.collections.MappedCollection;

import eu.dnetlib.miscutils.functional.UnaryFunction;

import eu.dnetlib.miscutils.functional.string.EscapeHtml;

@Controller

public class MDStoreInspector extends AbstractInspectorController {

	private static final Log log = LogFactory.getLog(MDStoreInspector.class); // NOPMD by marko on 11/24/08 5:02 PM

	@Resource(name = "mongodbMDStoreDao")

	private MDStoreDao mdstoreDao;

	@Autowired

	private MDStoreTransactionManager transactionManager;

	@Resource

	private MDStoreFeeder feeder;

	class MDStoreDescription {

		private String id;

		private int size;

		public MDStoreDescription(final String id, final int size) {

			super();

			this.id = id;

			this.size = size;

		}

		public String getFormat() {

			return mdstoreDao.getMDStore(id).getFormat();

		}

		public String getLayout() {

			return mdstoreDao.getMDStore(id).getLayout();

		}

		public String getInterpretation() {

			return mdstoreDao.getMDStore(id).getInterpretation();

		}

		public boolean getIndexed() {

			String currentid;

			try {

				currentid = transactionManager.getMDStoreCollection(id);

				return ((MongoMDStore) mdstoreDao.getMDStore(currentid)).isIndexed();

			} catch (MDStoreServiceException e) {

				return false;

			}

		}

		public String getId() {

			return id;

		}

		public void setId(final String id) {

			this.id = id;

		}

		public int getSize() {

			return size;

		}

		public void setSize(final int size) {

			this.size = size;

		}

	}

	@RequestMapping(value = "/inspector/mdstores.do")

	public void mdstores(final Model model) {

		model.addAttribute("mdstores", MappedCollection.listMap(mdstoreDao.listMDStores(), new UnaryFunction<MDStoreDescription, String>() {

			@Override

			public MDStoreDescription evaluate(final String arg) {

				try {

					String currentid = transactionManager.getMDStoreCollection(arg);

					MDStore mdstore = mdstoreDao.getMDStore(currentid);

					return new MDStoreDescription(arg, mdstore.getSize());

				} catch (MDStoreServiceException e) {

					return null;

				}

			}

		}));

	}

	@RequestMapping(value = "/inspector/mdstore.do", method = RequestMethod.GET)

	public void mdstore(final Model model,

			@RequestParam("id") final String id,

			@RequestParam(value = "start", required = false) final Integer startParam,

			@RequestParam(value = "regex", required = false) final String regex) throws MDStoreServiceException {

		int pageSize = 10;

		int start = 0;

		if (startParam != null) {

			start = startParam;

		}

		String currentId = transactionManager.getMDStoreCollection(id);

		MDStore mdstore = mdstoreDao.getMDStore(currentId);

		ResultSetListener rs = mdstore.deliver(null, null, regex);

		List<String> page = rs.getResult(1 + start, start + pageSize);

		model.addAttribute("id", id);

		model.addAttribute("start", start);

		model.addAttribute("regex", regex);

		model.addAttribute("nextPage", start + pageSize);

		model.addAttribute("prevPage", Math.max(0, start - pageSize));

		model.addAttribute("size", rs.getSize());

		model.addAttribute("page", listMap(page, new EscapeHtml()));

	}

	@RequestMapping(value = "/inspector/mdstore.do", method = RequestMethod.POST)

	public String bulkReplace(final Model model,

			@RequestParam("id") final String id,

			@RequestParam("regex") final String regex,

			@RequestParam("replace") final String replace,

			@RequestParam(value = "checkReplace", required = false) final Boolean checkReplace) throws MDStoreServiceException {

		log.debug("regex: " + regex);

		log.debug("replace: " + replace);

		String currentId = transactionManager.getMDStoreCollection(id);

		MongoMDStore mdstore = (MongoMDStore) mdstoreDao.getMDStore(currentId);

		boolean replaceEnable = (checkReplace != null) && (checkReplace == true);

		if (replaceEnable) {

			mdstore.replace(regex, replace);

		} else {

			model.addAttribute("regex", regex);

		}

		return "redirect:mdstore.do?id=" + id;

	}

	@RequestMapping(value = "/inspector/mdstoreEditResult.do")

	public void mdstoreEditRecord(final Model model,

			@RequestParam("id") final String id,

			@RequestParam("start") final int start,

			@RequestParam("index") final int index) throws MDStoreServiceException {

		String currentId = transactionManager.getMDStoreCollection(id);

		MDStore mdstore = mdstoreDao.getMDStore(currentId);

		ResultSetListener rs = mdstore.deliver(null, null, null);

		List<String> page = rs.getResult(1 + index + start, 1 + index + start);

		model.addAttribute("id", id);

		model.addAttribute("start", start);

		model.addAttribute("record", new EscapeHtml().evaluate(Iterables.getOnlyElement(page)));

	}

	@RequestMapping(value = "/inspector/mdstoreSaveRecord.do")

	public String mdstoreSaveRecord(final Model model,

			@RequestParam("id") final String id,

			@RequestParam("start") final int start,

			@RequestParam("record") final String record) {

		MDStore mdstore = mdstoreDao.getMDStore(id);

		mdstore.feed(Lists.newArrayList(record), true);

		return "redirect:mdstore.do?id=" + id + "&start=" + start;

	}

	@RequestMapping(value = "/inspector/mdstoreDeleteRecord.do")

	public String mdstoreDeleteRecord(final Model model,

			@RequestParam("id") final String id,

			@RequestParam("start") final int start,

			@RequestParam("index") final int index) throws MDStoreServiceException {

		String currentId = transactionManager.getMDStoreCollection(id);

		MDStore mdstore = mdstoreDao.getMDStore(currentId);

		ResultSetListener rs = mdstore.deliverIds(null, null, null);

		List<String> page = rs.getResult(1 + index + start, 1 + index + start);

		String recordId = Iterables.getOnlyElement(page);

		log.fatal("deleting record " + recordId);

		mdstore.deleteRecord(recordId);

		return "redirect:mdstore.do?id=" + id + "&start=" + start;

	}

	@RequestMapping(value = "/inspector/mdstoresRefreshSizes.do")

	public String mdstoresRefreshSizes(final Model model) throws MDStoreServiceException {

		for (String mdId : mdstoreDao.listMDStores()) {

			String currentId = transactionManager.getMDStoreCollection(mdId);

			feeder.touchSize(mdId, mdstoreDao.getMDStore(currentId).getSize());

		}

		return "redirect:mdstores.do";

	}

	@RequestMapping(value = "/inspector/ensure.do")

	public String mdstoreEnsureIndex(final Model model, @RequestParam("id") final String id) throws MDStoreServiceException {

		String currentId = transactionManager.getMDStoreCollection(id);

		MongoMDStore mdStore = (MongoMDStore) mdstoreDao.getMDStore(currentId);

		log.info("manual ensureIndex for mdId: " + id);

		mdStore.ensureIndices();

		return "redirect:mdstores.do";

	}

	@RequestMapping(value = "/inspector/infoTransaction.do")

	public void mdstoreInfoTransaction(final Model model, @RequestParam("id") final String id) throws MDStoreServiceException {

		MDStoreManagerInfo info = transactionManager.getInfoForCurrentMdStore(id);

		model.addAttribute("info", info);

	}

	@RequestMapping(value = "/inspector/dropUsedCollection.do")

	public String dropUsedCollection(final Model model, @RequestParam("mdId") final String mdId, @RequestParam("id") final String id)

			throws MDStoreServiceException {

		transactionManager.dropUsed(mdId, id);

		return "redirect:mdstores.do?id=" + mdId;

	}

	@RequestMapping(value = "/inspector/invalidTransactionCollection.do")

	public String invalidTransactionCollection(final Model model, @RequestParam("mdId") final String mdId, @RequestParam("id") final String id)

			throws MDStoreServiceException {

		transactionManager.dropTransaction(mdId, id);

		return "redirect:mdstores.do?id=" + mdId;

	}

}