init: 导入项目
This commit is contained in:
fengjun
273
Webplatform/src/main/java/com/wb/interact/DBE.java
Normal file
273
Webplatform/src/main/java/com/wb/interact/DBE.java
Normal file
@@ -0,0 +1,273 @@
|
||||
package com.wb.interact;
|
||||
|
||||
import java.io.File;
|
||||
import java.sql.Connection;
|
||||
import java.sql.ResultSet;
|
||||
import java.util.ArrayList;
|
||||
import java.util.HashMap;
|
||||
import java.util.HashSet;
|
||||
import java.util.Map.Entry;
|
||||
import java.util.Set;
|
||||
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
|
||||
import org.json.JSONArray;
|
||||
import org.json.JSONObject;
|
||||
|
||||
import com.wb.util.WebUtil;
|
||||
|
||||
public class DBE {
|
||||
/**
|
||||
* 获取数据库表树数据。
|
||||
*/
|
||||
public static void getTree(HttpServletRequest request, HttpServletResponse response) throws Exception {
|
||||
String type = request.getParameter("type");
|
||||
String jndi = request.getParameter("jndi");
|
||||
String schema = request.getParameter("schema");
|
||||
String result;
|
||||
|
||||
if ("db".equals(type)) {
|
||||
result = getSchemaList(jndi, null);
|
||||
//如果没有Schema直接返回表列表
|
||||
if (result == null)
|
||||
result = getTableList(jndi, null, null);
|
||||
} else if ("schema".equals(type))
|
||||
result = getTableList(jndi, schema, null);
|
||||
else
|
||||
result = getDbList();
|
||||
WebUtil.send(response, result);
|
||||
}
|
||||
|
||||
/**
|
||||
* 如果用户包含演示角色且非管理员，SQL语句仅允许执行select * from table，否则抛出异常。
|
||||
*/
|
||||
public static void checkSelectSql(HttpServletRequest request, HttpServletResponse response) throws Exception {
|
||||
String sql = request.getParameter("sql");
|
||||
String roles[] = (String[]) WebUtil.fetchObject(request, "sys.roles");
|
||||
if (com.wb.util.StringUtil.indexOf(roles, "demo") != -1 && com.wb.util.StringUtil.indexOf(roles, "admin") == -1
|
||||
&& (!sql.startsWith("select * from ") || !com.wb.util.StringUtil.checkName(sql.substring(14), true)
|
||||
|| sql.substring(14).equalsIgnoreCase("WB_USER")))
|
||||
com.wb.util.SysUtil.accessDenied();
|
||||
}
|
||||
|
||||
/**
|
||||
* 从变量sys.jndi节点获得所有配置的jndi，并生成树组件脚本。
|
||||
*/
|
||||
public static String getDbList() throws Exception {
|
||||
JSONObject config = com.wb.util.JsonUtil.readObject(new File(com.wb.common.Base.path, "wb/system/var.json"));
|
||||
HashMap<String, Object> map = new HashMap<String, Object>();
|
||||
ArrayList<Entry<String, Object>> sortedItems;
|
||||
config = config.optJSONObject("sys").optJSONObject("jndi");
|
||||
Set<Entry<String, Object>> es = config.entrySet();
|
||||
String key;
|
||||
JSONObject jo;
|
||||
JSONArray ja = new JSONArray();
|
||||
|
||||
// 默认jndi，插入到首行
|
||||
config.remove("default");
|
||||
for (Entry<String, Object> e : es) {
|
||||
key = e.getKey();
|
||||
map.put(key, ((JSONArray) e.getValue()).optString(0));
|
||||
}
|
||||
sortedItems = com.wb.util.SortUtil.sortKey(map);
|
||||
jo = new JSONObject();
|
||||
jo.put("text", "default");
|
||||
jo.put("jndi", "default");
|
||||
jo.put("type", "db");
|
||||
jo.put("iconCls", "db_icon");
|
||||
ja.put(jo);
|
||||
|
||||
for (Entry<String, Object> e : sortedItems) {
|
||||
jo = new JSONObject();
|
||||
jo.put("text", e.getKey());
|
||||
jo.put("jndi", e.getKey());
|
||||
jo.put("type", "db");
|
||||
jo.put("iconCls", "db_icon");
|
||||
ja.put(jo);
|
||||
}
|
||||
return ja.toString();
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取指定jndi所有数据库表JSON脚本。
|
||||
* @param jndi jndi名称。
|
||||
* @param schema 表Schema。
|
||||
* @param tables 已经配置的表定义。
|
||||
* @return 表列表JSON脚本。
|
||||
*/
|
||||
public static String getTableList(String jndi, String schema, HashSet<String> tables) throws Exception {
|
||||
Connection conn = null;
|
||||
ResultSet rs = null;
|
||||
boolean isFirst = true, hasTableDefine = tables != null;
|
||||
String types[] = { "TABLE" }, tableSchema, tableName, tableText, upperTableName,
|
||||
jndiText = com.wb.util.StringUtil.quote(jndi);
|
||||
StringBuilder buf = new StringBuilder();
|
||||
ArrayList<Entry<String, String>> sortedEntries;
|
||||
HashMap<String, String> tableMap = new HashMap<String, String>();
|
||||
|
||||
try {
|
||||
conn = com.wb.util.DbUtil.getConnection(jndi);
|
||||
rs = conn.getMetaData().getTables(null, schema, null, types);
|
||||
while (rs.next()) {
|
||||
tableSchema = com.wb.util.StringUtil.opt(rs.getString(2));
|
||||
tableName = rs.getString(3);
|
||||
tableMap.put(tableName, tableSchema);
|
||||
}
|
||||
sortedEntries = com.wb.util.SortUtil.sortKey(tableMap);
|
||||
buf.append('[');
|
||||
for (Entry<String, String> entry : sortedEntries) {
|
||||
if (isFirst)
|
||||
isFirst = false;
|
||||
else
|
||||
buf.append(',');
|
||||
tableName = entry.getKey();
|
||||
tableText = com.wb.util.StringUtil.quote(tableName);
|
||||
tableSchema = com.wb.util.StringUtil.quote(entry.getValue());
|
||||
buf.append("{\"text\":");
|
||||
buf.append(tableText);
|
||||
buf.append(",\"type\":\"table\",\"table\":");
|
||||
buf.append(tableText);
|
||||
buf.append(",\"schema\":");
|
||||
buf.append(tableSchema);
|
||||
buf.append(",\"jndi\":");
|
||||
buf.append(jndiText);
|
||||
buf.append(",\"leaf\":true,\"iconCls\":\"");
|
||||
upperTableName = tableName.toUpperCase();
|
||||
if (hasTableDefine && tables.contains(upperTableName)) {
|
||||
tables.remove(upperTableName);
|
||||
buf.append("table_add_icon\"}");
|
||||
} else {
|
||||
buf.append("table_icon\"}");
|
||||
}
|
||||
}
|
||||
// 不匹配的表添加到最后
|
||||
if (hasTableDefine && schema == null) {
|
||||
for (String fullName : tables) {
|
||||
fullName = com.wb.util.StringUtil.quote(fullName);
|
||||
buf.append(",{\"text\":");
|
||||
buf.append(fullName);
|
||||
buf.append(",\"type\":\"table\",\"table\":");
|
||||
buf.append(fullName);
|
||||
buf.append(",\"schema\":\"\",\"jndi\":");
|
||||
buf.append(jndiText);
|
||||
buf.append(",\"leaf\":true,\"iconCls\":\"table_delete_icon\"}");
|
||||
}
|
||||
}
|
||||
buf.append(']');
|
||||
return buf.toString();
|
||||
} finally {
|
||||
com.wb.util.DbUtil.close(rs);
|
||||
com.wb.util.DbUtil.close(conn);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取指定jndi所有Schema列表。
|
||||
* @param jndi jndi名称。
|
||||
* @param tables 已经配置的表定义。
|
||||
* @return Schema列表JSON脚本。 如果没有Schema返回null。
|
||||
*/
|
||||
public static String getSchemaList(String jndi, HashSet<String> tables) throws Exception {
|
||||
Connection conn = null;
|
||||
ResultSet rs = null;
|
||||
String types[] = { "TABLE" }, schema, tableName, upperTableName, jndiText = com.wb.util.StringUtil.quote(jndi);
|
||||
StringBuilder buf = new StringBuilder();
|
||||
HashMap<String, Boolean> schemaMap = new HashMap<String, Boolean>();
|
||||
ArrayList<Entry<String, Boolean>> entryList;
|
||||
boolean isFirst = true, hasTableDefine = tables != null;
|
||||
|
||||
try {
|
||||
conn = com.wb.util.DbUtil.getConnection(jndi);
|
||||
rs = conn.getMetaData().getTables(null, null, null, types);
|
||||
while (rs.next()) {
|
||||
schema = com.wb.util.StringUtil.opt(rs.getString(2));
|
||||
tableName = rs.getString(3);
|
||||
upperTableName = tableName.toUpperCase();
|
||||
schemaMap.put(schema, true);
|
||||
if (hasTableDefine && tables.contains(upperTableName))
|
||||
tables.remove(upperTableName);
|
||||
}
|
||||
//如果不包含Schema返回null
|
||||
if (schemaMap.isEmpty() || schemaMap.size() == 1 && schemaMap.containsKey(""))
|
||||
return null;
|
||||
entryList = com.wb.util.SortUtil.sortKey(schemaMap);
|
||||
buf.append('[');
|
||||
for (Entry<String, Boolean> entry : entryList) {
|
||||
schema = com.wb.util.StringUtil.quote(entry.getKey());
|
||||
if (isFirst)
|
||||
isFirst = false;
|
||||
else
|
||||
buf.append(',');
|
||||
buf.append("{\"text\":");
|
||||
buf.append(schema);
|
||||
buf.append(",\"jndi\":");
|
||||
buf.append(jndiText);
|
||||
buf.append(",\"schema\":");
|
||||
buf.append(schema);
|
||||
buf.append(",\"type\":\"schema\",\"iconCls\":\"db_form_icon\"}");
|
||||
}
|
||||
// 不匹配的表添加到最后
|
||||
if (hasTableDefine) {
|
||||
for (String fullName : tables) {
|
||||
fullName = com.wb.util.StringUtil.quote(fullName);
|
||||
buf.append(",{\"text\":");
|
||||
buf.append(fullName);
|
||||
buf.append(",\"type\":\"table\",\"table\":");
|
||||
buf.append(fullName);
|
||||
buf.append(",\"schema\":\"\",\"jndi\":");
|
||||
buf.append(jndiText);
|
||||
buf.append(",\"leaf\":true,\"iconCls\":\"table_delete_icon\"}");
|
||||
}
|
||||
}
|
||||
buf.append(']');
|
||||
return buf.toString();
|
||||
} finally {
|
||||
com.wb.util.DbUtil.close(rs);
|
||||
com.wb.util.DbUtil.close(conn);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 从指定表下载二进制字段内容。
|
||||
*/
|
||||
public static void downloadBlob(HttpServletRequest request, HttpServletResponse response) throws Exception {
|
||||
String jndi = request.getParameter("__jndi"), tableName = request.getParameter("__tableName"),
|
||||
fieldName = request.getParameter("__fieldName");
|
||||
String selectSql = com.wb.util.DbUtil.buildSQLs(jndi, tableName, false, 1, null, new JSONObject().put(fieldName, 1), null,
|
||||
null)[3];
|
||||
ResultSet rs = (ResultSet) com.wb.util.DbUtil.run(request, selectSql, jndi);
|
||||
com.wb.util.DbUtil.outputBlob(rs, request, response, "download");
|
||||
}
|
||||
|
||||
/**
|
||||
* 上传文件数据至指定数据库表二进制字段。
|
||||
*/
|
||||
public static void uploadBlob(HttpServletRequest request, HttpServletResponse response) throws Exception {
|
||||
setBlob(request, false);
|
||||
}
|
||||
|
||||
/**
|
||||
* 上传文件数据至指定数据库表二进制字段。
|
||||
*/
|
||||
public static void clearBlob(HttpServletRequest request, HttpServletResponse response) throws Exception {
|
||||
setBlob(request, true);
|
||||
}
|
||||
|
||||
/**
|
||||
* 上传文件数据至指定数据库表二进制字段或清除该字段。
|
||||
* @param isClear 是否清除二进制字段，true清除，false更新。
|
||||
*/
|
||||
private static void setBlob(HttpServletRequest request, boolean isClear) throws Exception {
|
||||
String jndi = WebUtil.fetch(request, "__jndi"), tableName = WebUtil.fetch(request, "__tableName"),
|
||||
fieldName = WebUtil.fetch(request, "__fieldName");
|
||||
|
||||
if (isClear)
|
||||
request.setAttribute(fieldName, "");
|
||||
else
|
||||
request.setAttribute(fieldName, request.getAttribute("file"));
|
||||
String updateSql = com.wb.util.DbUtil.buildSQLs(jndi, tableName, false, 1, null, new JSONObject().put(fieldName, 1), null,
|
||||
null)[1];
|
||||
com.wb.util.DbUtil.run(request, updateSql, jndi);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user