ars.database.service.StandardGeneralService Maven / Gradle / Ivy
package ars.database.service;
import java.io.File;
import java.io.Serializable;
import java.util.Map;
import java.util.List;
import java.util.UUID;
import java.util.HashMap;
import java.util.Map.Entry;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.streaming.SXSSFWorkbook;
import ars.util.Beans;
import ars.util.Dates;
import ars.util.Files;
import ars.util.Nfile;
import ars.util.Strings;
import ars.file.office.Excels;
import ars.invoke.request.Requester;
import ars.database.model.TreeModel;
import ars.database.repository.Query;
import ars.database.repository.Repositories;
import ars.database.service.ExcelAdapter;
import ars.database.service.ImportService;
import ars.database.service.AbstractService;
import ars.database.service.SimpleExcelAdapter;
/**
* 通用业务操作接口抽象实现
*
* @author yongqiangwu
*
* @param
* 数据模型
*/
public abstract class StandardGeneralService extends AbstractService {
private String directory = Strings.TEMP_PATH;
private ExcelAdapter excelAdapter = new SimpleExcelAdapter();
public String getDirectory() {
return directory;
}
public void setDirectory(String directory) {
this.directory = directory;
}
public ExcelAdapter getExcelAdapter() {
return excelAdapter;
}
public void setExcelAdapter(ExcelAdapter excelAdapter) {
this.excelAdapter = excelAdapter;
}
/**
* 获取查询有效参数（排除参数值为空的参数）
*
* @param parameters
* 参数键/值对
* @return 有效参数键/值对
*/
protected Map getEffectiveParameters(Map parameters) {
if (parameters == null || parameters.isEmpty()) {
return new HashMap(0);
}
String emptySuffix = new StringBuilder(Query.DELIMITER).append(Query.EMPTY).toString();
String nonemptySuffix = new StringBuilder(Query.DELIMITER).append(Query.NOT_EMPTY).toString();
Map effectives = new HashMap(parameters.size());
for (Entry entry : parameters.entrySet()) {
String key = entry.getKey();
Object value = entry.getValue();
if (Beans.isEmpty(value) && !key.endsWith(emptySuffix) && !key.endsWith(nonemptySuffix)) {
continue;
}
effectives.put(key, value);
}
return effectives;
}
/**
* 获取数据查询对象
*
* @param requester
* 请求对象
* @param parameters
* 数据过滤参数
* @return 数据查询对象
*/
protected Query getQuery(Requester requester, Map parameters) {
return this.getQuery(requester).custom(parameters);
}
/**
* 数据批量导入
*
* @param requester
* 请求对象
* @param file
* 文件对象
* @param start
* 开始数据行（从0开始）
* @param adapter
* Excel文件适配对象
* @param parameters
* 请求参数
* @return 数据导入结果对象
* @throws Exception 操作异常
*/
protected ImportService.Result import_(final Requester requester, final Nfile file, final Integer start,
final ExcelAdapter adapter, Map parameters) throws Exception {
long timestamp = System.currentTimeMillis();
final Workbook failed = new SXSSFWorkbook(100);
final ImportService.Result result = new ImportService.Result();
final String[] titles = start > 0 ? Excels.getTitles(file, start - 1) : Strings.EMPTY_ARRAY;
try {
adapter.begin(requester, this);
Excels.iteration(file, start, new Excels.Reader() {
@Override
public T read(Row row) {
if (row == null || Excels.isEmpty(row)) {
return null;
}
try {
T entity = adapter.read(requester, StandardGeneralService.this, row);
if (entity != null) {
saveObject(requester, entity);
}
} catch (Exception e) {
result.setFailed(result.getFailed() + 1);
Sheet sheet = failed.getSheetAt(failed.getNumberOfSheets() - 1);
int index = sheet.getLastRowNum() + 1;
if (result.getFailed() == 1 || result.getFailed() % 60000 == 1) {
sheet = failed.createSheet();
Excels.setTitles(sheet.createRow(0), titles);
index = 1;
}
Row target = sheet.createRow(index);
Excels.copy(row, target);
int columns = titles.length > 0 ? titles.length : row.getLastCellNum();
Excels.setValue(target.createCell(columns), e.getMessage());
} finally {
result.setTotal(result.getTotal() + 1);
}
return null;
}
});
if (result.getFailed() > 0) {
String name = new StringBuilder(UUID.randomUUID().toString()).append(".xlsx").toString();
File attachment = new File(this.directory, name);
Excels.write(failed, attachment);
result.setFile(name);
result.setSize(Files.getUnitSize(attachment.length()));
}
} finally {
failed.close();
adapter.complete(requester, this);
}
result.setSpend(Dates.getUnitTime(System.currentTimeMillis() - timestamp));
return result;
}
/**
* 数据批量导出
*
* @param requester
* 请求对象
* @param adapter
* Excel文件适配对象
* @param parameters
* 请求参数
* @return 导出结果
* @throws Exception 操作异常
*/
public ExportService.Result export(Requester requester, ExcelAdapter adapter, Map parameters)
throws Exception {
parameters = requester.getParameters();
Integer page = Beans.toInteger(Integer.class, parameters.remove(Query.PAGE));
Integer size = Beans.toInteger(Integer.class, parameters.remove(Query.SIZE));
long timestamp = System.currentTimeMillis();
final ExportService.Result result = new ExportService.Result();
Workbook workbook = new SXSSFWorkbook(100);
try {
adapter.begin(requester, this);
Sheet sheet = workbook.createSheet();
String[] titles = adapter.getTitles(requester, this);
Excels.setTitles(sheet.createRow(0), titles);
if (page == null || size == null) {
int count = this.getQuery(requester, parameters).count();
int length = 1000;
int pages = count / length;
int remain = count % length;
if (remain > 0) {
pages++;
}
int total = 1;
for (int i = 1; i < pages + 1; i++) {
if (total > 1 && total % 60000 == 1) {
sheet = workbook.createSheet();
Excels.setTitles(sheet.createRow(0), titles);
}
int end = i * length;
List objects = this.getQuery(requester, parameters).paging(i, end > count ? remain : length)
.list();
for (int n = 0; n < objects.size(); n++) {
adapter.write(requester, this, objects.get(n), sheet.createRow(n + 1));
result.setTotal(result.getTotal() + 1);
}
}
} else {
List objects = this.getQuery(requester, parameters).paging(page, size).list();
for (int i = 0; i < objects.size(); i++) {
adapter.write(requester, this, objects.get(i), sheet.createRow(i + 1));
}
}
String name = new StringBuilder(UUID.randomUUID().toString()).append(".xlsx").toString();
File attachment = new File(this.directory, name);
Excels.write(workbook, attachment);
result.setFile(name);
result.setSize(Files.getUnitSize(attachment.length()));
result.setSpend(Dates.getUnitTime(System.currentTimeMillis() - timestamp));
} finally {
workbook.close();
adapter.complete(requester, this);
}
return result;
}
/**
* 新增对象实体
*
* @param requester
* 请求对象
* @param parameters
* 对象实体参数
* @return 新增对象实体主键
*/
public Serializable add(Requester requester, Map parameters) {
T entity = Beans.getInstance(this.getModel());
this.initObject(requester, entity, parameters);
return this.saveObject(requester, entity);
}
/**
* 删除对象
*
* @param requester
* 请求对象
* @param parameters
* 数据过滤参数
*/
public void delete(Requester requester, Map parameters) {
Map effectives = this.getEffectiveParameters(parameters);
if (!effectives.isEmpty()) {
List entities = this.getQuery(requester, effectives).list();
for (int i = 0; i < entities.size(); i++) {
this.deleteObject(requester, entities.get(i));
}
}
}
/**
* 修改对象实体
*
* @param requester
* 请求对象
* @param parameters
* 对象实体参数
*/
public void update(Requester requester, Map parameters) {
String primary = this.getRepository().getPrimary();
Object[] identifiers = Beans.toArray(Object.class, parameters.get(primary));
if (identifiers.length > 0) {
List entities = this.getQuery(requester).or(primary, identifiers).list();
for (int i = 0; i < entities.size(); i++) {
T entity = entities.get(i);
this.initObject(requester, entity, parameters);
this.updateObject(requester, entity);
}
}
}
/**
* 统计数量
*
* @param requester
* 请求对象
* @param parameters
* 过滤参数
* @return 数量
*/
public int count(Requester requester, Map parameters) {
Map effectives = this.getEffectiveParameters(parameters);
effectives.remove(Query.PAGE);
effectives.remove(Query.SIZE);
return this.getQuery(requester, effectives).count();
}
/**
* 数据统计
*
* @param requester
* 请求对象
* @param parameters
* 过滤参数
* @return 统计数据列表
*/
public List stats(Requester requester, Map parameters) {
return this.getQuery(requester, parameters).stats();
}
/**
* 获取单个对象
*
* @param requester
* 请求对象
* @param parameters
* 过滤参数
* @return 对象实例
*/
public T object(Requester requester, Map parameters) {
Map effectives = this.getEffectiveParameters(parameters);
if (effectives.isEmpty()) {
return null;
}
return this.getQuery(requester, effectives).single();
}
/**
* 获取对象列表
*
* @param requester
* 请求对象
* @param parameters
* 过滤参数
* @return 对象实例列表
*/
public List objects(Requester requester, Map parameters) {
return this.getQuery(requester, parameters).list();
}
/**
* 获取树对象列表
*
* @param requester
* 请求对象
* @param parameters
* 过滤参数
* @return 树对象实例列表
*/
@SuppressWarnings({ "rawtypes", "unchecked" })
public List trees(Requester requester, Map parameters) {
List objects = this.getQuery(requester, parameters).list();
return (List) Repositories.mergeTrees((List) objects);
}
/**
* 数据批量导入
*
* @param requester
* 请求对象
* @param file
* 文件对象
* @param start
* 开始数据行（从0开始）
* @param parameters
* 请求参数
* @return 数据导入结果对象
* @throws Exception 操作异常
*/
public ImportService.Result import_(Requester requester, Nfile file, Integer start, Map parameters)
throws Exception {
return this.import_(requester, file, start, this.excelAdapter, parameters);
}
/**
* 数据批量导出
*
* @param requester
* 请求对象
* @param parameters
* 请求参数
* @return 导出结果
* @throws Exception 操作异常
*/
public ExportService.Result export(Requester requester, Map parameters) throws Exception {
return this.export(requester, this.excelAdapter, parameters);
}
/**
* 下载批量导入（失败）/导出文件
*
* @param requester
* 请求对象
* @param name
* 文件名称
* @param parameters
* 请求参数
* @return 文件对象
*/
public Nfile file(Requester requester, String name, Map parameters) {
return new Nfile(new File(this.directory, name));
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy