Zhenfei.Li
5 years ago
11 changed files with 348 additions and 24 deletions
@ -0,0 +1,205 @@
|
||||
package com.fr.plugin.decision.request.handler; |
||||
|
||||
import com.finebi.activate.foundation.StableManager; |
||||
import com.finebi.burger.api.bean.configuration.table.attachment.FineAttachment; |
||||
import com.finebi.burger.api.bean.configuration.table.field.FineBusinessField; |
||||
import com.finebi.burger.api.request.configuration.table.DataBaseAddTableRequestBean; |
||||
import com.finebi.burger.api.response.configuration.table.TableAddResponseBean; |
||||
import com.finebi.direct.burger.api.service.FineTableService; |
||||
import com.finebi.direct.burger.impl.service.helper.table.FineTableUtils; |
||||
import com.finebi.direct.common.authority.api.FineAuthorityUtils; |
||||
import com.finebi.foundation.api.reponse.FineRespond; |
||||
import com.finebi.foundation.api.service.FineService; |
||||
import com.finebi.foundation.api.service.context.BaseContext; |
||||
import com.finebi.utils.CompareUtils; |
||||
import com.fr.decision.webservice.v10.user.UserService; |
||||
import com.fr.json.JSONArray; |
||||
import com.fr.json.JSONObject; |
||||
import com.fr.log.FineLoggerFactory; |
||||
import com.fr.plugin.decision.HWUtils; |
||||
import com.fr.plugin.decision.core.ActionFactory; |
||||
import com.fr.plugin.decision.core.ExcelDmlBean; |
||||
import com.fr.plugin.decision.core.action.ExcelDmlAction; |
||||
import com.fr.plugin.decision.dao.ExcelLinkService; |
||||
import com.fr.plugin.decision.utils.JdbcUtils; |
||||
import com.fr.stable.StringUtils; |
||||
import com.fr.third.fasterxml.jackson.core.JsonGenerationException; |
||||
import com.fr.third.fasterxml.jackson.databind.JsonMappingException; |
||||
import com.fr.third.fasterxml.jackson.databind.ObjectMapper; |
||||
import com.fr.third.org.apache.poi.hssf.usermodel.HSSFSheet; |
||||
import com.fr.third.org.apache.poi.hssf.usermodel.HSSFWorkbook; |
||||
import com.fr.third.v2.org.apache.poi.xssf.usermodel.XSSFSheet; |
||||
import com.fr.third.v2.org.apache.poi.xssf.usermodel.XSSFWorkbook; |
||||
import com.finebi.direct.common.api.cell.context.CellCreator; |
||||
import com.finebi.common.impl.vo.table.FineBusinessTable; |
||||
import com.finebi.common.impl.vo.table.FineExcelBusinessTable; |
||||
import com.finebi.direct.common.authority.decision.role.FineAuthorityUser; |
||||
import com.fr.web.utils.WebUtils; |
||||
|
||||
import javax.servlet.http.HttpServletRequest; |
||||
|
||||
import javax.servlet.http.HttpServletResponse; |
||||
import java.io.IOException; |
||||
import java.sql.Connection; |
||||
import java.util.HashMap; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
|
||||
public class UpgradeHandler extends HWAbstractHandler { |
||||
public static final String PATH="/excelmng/upgrade"; |
||||
@Override |
||||
protected void deal(HttpServletRequest req, HttpServletResponse res, JSONObject para, JSONObject result) throws Exception { |
||||
|
||||
String connection = para.getString("connection"); |
||||
|
||||
Connection conn = JdbcUtils.getConnection(connection); |
||||
|
||||
if (conn == null){ |
||||
FineLoggerFactory.getLogger().error("HW_excel导入：数据链接 " + connection + " 获取异常" ); |
||||
WebUtils.printAsJSON(res, new JSONObject().put("status","fail").put("errorText","获取数据连接异常")); |
||||
return; |
||||
} |
||||
|
||||
final String createUser = UserService.getInstance().getCurrentUserIdFromCookie(req); |
||||
|
||||
JSONArray items = JSONArray.create(); |
||||
|
||||
JSONArray importDbResult = JSONArray.create(); |
||||
JSONArray addDbResult = JSONArray.create(); |
||||
JSONObject finalResult = JSONObject.create(); |
||||
|
||||
List<FineBusinessTable> allTables = CellCreator.getCellProvider().tableCell().getAllTableWithoutAuth(); |
||||
allTables.forEach(fineBusinessTable -> { |
||||
Map<String, List> values = new HashMap<String, List>(); |
||||
if (fineBusinessTable instanceof FineExcelBusinessTable) { |
||||
|
||||
FineAttachment sheetInfo = ((FineExcelBusinessTable) fineBusinessTable).getBaseAttach(); |
||||
String attachId = sheetInfo.getId(); |
||||
String fileName = sheetInfo.getFileName(); |
||||
int sheetNo = 0; |
||||
|
||||
List<FineBusinessField> fields = ((FineExcelBusinessTable) fineBusinessTable).getExcelFields(); |
||||
|
||||
if (fileName.endsWith(".xlsx")) { |
||||
XSSFWorkbook excelWork = HWUtils.getExcelWork(attachId); |
||||
if(excelWork != null){ |
||||
XSSFSheet sheet = excelWork.getSheetAt(sheetNo); |
||||
JSONObject tableCol = JSONObject.create(); |
||||
for(int i = 0; i < fields.size(); i++){ |
||||
FineBusinessField field = fields.get(i); |
||||
tableCol.put(field.getName(), i); |
||||
|
||||
JSONObject item = new JSONObject(); |
||||
item.put("colLength", "255"); |
||||
item.put("colName", field.getName()); |
||||
item.put("colType", this.getFieldSqlType(field.getType())); |
||||
item.put("excelCol", i); |
||||
item.put("id", field.getId()); |
||||
|
||||
items.put(item); |
||||
} |
||||
values = HWUtils.getSheetContent(sheet, 0, tableCol); |
||||
} |
||||
} else if (fileName.endsWith(".xls")) { |
||||
HSSFWorkbook excel2003Work = HWUtils.getExcel2003Work(attachId); |
||||
if(excel2003Work != null){ |
||||
HSSFSheet sheetAt = excel2003Work.getSheetAt(sheetNo); |
||||
JSONObject tableCol = JSONObject.create(); |
||||
for(int i = 0; i < fields.size(); i++){ |
||||
FineBusinessField field = fields.get(i); |
||||
tableCol.put(field.getName(), i); |
||||
|
||||
JSONObject item = new JSONObject(); |
||||
item.put("colLength", "255"); |
||||
item.put("colName", field.getName()); |
||||
item.put("colType", this.getFieldSqlType(field.getType())); |
||||
item.put("excelCol", i); |
||||
item.put("id", field.getId()); |
||||
|
||||
items.put(item); |
||||
} |
||||
values = HWUtils.get2003SheetContent(sheetAt,0, tableCol); |
||||
} |
||||
} |
||||
|
||||
|
||||
String name = fineBusinessTable.getName(); |
||||
String schema = ""; |
||||
String tableName = fineBusinessTable.getName(); |
||||
|
||||
ExcelDmlAction excelDmlAction = ActionFactory.getInstance().create("add"); |
||||
|
||||
excelDmlAction.setExcelDmlBean( new ExcelDmlBean(connection, schema, tableName, items, values) ); |
||||
|
||||
try { |
||||
JSONObject dmlResult = excelDmlAction.execute(); |
||||
importDbResult.put(dmlResult); |
||||
|
||||
if (StringUtils.equals("success", dmlResult.getString("status")) && dmlResult.getBoolean("addLink")) { |
||||
ExcelLinkService.getInstance().addLink(name, connection, schema, tableName, createUser); |
||||
FineLoggerFactory.getLogger().info("HW_excel导入：FineDB 新增数据成功" + name); |
||||
} |
||||
} catch(Exception e){ |
||||
e.printStackTrace(); |
||||
} |
||||
|
||||
ObjectMapper mapper = new ObjectMapper(); |
||||
JSONObject table = JSONObject.create().put("tableName", fineBusinessTable.getName()).put("connectionName", connection); |
||||
JSONArray tables = JSONArray.create().put(table); |
||||
JSONObject tablesBean = JSONObject.create().put("tables", tables); |
||||
try{ |
||||
FineAuthorityUser authorityUser = FineAuthorityUtils.getAuth(req); |
||||
String packid = "__my_analysis__" + authorityUser.getUserId() + "1"; |
||||
try { |
||||
String tableString = tablesBean.toString(); |
||||
FineBusinessTable[] businessTables = FineTableUtils.createBusinessTablesFromDBNewAddTableInfo(mapper.readValue(tableString, DataBaseAddTableRequestBean.class), authorityUser.getUserId()); |
||||
TableAddResponseBean responed = getTableService().addNewTable(packid, authorityUser, businessTables); |
||||
addDbResult.put(FineRespond.success(responed)); |
||||
} catch (JsonMappingException e) { |
||||
e.printStackTrace(); |
||||
} catch (JsonGenerationException e) { |
||||
e.printStackTrace(); |
||||
} catch (IOException e) { |
||||
e.printStackTrace(); |
||||
} |
||||
} |
||||
catch(Exception e){ |
||||
e.printStackTrace(); |
||||
} |
||||
|
||||
} |
||||
}); |
||||
|
||||
WebUtils.printAsJSON(res, finalResult.put("importDbResult", importDbResult).put("addDbResult", addDbResult)); |
||||
} |
||||
|
||||
@Override |
||||
public String getPath() { |
||||
return PATH; |
||||
} |
||||
|
||||
protected static FineTableService getTableService() { |
||||
return getBean(FineTableService.class); |
||||
} |
||||
|
||||
private static <T extends FineService> T getBean(Class<T> annotatedClass) { |
||||
return getContext().getServiceBean(annotatedClass); |
||||
} |
||||
/** |
||||
* service获取集合 |
||||
*/ |
||||
private static BaseContext getContext() { |
||||
return StableManager.getContext(); |
||||
} |
||||
|
||||
private int getFieldSqlType(int fieldType) { |
||||
if (CompareUtils.isEqual(fieldType, 16)) { |
||||
return 12; |
||||
} else if (CompareUtils.isEqual(fieldType, 32)) { |
||||
return 8; |
||||
} else if (CompareUtils.isEqual(fieldType, 48)) { |
||||
return 91; |
||||
} |
||||
return 12; |
||||
} |
||||
} |
||||
@ -0,0 +1,15 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?> |
||||
<module type="JAVA_MODULE" version="4"> |
||||
<component name="NewModuleRootManager"> |
||||
<output url="file://$USER_HOME$/env/direct-bi/WEB-INF/plugins/plugin-com.fr.plugin.hw.import.excel-2.0.0/classes" /> |
||||
<output-test url="file://$MODULE_DIR$/../../../../direct-bi/nuclear-maven/classes/test/main" /> |
||||
<exclude-output /> |
||||
<content url="file://$MODULE_DIR$"> |
||||
<sourceFolder url="file://$MODULE_DIR$/java" isTestSource="false" /> |
||||
</content> |
||||
<orderEntry type="inheritedJdk" /> |
||||
<orderEntry type="sourceFolder" forTests="false" /> |
||||
<orderEntry type="library" name="lib" level="project" /> |
||||
<orderEntry type="module" module-name="direct-common-adapter" /> |
||||
</component> |
||||
</module> |
||||
@ -0,0 +1,59 @@
|
||||
!(function () { |
||||
var UpgradePane = BI.inherit(BI.Widget, { |
||||
props: { |
||||
baseCls: "dec-mng-upgrade" |
||||
}, |
||||
|
||||
_store: function () { |
||||
return BI.Models.getModel("dec.model.mng.upgrade"); |
||||
}, |
||||
|
||||
beforeInit: function (callback) { |
||||
this.store.initData(callback); |
||||
}, |
||||
|
||||
render: function () { |
||||
var self = this; |
||||
return { |
||||
type: "bi.vertical", |
||||
width: 640, |
||||
vgap: 15, |
||||
items: [{ |
||||
type: "bi.vertical_adapt", |
||||
items: [{ |
||||
type: "bi.label", |
||||
textAlign: "right", |
||||
cls: "dec-font-weight-bold", |
||||
text: "数据源:", |
||||
title: "数据源", |
||||
width: 100 |
||||
}, { |
||||
el: { |
||||
type: "bi.text_value_combo", |
||||
lgap: 4, |
||||
cls: "bi-border", |
||||
textAlign: "left", |
||||
items: this.model.connectionItems, |
||||
width: 480, |
||||
height: 22, |
||||
ref: function (_ref) { |
||||
self.connection = _ref; |
||||
} |
||||
}, |
||||
lgap: 10 |
||||
}] |
||||
}, { |
||||
type: "bi.button", |
||||
width: 200, |
||||
text: "升级旧Excel", |
||||
handler: function () { |
||||
Dec.HW.upgrade({ |
||||
connection: self.connection.getValue()[0] |
||||
}); |
||||
} |
||||
}] |
||||
}; |
||||
} |
||||
}); |
||||
BI.shortcut("dec.mng.upgrade", UpgradePane); |
||||
})(); |
||||
@ -0,0 +1,25 @@
|
||||
!(function () { |
||||
var Model = BI.inherit(Fix.Model, { |
||||
state: function () { |
||||
return { |
||||
connectionItems: [], |
||||
}; |
||||
}, |
||||
|
||||
actions: { |
||||
initData: function (callback) { |
||||
var self = this; |
||||
Dec.Utils.getConnections(function (res) { |
||||
self.model.connectionItems = BI.map(res, function (i, val) { |
||||
return { |
||||
text: val.connectionName || "", |
||||
value: val.connectionName || "" |
||||
}; |
||||
}); |
||||
callback(); |
||||
}); |
||||
} |
||||
} |
||||
}); |
||||
BI.model("dec.model.mng.upgrade", Model); |
||||
})(); |
||||
Loading…
Reference in new issue