InfoCollectionCommand.java

package org.fastcatsearch.cli.command;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.TreeSet;

import org.fastcatsearch.cli.CommandException;
import org.fastcatsearch.cli.CommandResult;
import org.fastcatsearch.cli.ConsoleSessionContext;
import org.fastcatsearch.cli.command.exception.CollectionNotDefinedException;
import org.fastcatsearch.cli.command.exception.CollectionNotFoundException;
import org.fastcatsearch.ir.IRService;
import org.fastcatsearch.ir.config.CollectionContext;
import org.fastcatsearch.ir.search.CollectionHandler;
import org.fastcatsearch.ir.settings.Schema;
import org.fastcatsearch.ir.settings.SchemaSetting;
import org.fastcatsearch.ir.util.Formatter;
import org.fastcatsearch.service.ServiceManager;

public class InfoCollectionCommand extends CollectionExtractCommand {

	private String[] header = new String[] { "collection name", "total field count", "indexing field count",
			"fileter field count", "group field count", "sort field count", "data source Type", "status", "start date",
			"run time" };

	private ArrayList<Object[]> data = new ArrayList<Object[]>();

	@Override
	public boolean isCommand(String[] cmd) {
		return isCommand(CMD_INFO_COLLECTION, cmd);
	}

	@Override
	public CommandResult doCommand(String[] cmd, ConsoleSessionContext context) throws IOException, CommandException {
		if ((cmd.length == 0) || (cmd.length > 2)) {
			return new CommandResult("invalid Command", CommandResult.Status.SUCCESS);
		} else {
			String inputCollection = "";

			if (cmd.length == 2)
				inputCollection = cmd[1].trim();
			else if (cmd.length == 1) {
				try {
					inputCollection = extractCollection(context);
					checkCollectionExists(inputCollection);
				} catch (CollectionNotDefinedException e) {
					return new CommandResult("Collection Not defined", CommandResult.Status.SUCCESS);
				} catch (CollectionNotFoundException e) {
					return new CommandResult("collection " + inputCollection + " is not exists",
							CommandResult.Status.SUCCESS);
				}
			}

			TreeSet ts = getCollectionList();

			if (ts.size() == 0) {
				return new CommandResult("there is no Collection", CommandResult.Status.SUCCESS);
			}

			// 입력된 컬렉션이 현재 컬렉션 리스트에 있을 때
			if (getCollectionData(inputCollection))
				return new CommandResult(printData(data, header), CommandResult.Status.SUCCESS);
			else
				return new CommandResult("invalid Command", CommandResult.Status.SUCCESS);

		}
	}

	private boolean getCollectionData(String collectionId) {
		IRService irService = ServiceManager.getInstance().getService(IRService.class);
		CollectionContext collectionContext = irService.collectionContext(collectionId);
		Schema schema = collectionContext.schema();
		String sourceReaderType = "";//collectionContext.dataSourceConfig().getReaderType();
		CollectionHandler ch = irService.collectionHandler(collectionId);
		boolean isRunning = (ch == null ? false : true);
		String durationStr = "";
		String strStartTime = "";
		if (ch != null) {
			long startTime = ch.getStartedTime();
			long duration = System.currentTimeMillis() - startTime;
			strStartTime = new Date(startTime).toString();
			durationStr = Formatter.getFormatTime(duration);
		}
		SchemaSetting schemaSetting = schema.schemaSetting();
		addRecord(data, collectionId, schemaSetting.getFieldSettingList().size() + "", schemaSetting.getIndexSettingList().size() + "",
				schemaSetting.getFieldIndexSettingList().size() + "", sourceReaderType, (isRunning ? "Running" : "stop"),
				strStartTime, durationStr);

		return true;
	}

	private void addRecord(List<Object[]> data, String cn, String ftc, String ifc, String ffc,
			String dst, String status, String sdate, String runtime) {
		data.add(new Object[] { cn, ftc, ifc, ffc, dst, status, sdate, runtime });
	}

	// collection name", "total field count", "indexing field count",	"fileter
	// field count", "group field count", "sort field count", "data source
	// Type", "status", "start date",	"run time"
}