dbutil.java

来自「工厂版本管理系统,STRUTS2框架,用于管理商品的版本,便于有效的控制版本」· Java 代码 · 共 91 行

package com.utstar.fcs.domain;

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

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.bluesky.common.dao.ICommonDao;
import com.bluesky.common.domain.IReposity;
import com.bluesky.util.ApplicationContextUtil;
import com.utstar.fcs.domain.workinstruction.FieldType;
import com.utstar.fcs.domain.workinstruction.StationType;
import com.utstar.fcs.domain.workinstruction.WorkInstruction;
import com.utstar.fcs.domain.workinstruction.WorkInstructionVersion;

public class DBUtil {
	private Log log = LogFactory.getLog(this.getClass());

	private ICommonDao commonDao;
	private IReposity reposity;

	public DBUtil() {
		log.debug(this.getClass() + ".setUp");
		commonDao = (ICommonDao) ApplicationContextUtil.getFactory().getBean(
				"commonDao");
		reposity = (IReposity) ApplicationContextUtil.getFactory().getBean(
				"reposity");
	}

	private String getRandomString() {
		return "random-string";
	}

	private void insertFieldType() {
		java.util.Iterator<FieldType> it = FieldType.getAll().iterator();

		while (it.hasNext()) {
			reposity.save(it.next());
		}
	}

	private void insertWorkInstruction() {
		StationType stationType = StationType.getSample();
		reposity.save(stationType);

		List<StationType> stationTypeList = new ArrayList<StationType>();
		stationTypeList.add(stationType);
		
		WorkInstruction wi = WorkInstruction.getSample(stationTypeList);
		reposity.save(wi);

		WorkInstructionVersion wiv = WorkInstructionVersion.getSample(wi);
		reposity.save(wiv);
	}

	private void insertData() {
		log.debug("begin insert data");

		insertFieldType();
		insertWorkInstruction();

		log.debug("end insert data");
	}

	private void deleteData() {
		log.debug("begin delete data");
		

		String[] tableNames = new String[] { "wi_Field_Type",
				"wi_Field", "wi_string_field","wi_memo_field", "wi_file_field","wi_table_field","wi_table_field_data",
				"wi_Field_Definition",
				"wi_Station_Type", "wi_Station", "wi_Station_Version",
				"wi_Work_Instruction", "wi_Work_Instruction_Version",
				"wi_flow_chart_item" };

		commonDao.truncateDB(tableNames);

		log.debug("end delete data");
	}

	public static void main(String[] argv) {

		DBUtil dbUtil = new DBUtil();

		dbUtil.deleteData();
		dbUtil.insertData();

	}
}