/jeecg-framework/src/main/java/test/JeecgInitDB.java

package test;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.jeecgframework.web.cgform.entity.config.CgFormFieldEntity;
import org.jeecgframework.web.cgform.entity.config.CgFormHeadEntity;
import org.jeecgframework.web.system.pojo.base.TSAttachment;
import org.jeecgframework.web.system.pojo.base.TSBaseUser;
import org.jeecgframework.web.system.pojo.base.TSDepart;
import org.jeecgframework.web.system.pojo.base.TSFunction;
import org.jeecgframework.web.system.pojo.base.TSLog;
import org.jeecgframework.web.system.pojo.base.TSRole;
import org.jeecgframework.web.system.pojo.base.TSType;
import org.jeecgframework.web.system.pojo.base.TSTypegroup;
import org.jeecgframework.web.system.pojo.base.TSUser;
import freemarker.template.Configuration;
import freemarker.template.DefaultObjectWrapper;
import freemarker.template.Template;


/** 
 * @Description 
 * @ClassName: JeecgInitDB
 * @author tanghan
 * @date 2013-7-19 下午04:24:51  
 */

public class JeecgInitDB {
   
    private static Connection con=null;
    
    
	public static Connection getConnection() throws ClassNotFoundException, SQLException{
		if(con == null){
			Class.forName("com.mysql.jdbc.Driver");
            con=DriverManager.getConnection("jdbc:mysql://127.0.0.1:3306/jeecg", "root","root");
		}
		return con;
	}
	
	public static void main(String[] args) throws Exception {
		Configuration cfg = new Configuration();
		String sql1="select * from t_s_attachment";
		String sql2="select * from t_s_base_user";
		String sql3="select * from t_s_depart";
		String sql4="select * from t_s_role";
		String sql5="select * from t_s_user";
		String sql6="select * from t_s_typegroup";
		String sql7 = "select * from t_s_function";
		String sql8 = "select * from t_s_type";
		String sql9 = "select * from t_s_log limit 100";
		String sql10 = "select * from cgform_field where table_id =";  //此处由于需要更具cgform_head生成，故只能单个生成
		String sql11 = "select * from cgform_head ";
        Statement st=null;
        ResultSet rs=null;
		try {
			cfg.setDirectoryForTemplateLoading(new File("E:/Workspace-jeecg/jeecg-v3-simple-new/src/test"));
			cfg.setObjectWrapper(new DefaultObjectWrapper());
			Template temp = cfg.getTemplate("init.ftl","UTF-8");
            con=getConnection();
            st=con.createStatement();
            rs=st.executeQuery(sql1);
            int i=1;
            Map<String, Object> root = new HashMap<String, Object>();
            List att = new ArrayList();
            while(rs.next())
            {
                 TSAttachment tsAttachment = new TSAttachment();
            	 tsAttachment.setId(i+"");
            	 tsAttachment.setAttachmenttitle(rs.getString("attachmenttitle"));
            	 tsAttachment.setRealpath(rs.getString("attachmenttitle"));
            	 tsAttachment.setSwfpath(rs.getString("swfpath"));
            	 tsAttachment.setExtend(rs.getString("extend"));
            	 att.add(tsAttachment);
                 i++;
            }
            root.put("animals", att);
            
            
            rs=st.executeQuery(sql2);
            i=1;
            List user = new ArrayList();
            while(rs.next())
            {
                 TSBaseUser baseUser = new TSBaseUser();
                 baseUser.setId(i+"");
                 baseUser.setUserKey(rs.getString("userkey"));
                 baseUser.setStatus(rs.getShort("status"));
                 baseUser.setRealName(rs.getString("realname"));
                 baseUser.setUserName(rs.getString("username"));
                 baseUser.setPassword(rs.getString("password"));
                 baseUser.setActivitiSync(rs.getShort("activitisync"));
                 user.add(baseUser);
                 i++;
            }
            root.put("baseuser", user);
            
            
            rs=st.executeQuery(sql3);
            List dep = new ArrayList();
            i=1;
            while(rs.next())
            {
                 TSDepart tsDepart = new TSDepart();
                 tsDepart.setId(i+"");
                 tsDepart.setDepartname(rs.getString("departname"));
                 tsDepart.setDescription(rs.getString("description"));
                 dep.add(tsDepart);
                 i++;
            }
            root.put("depart", dep);    
            
            
            rs=st.executeQuery(sql4);
            List role = new ArrayList();
            i=1;
            while(rs.next())
            {
            	org.jeecgframework.core.util.LogUtil.info(rs.getString("rolename"));
                 TSRole tsRole = new TSRole();
                 tsRole.setId(i+"");
                 tsRole.setRoleName(rs.getString("rolename"));
                 tsRole.setRoleCode(rs.getString("rolecode"));
                 role.add(tsRole);
                 i++;
            }
            root.put("role", role);
            
            rs=st.executeQuery(sql5);
            List susers = new ArrayList();
            i=1;
            while(rs.next())
            {
                 TSUser suer = new TSUser();
                 suer.setId(i+"");
//                 org.jeecgframework.core.util.LogUtil.info(rs.getString("signaturefile"));
//                 suer.setSignatureFile(rs.getString("signaturefile"));
                 suer.setMobilePhone(rs.getString("mobilephone"));
                 suer.setOfficePhone(rs.getString("officephone"));
                 suer.setEmail(rs.getString("email"));
                 susers.add(suer);
                 i++;
            }
            root.put("suser", susers);
            
            rs=st.executeQuery(sql6);
            List typegroup = new ArrayList();
            i=1;
            while(rs.next())
            {
                 TSTypegroup tsTypegroup = new TSTypegroup();
                 tsTypegroup.setId(i+"");
//                 org.jeecgframework.core.util.LogUtil.info(rs.getString("signaturefile"));
//                 suer.setSignatureFile(rs.getString("signaturefile"));
                 tsTypegroup.setTypegroupname(rs.getString("typegroupname"));
                 tsTypegroup.setTypegroupcode(rs.getString("typegroupcode"));
                 typegroup.add(tsTypegroup);
                 i++;
            }
            root.put("typegroup", typegroup);
            
            rs=st.executeQuery(sql7);
            List function = new ArrayList();
            i=1;
            while(rs.next())
            {
                 TSFunction tsFunction = new TSFunction();
                 tsFunction.setId(i+"");
                 tsFunction.setFunctionName(rs.getString("functionName"));
                 tsFunction.setFunctionUrl(rs.getString("functionUrl"));
                 tsFunction.setFunctionLevel(rs.getShort("functionLevel"));
                 tsFunction.setFunctionOrder(rs.getString("functionOrder"));
                 function.add(tsFunction);
                 i++;
            }
            root.put("menu", function);
            
            rs=st.executeQuery(sql8);
            List type = new ArrayList();
            i=1;
            while(rs.next())
            {
                 TSType tsType = new TSType();
                 tsType.setId(i+"");
//                 org.jeecgframework.core.util.LogUtil.info(rs.getString("signaturefile"));
//                 suer.setSignatureFile(rs.getString("signaturefile"));
                 tsType.setTypename(rs.getString("typename"));
                 tsType.setTypecode(rs.getString("typecode"));
                 type.add(tsType);
                 i++;
            }
            root.put("type", type);
            
            rs=st.executeQuery(sql9);
            List log = new ArrayList();
            i=1;
            while(rs.next())
            {
                TSLog slog = new TSLog();
                slog.setId(i+"");
//                 org.jeecgframework.core.util.LogUtil.info(rs.getString("signaturefile"));
//                 suer.setSignatureFile(rs.getString("signaturefile"));
//                 tsType.setTypename(rs.getString("typename"));
//                 tsType.setTypecode(rs.getString("typecode"));
//                 type.add(tsType);
                   slog.setId(i+"");
//                   org.jeecgframework.core.util.LogUtil.info(rs.getString("logcontent"));
                   slog.setLogcontent(rs.getString("logcontent"));
                   slog.setLoglevel(rs.getShort("loglevel"));
                   slog.setBroswer(rs.getString("broswer"));
                   slog.setNote(rs.getString("note"));
                   slog.setOperatetime(rs.getTimestamp("operatetime"));
                   slog.setOperatetype(rs.getShort("operatetype"));
                   log.add(slog);
                 i++;
            }
            root.put("log", log);
            
            rs=st.executeQuery(sql11);
            List cghead = new ArrayList();
            i=1;
            while(rs.next())
            {
                CgFormHeadEntity head = new CgFormHeadEntity();
                head.setId(i+"");
                head.setTableName(rs.getString("table_name"));
                head.setIsTree(rs.getString("is_tree"));
                head.setIsPagination(rs.getString("is_pagination"));
                head.setQuerymode(rs.getString("queryMode"));
                head.setIsCheckbox(rs.getString("is_checkbox"));
                head.setIsDbSynch(rs.getString("is_dbsynch"));
                head.setContent(rs.getString("content"));
                head.setJformVersion(rs.getString("JFORM_VERSION"));
                head.setJformType(rs.getInt("jform_type"));
                head.setColumns(getCgFormItem(sql10, rs.getString("id")));
                cghead.add(head);
                i++;
            }
            root.put("cghead", cghead);
//            rs=st.executeQuery(sql10);
//            List cgfield = new ArrayList();
//            i=1;
//            while(rs.next())
//            {
//            	CgFormFieldEntity filed = new CgFormFieldEntity();
//                filed.setFieldName(rs.getString("field_name"));
//                filed.setLength(rs.getInt("length"));
//                filed.setType(rs.getString("type"));
//                filed.setPointLength(rs.getInt("point_length"));
//                filed.setIsNull(rs.getString("is_null"));
//                filed.setIsKey(rs.getString("is_key"));
//                filed.setIsQuery(rs.getString("is_query"));
//                filed.setIsShow(rs.getString("is_show"));
//                filed.setShowType(rs.getString("show_type"));
//                filed.setOrderNum(rs.getInt("order_num"));
//                filed.setFieldHref(rs.getString("field_href"));
//                filed.setFieldLength(rs.getInt("field_length"));
//                filed.setFieldValidType(rs.getString("field_valid_type"));
//                filed.setQueryMode(rs.getString("query_mode"));
//                filed.setContent(rs.getString("content"));
//                filed.setDictTable(rs.getString("dict_table"));
//                filed.setDictField(rs.getString("dict_field"));
//                filed.setMainField(rs.getString("main_field"));
//                filed.setMainTable(rs.getString("main_table"));
//                cgfield.add(filed);
//                i++;
//            }
//            root.put("cgfield", cgfield);
            
            
            Writer out = new OutputStreamWriter(new FileOutputStream("RepairServiceImpl.java"), "UTF-8");
            temp.process(root, out);
            out.flush();
            out.close();
            org.jeecgframework.core.util.LogUtil.info("Successfull................");
		} catch (IOException e) {
			e.printStackTrace();
		}
	}
	
	
	
	/**
	 * 获取表单字段方法
	 * @param sql10
	 * @param cgformhead_id
	 * @return
	 * @throws Exception
	 */
	public static List getCgFormItem(String sql10,String cgformhead_id) throws Exception{
		Statement st =con.createStatement();
		ResultSet rs = st.executeQuery(sql10+"'"+cgformhead_id.trim()+"'");
        List cgfield = new ArrayList();
        int i=1;
        while(rs.next())
        {
        	CgFormFieldEntity filed = new CgFormFieldEntity();
            filed.setFieldName(rs.getString("field_name"));
            filed.setLength(rs.getInt("length"));
            filed.setType(rs.getString("type"));
            filed.setPointLength(rs.getInt("point_length"));
            filed.setIsNull(rs.getString("is_null"));
            filed.setIsKey(rs.getString("is_key"));
            filed.setIsQuery(rs.getString("is_query"));
            filed.setIsShow(rs.getString("is_show"));
            filed.setShowType(rs.getString("show_type"));
            filed.setOrderNum(rs.getInt("order_num"));
            filed.setFieldHref(rs.getString("field_href"));
            filed.setFieldLength(rs.getInt("field_length"));
            filed.setFieldValidType(rs.getString("field_valid_type"));
            filed.setQueryMode(rs.getString("query_mode"));
            filed.setContent(rs.getString("content"));
            filed.setDictTable(rs.getString("dict_table"));
            filed.setDictField(rs.getString("dict_field"));
            filed.setMainField(rs.getString("main_field"));
            filed.setMainTable(rs.getString("main_table"));
            cgfield.add(filed);
            i++;
        }
        return cgfield;
	}
	
}