package cn.com.smartbi;
import java.sql.Connection;
import java.sql.Statement;
import javax.servlet.http.HttpSession;
import org.apache.log4j.Logger;
import smartbi.daq.excelimport.DefaultRowDataImporter;
import smartbi.daq.excelimport.ImportFileConfig;
import smartbi.daq.excelimport.RowData;
import smartbi.daq.excelimport.RowDataImporterResult;
import smartbi.net.sf.json.JSONArray;
import smartbi.net.sf.json.JSONObject;
import smartbi.state.IStateModule;
import smartbi.state.StateModule;
import smartbi.usermanager.User;
import smartbi.usermanager.UserManagerModule;
import smartbi.util.StringUtil;
public class MyRowDataImporter extends DefaultRowDataImporter {
	private static final Logger LOG = Logger.getLogger(MyRowDataImporter.class);
	
	/**
	 * 执行行数据导入逻辑,删除对应用户原有的记录
	 */
	@Override
	public RowDataImporterResult process(RowData rowData, ImportFileConfig config) throws Exception {
		if(!isDeleteRows()){
			JSONObject jObj = config.getConfig();
			User user = (User) UserManagerModule.getInstance().getCurrentUser();
			String username = user.getName();
			String column = "";
			String dataprocessMode = jObj.optString("dataprocessMode");
			JSONArray json = jObj.optJSONArray("columnConfigs");
			for(int i=0;i<json.length();i++){
				JSONObject job = json.getJSONObject(i);
				if(job.has("javaClassName")&&"smartbi.daq.excelimport.CurrentUserIDCellDataGenerator".equals(job.getString("javaClassName"))){
					column = job.getString("fieldName"); //获取用户ID字段
				}
			}
			//新增的数据处理模式
			if (StringUtil.equals("UserFull", dataprocessMode)) {
				String tableName = jObj.optString("tableName");
				String deleteSql = "delete from " + tableName +" where "+column+"='"+username+"'";
				// String dsId = jObj.optString("dataSourceId");
				Connection conn = config.getConn();
				try {
					Statement stat = conn.createStatement();
					stat.execute(deleteSql);
					setIsDeleteRows("true");  //由于每导入一行数据会执行该方法，删除数据只需要执行一次，故加此状态判断
				} catch (Exception e) {
					LOG.error(e.getMessage(), e);
				}
			}
		}
		return super.process(rowData, config);
	}
	
	//用户全量excel导入时,在处理每行数据时,判断是否已经处理过原有数据
	public boolean isDeleteRows(){
		IStateModule stateModule = StateModule.getInstance();
		HttpSession session = stateModule.getSession();
		String isDeleteRows = (String) session.getAttribute("isDeleteRows");
		if("true".equals(isDeleteRows)){
			return true;
		}
		return false;
	}
	public void setIsDeleteRows(String attr){
		IStateModule stateModule = StateModule.getInstance();
		HttpSession session = stateModule.getSession();
		session.setAttribute("isDeleteRows", attr);
	}
	
}