• Home
• org.docshare
• yangmvc
• 2022.12.15
• source code
• DBHelper.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

org.docshare.orm.DBHelper Maven / Gradle / Ivy

package org.docshare.orm;

import java.io.InputStream;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
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.docshare.log.Log;
import org.docshare.mvc.Config;
import org.docshare.mvc.except.MVCException;
import org.docshare.orm.ColumnDesc.ExportTo;
import org.docshare.orm.access.AccessDBHelper;
import org.docshare.orm.mysql.MySQLDBHelper;
import org.docshare.orm.postgres.PostgresDBHelper;
import org.docshare.util.FileTool;
import org.docshare.util.TextTool;

import com.alibaba.fastjson.JSON;

public abstract class DBHelper {

	public static final String DB_MYSQL = "mysql";
	public static final String DB_POSTGRES = "postgres";
	
	
	static ThreadLocal locals = new ThreadLocal();

	/**
	 * 使用该函数获取DBHelper对象。 本类的设计为同一个线程公用一个DBHelper
	 * @param type 数据库类型
	 * @return DBHelper对象
	 */
	public static DBHelper getIns(String type) {
		DBHelper ins = locals.get();
		if(ins == null || ! useCache){
			if(type == null  || type.equals("mysql")){
				ins = new MySQLDBHelper();	
			}else if(type.equals("postgres")){
				ins = new PostgresDBHelper();
			}else if(type.equals("access")){
				ins = new AccessDBHelper();
			}
			locals.set(ins);
		}
		return ins;
	}
	public static DBHelper getIns(){
		return getIns(Config.dbtype);
	}
	public Connection getConnection(){
		return con;
	}

	public static boolean useCache = true;

	public static void disableCache() { //用于Bae这种变态的环境。
		useCache = false;
	}

	public static void removeThreadLocal() {
		try{
			locals.remove();
		}catch (Exception e) {
			//Log.e(e);
		}
	}

	protected Connection con = null;
	protected Object last_id;

	/**
	 * 连接数据库
	 */
	public abstract void conn();

	/**
	 * 获取该表的外键信息
	 * @param table 表名
	 * @param map 字段描述
	 * @throws SQLException 数据库异常
	 */
	public void getFKey(String table, HashMap map) throws SQLException {
			conn();
			
			
			DatabaseMetaData dbmd = con.getMetaData();
			
			//导入键   --- 我的键是别人的主键
			ResultSet rs = dbmd.getImportedKeys(Config.dbname, null, table);
			
	//		ResultSetMetaData meta = rs.getMetaData();
	//		Log.d("column count "+meta.getColumnCount());
	//		for(int i=1;i<=meta.getColumnCount();i++){
	//			Log.d(meta.getColumnName(i));
	//		}
			
			while(rs.next()){
				String pk_table = rs.getString("PKTABLE_NAME");
				String pk_column = rs.getString("PKCOLUMN_NAME");
				String my_column = rs.getString("FKCOLUMN_NAME");
				ColumnDesc desc = map.get(my_column);
				if(desc == null) continue;
				
				desc.pk_column = pk_column;
				desc.pk_table = pk_table;
			}
			rs.close();
			//导出键值--- 我的主键是别人的外键。
			rs = dbmd.getExportedKeys(Config.dbname, null, table);
			while(rs.next()){
				ExportTo eTo = new ExportTo();
                //String pkTableName = rs.getString("PKTABLE_NAME");//主键表名   
                String pkColumnName = rs.getString("PKCOLUMN_NAME");//主键列名    
                String fkTableName = rs.getString("FKTABLE_NAME");//外键表名  
                String fkColumnName = rs.getString("FKCOLUMN_NAME"); //外键列名    
                eTo.table = fkTableName;
                eTo.column = fkColumnName;
                ColumnDesc desc = map.get(pkColumnName);
				if(desc == null) continue;
				if(desc.exportKey==null){
					desc.exportKey = new ArrayList();
				}
				desc.exportKey.add(eTo);
			}
			rs.close();
		}

	ArrayList