【feat】 提交所有改动

This commit is contained in:
Kris 2025-07-24 14:37:33 +08:00
parent 15e755ff4b
commit 7893f8fdb3
12 changed files with 111 additions and 177 deletions

View File

@ -1,11 +1,11 @@
# 程序配置 # 程序配置
# xxl job数据库链接 # xxl job数据库链接
#cook #cook
#dbUrl="jdbc:mysql://127.0.0.1:3306/data-dump-xxl-job?useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&serverTimezone=Asia/Shanghai" #dbUrl="jdbc:mysql://127.0.0.1:3306/data-dump-xxl-job?useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&useSSL=false&serverTimezone=Asia/Shanghai"
#dbUsername="root" #dbUsername="root"
#dbPassword="celnet@2025.bln" #dbPassword="celnet@2025.bln"
#携科 #携科
dbUrl="jdbc:mysql://127.0.0.1:3306/data-dump-xxl-job?useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&useSSL=false&serverTimezone=Asia/Shanghai" dbUrl="jdbc:mysql://127.0.0.1:3306/data-dump-xxl-job?useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&serverTimezone=Asia/Shanghai&useSSL=false"
dbUsername="root" dbUsername="root"
dbPassword="Celnet2025.QY" dbPassword="Celnet2025.QY"
#其它 #其它

View File

@ -110,8 +110,6 @@ public class FileManagerController {
} }
if ("Attachment".equals(api)){ if ("Attachment".equals(api)){
fileService.uploadFileToAttachment(api, blobField, param.getSingleThread()); fileService.uploadFileToAttachment(api, blobField, param.getSingleThread());
}else if("Document".equals(api)){
fileService.uploadFileToDocument(api, blobField, param.getSingleThread());
}else { }else {
fileService.uploadFile(api, blobField, param.getSingleThread()); fileService.uploadFile(api, blobField, param.getSingleThread());
} }

View File

@ -2,12 +2,16 @@ package com.celnet.datadump.job;
import com.alibaba.fastjson.JSON; import com.alibaba.fastjson.JSON;
import com.celnet.datadump.config.SalesforceConnect; import com.celnet.datadump.config.SalesforceConnect;
import com.celnet.datadump.entity.DataObject;
import com.celnet.datadump.param.SalesforceParam; import com.celnet.datadump.param.SalesforceParam;
import com.celnet.datadump.service.CommonService; import com.celnet.datadump.service.CommonService;
import com.celnet.datadump.service.DataImportBatchService; import com.celnet.datadump.service.DataImportBatchService;
import com.celnet.datadump.service.DataImportNewService; import com.celnet.datadump.service.DataImportNewService;
import com.celnet.datadump.service.DataObjectService;
import com.celnet.datadump.util.DataUtil;
import com.xxl.job.core.biz.model.ReturnT; import com.xxl.job.core.biz.model.ReturnT;
import com.xxl.job.core.handler.annotation.XxlJob; import com.xxl.job.core.handler.annotation.XxlJob;
import com.xxl.job.core.log.XxlJobLogger;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
@ -31,6 +35,9 @@ public class DataDumpNewJob {
@Autowired @Autowired
private DataImportBatchService dataImportBatchService; private DataImportBatchService dataImportBatchService;
@Autowired
private DataObjectService dataObjectService;
/** /**
* bulk批量大数据生成newSFID * bulk批量大数据生成newSFID
@ -213,7 +220,7 @@ public class DataDumpNewJob {
/** /**
* 文件下载 * 文件下载
* @param paramStr * @param paramStr
* @author kris * @author kris
* @return * @return
@ -236,6 +243,44 @@ public class DataDumpNewJob {
return dataImportNewService.dumpFileNew(param); return dataImportNewService.dumpFileNew(param);
} }
/**
* 文件上传
* @param paramStr
* @author kris
* @return
* @throws Exception
*/
@XxlJob("uploadFileNewJob")
public ReturnT<String> uploadFileNewJob(String paramStr) throws Exception {
log.info("uploadFileNewJob execute start ..................");
SalesforceParam param = new SalesforceParam();
try {
if (StringUtils.isNotBlank(paramStr)) {
param = JSON.parseObject(paramStr, SalesforceParam.class);
}
} catch (Throwable throwable) {
return new ReturnT<>(500, "参数解析失败!");
}
// 参数转换
param.setBeginCreateDate(param.getBeginDate());
param.setEndCreateDate(param.getEndDate());
for (String api : DataUtil.toIdList(param.getApi())) {
DataObject dataObject = dataObjectService.getById(api);
String blobField = dataObject.getBlobField();
if (StringUtils.isBlank(blobField)) {
log.error("api:{} does not have blob field", api);
XxlJobLogger.log("api:{} does not have blob field", api);
}
if ("Attachment".equals(api)){
dataImportNewService.uploadFileToAttachmentNew(param,dataObject);
}else {
dataImportNewService.uploadFileNew(param,dataObject);
}
}
return ReturnT.SUCCESS;
}
/** /**
* 增量任务 * 增量任务
* *

View File

@ -122,8 +122,6 @@ public class DumpFileJob {
} }
if ("Attachment".equals(api)){ if ("Attachment".equals(api)){
fileService.uploadFileToAttachment(api, blobField, param.getSingleThread()); fileService.uploadFileToAttachment(api, blobField, param.getSingleThread());
}else if("Document".equals(api)){
fileService.uploadFileToDocument(api, blobField, param.getSingleThread());
}else { }else {
fileService.uploadFile(api, blobField, param.getSingleThread()); fileService.uploadFile(api, blobField, param.getSingleThread());
} }

View File

@ -1,5 +1,6 @@
package com.celnet.datadump.service; package com.celnet.datadump.service;
import com.celnet.datadump.entity.DataObject;
import com.celnet.datadump.param.SalesforceParam; import com.celnet.datadump.param.SalesforceParam;
import com.xxl.job.core.biz.model.ReturnT; import com.xxl.job.core.biz.model.ReturnT;
@ -36,4 +37,9 @@ public interface DataImportNewService {
ReturnT<String> dumpFileNew(SalesforceParam param) throws Exception; ReturnT<String> dumpFileNew(SalesforceParam param) throws Exception;
void uploadFileToAttachmentNew(SalesforceParam param, DataObject dataObject) throws Exception;
void uploadFileNew(SalesforceParam param, DataObject dataObject) throws Exception;
} }

View File

@ -36,6 +36,4 @@ public interface FileService {
void uploadFileToAttachment(String api, String field, Boolean singleThread); void uploadFileToAttachment(String api, String field, Boolean singleThread);
void uploadFileToDocument(String api, String field, Boolean singleThread);
} }

View File

@ -370,7 +370,7 @@ public class DataImportBatchServiceImpl implements DataImportBatchService {
if (reference == null){ if (reference == null){
continue; continue;
} }
log.info("----------" + dataField.getField() + "的引用类型 ------------" + reference); // log.info("----------" + dataField.getField() + "的引用类型 ------------" + reference);
List<Map<String, Object>> referenceMap = customMapper.list("new_id", reference, "new_id is not null limit 1"); List<Map<String, Object>> referenceMap = customMapper.list("new_id", reference, "new_id is not null limit 1");
if (referenceMap.isEmpty()){ if (referenceMap.isEmpty()){
QueryWrapper<DataObject> maxIndex = new QueryWrapper<>(); QueryWrapper<DataObject> maxIndex = new QueryWrapper<>();
@ -755,7 +755,7 @@ public class DataImportBatchServiceImpl implements DataImportBatchService {
if (reference_to == null){ if (reference_to == null){
continue; continue;
} }
log.info("----------" + dataField.getField() + "的引用类型 ------------" + reference_to); // log.info("----------" + dataField.getField() + "的引用类型 ------------" + reference_to);
//判断reference_to内是否包含User字符串 //判断reference_to内是否包含User字符串
if (reference_to.contains(",User") || reference_to.contains("User,")) { if (reference_to.contains(",User") || reference_to.contains("User,")) {
reference_to = "User"; reference_to = "User";

View File

@ -784,7 +784,7 @@ public class DataImportNewServiceImpl implements DataImportNewService {
if (reference_to == null){ if (reference_to == null){
continue; continue;
} }
log.info("----------" + dataField.getField() + "的引用类型 ------------" + reference_to); // log.info("----------" + dataField.getField() + "的引用类型 ------------" + reference_to);
//判断reference_to内是否包含User字符串 //判断reference_to内是否包含User字符串
if (reference_to.contains(",User") || reference_to.contains("User,")) { if (reference_to.contains(",User") || reference_to.contains("User,")) {
reference_to = "User"; reference_to = "User";
@ -1116,6 +1116,20 @@ public class DataImportNewServiceImpl implements DataImportNewService {
return ReturnT.SUCCESS; return ReturnT.SUCCESS;
} }
@Override
public void uploadFileToAttachmentNew(SalesforceParam param, DataObject dataObject) throws Exception {
}
@Override
public void uploadFileNew(SalesforceParam param, DataObject dataObject) throws Exception {
}
/** /**
* 下载文件 * 下载文件
*/ */
@ -1133,7 +1147,10 @@ public class DataImportNewServiceImpl implements DataImportNewService {
String token = Connection.getSessionHeader().getSessionId(); String token = Connection.getSessionHeader().getSessionId();
String name = getName(api); String name = "Name";
if (Const.CONTENT_VERSION.equalsIgnoreCase(api)) {
name = "Title,FileExtension";
}
log.info("api:{},field:{},开始时间:{},结束时间:{}", api, field,beginDateStr,endDateStr); log.info("api:{},field:{},开始时间:{},结束时间:{}", api, field,beginDateStr,endDateStr);
Map<String, String> headers = Maps.newHashMap(); Map<String, String> headers = Maps.newHashMap();
@ -1160,8 +1177,13 @@ public class DataImportNewServiceImpl implements DataImportNewService {
// 上传完毕 更新附件信息 // 上传完毕 更新附件信息
List<Map<String, Object>> maps = Lists.newArrayList(); List<Map<String, Object>> maps = Lists.newArrayList();
try { try {
String fileName;
id = (String) map.get("Id"); id = (String) map.get("Id");
String fileName = (String) map.get(name); if (Const.CONTENT_VERSION.equalsIgnoreCase(api)) {
fileName = map.get("Title") + "." + map.get("FileExtension");
}else {
fileName = (String) map.get("Name");
}
log.info("------------文件名:" + id + "_" + fileName); log.info("------------文件名:" + id + "_" + fileName);
// 判断路径是否为空 // 判断路径是否为空
if (StringUtils.isNotBlank(fileName)) { if (StringUtils.isNotBlank(fileName)) {
@ -1200,7 +1222,7 @@ public class DataImportNewServiceImpl implements DataImportNewService {
} }
Map<String, Object> paramMap = Maps.newHashMap(); Map<String, Object> paramMap = Maps.newHashMap();
paramMap.put("key", "is_dump"); paramMap.put("key", "is_dump");
paramMap.put("value", true); paramMap.put("value", 1);
maps.add(paramMap); maps.add(paramMap);
customMapper.updateById(api, maps, id); customMapper.updateById(api, maps, id);
TimeUnit.MILLISECONDS.sleep(1); TimeUnit.MILLISECONDS.sleep(1);

View File

@ -93,7 +93,7 @@ public class FileServiceImpl implements FileService {
extraSqlTmp += "AND Id > '" + maxId + "'"; extraSqlTmp += "AND Id > '" + maxId + "'";
} }
// 获取未存储的附件id // 获取未存储的附件id
List<Map<String, Object>> list = customMapper.list("Id, url", api, extraSqlTmp + " AND is_dump = true order by id asc limit 10"); List<Map<String, Object>> list = customMapper.list("Id, url", api, extraSqlTmp + " AND is_dump = 1 order by id asc limit 10");
if (CollectionUtils.isEmpty(list)) { if (CollectionUtils.isEmpty(list)) {
break; break;
} }
@ -162,19 +162,21 @@ public class FileServiceImpl implements FileService {
} }
} }
try { try {
String name = getName(api); String name = "Name";
if (Const.CONTENT_VERSION.equalsIgnoreCase(api)) {
name = "Title,FileExtension";
}
Map<String, String> headers = Maps.newHashMap(); Map<String, String> headers = Maps.newHashMap();
headers.put("Authorization", "Bearer " + token); headers.put("Authorization", "Bearer " + token);
headers.put("connection", "keep-alive"); headers.put("connection", "keep-alive");
long num = 0; long num = 0;
while (true) { while (true) {
// 获取未存储的附件id // 获取未存储的附件id
List<Map<String, Object>> list = customMapper.list("Id, " + name, api, extraSql + " AND is_dump = false limit 1000"); List<Map<String, Object>> list = customMapper.list("Id, " + name, api, extraSql + " AND is_dump = 0 limit 1000");
if (CollectionUtils.isEmpty(list)) { if (CollectionUtils.isEmpty(list)) {
log.info("无需要下载文件数据!!!"); log.info("无需要下载文件数据!!!");
break; break;
} }
String finalName = name;
for (Map<String, Object> map : list) { for (Map<String, Object> map : list) {
String finalDownloadUrl = downloadUrl; String finalDownloadUrl = downloadUrl;
Future<?> future = salesforceExecutor.execute(() -> { Future<?> future = salesforceExecutor.execute(() -> {
@ -185,8 +187,13 @@ public class FileServiceImpl implements FileService {
int failCount = 0; int failCount = 0;
while (true) { while (true) {
try { try {
String fileName;
id = (String) map.get("Id"); id = (String) map.get("Id");
String fileName = (String) map.get(finalName); if (Const.CONTENT_VERSION.equalsIgnoreCase(api)) {
fileName = map.get("Title") + "." + map.get("FileExtension");
}else {
fileName = (String) map.get("Name");
}
log.info("------------文件名:"+id + "_" + fileName); log.info("------------文件名:"+id + "_" + fileName);
// 判断路径是否为空 // 判断路径是否为空
if (StringUtils.isNotBlank(fileName)) { if (StringUtils.isNotBlank(fileName)) {
@ -430,7 +437,7 @@ public class FileServiceImpl implements FileService {
String name = "Name"; String name = "Name";
// contentVersion使用PathOnClient // contentVersion使用PathOnClient
if (Const.CONTENT_VERSION.equalsIgnoreCase(api)) { if (Const.CONTENT_VERSION.equalsIgnoreCase(api)) {
name = "PathOnClient"; name = "Title,FileExtension";
} }
return name; return name;
} }
@ -776,144 +783,4 @@ public class FileServiceImpl implements FileService {
} }
} }
@Override
public void uploadFileToDocument(String api, String field, Boolean singleThread) {
String uploadUrl = null;
List<Map<String, Object>> poll = customMapper.list("code,value","org_config",null);
for (Map<String, Object> map1 : poll) {
if ("FILE_UPLOAD_URL".equals(map1.get("code"))) {
uploadUrl = (String) map1.get("value");
}
}
if (StringUtils.isBlank(uploadUrl)) {
EmailUtil.send("UploadFile ERROR", "文件上传失败!上传地址未配置");
return;
}
log.info("upload file api:{}, field:{}", api, field);
PartnerConnection connect = salesforceTargetConnect.createConnect();
String token = connect.getSessionHeader().getSessionId();
List<Future<?>> futures = Lists.newArrayList();
String extraSql = "";
if (dataFieldService.hasDeleted(api)) {
extraSql += "AND IsDeleted = false ";
}
if (Const.FILE_TYPE == FileType.SERVER) {
// 检测路径是否存在
File excel = new File(Const.SERVER_FILE_PATH + "/" + api);
if (!excel.exists()) {
throw new RuntimeException("找不到文件路径");
}
}
try {
// 获取未存储的附件id
List<Map<String, Object>> list = customMapper.list("Id, Name, localUrl, Description", api, "is_upload = 0");
for (Map<String, Object> map : list) {
String finalUploadUrl = uploadUrl;
String id = null;
// 上传完毕 更新附件信息
List<Map<String, Object>> maps = Lists.newArrayList();
boolean isUpload = true;
int failCount = 0;
CloseableHttpResponse response = null;
String respContent = null;
while (true) {
try {
id = (String) map.get("Id");
String url_fileName = (String) map.get("localUrl");
// 判断路径是否为空
if (StringUtils.isNotBlank(url_fileName)) {
String filePath = Const.SERVER_FILE_PATH + "/" + url_fileName;
File file = new File(filePath);
boolean exists = file.exists();
if (!exists) {
log.info("文件不存在");
break;
}
String fileName = file.getName();
// 拼接url
String url = finalUploadUrl + String.format(Const.SF_UPLOAD_FILE_URL, api);
HttpPost httpPost = new HttpPost(url);
httpPost.setHeader("Authorization", "Bearer " + token);
httpPost.setHeader("connection", "keep-alive");
JSONObject credentialsJsonParam = new JSONObject();
credentialsJsonParam.put("Name", fileName);
MultipartEntityBuilder builder = MultipartEntityBuilder.create();
builder.addTextBody("data", credentialsJsonParam.toJSONString(), ContentType.APPLICATION_JSON);
builder.addBinaryBody("Body", new FileInputStream(file), ContentType.APPLICATION_OCTET_STREAM, fileName);
HttpEntity entity = builder.build();
httpPost.setEntity(entity);
CloseableHttpClient httpClient = HttpClients.createDefault();
response = httpClient.execute(httpPost);
if (response != null && response.getStatusLine() != null && response.getStatusLine().getStatusCode() < 400) {
HttpEntity he = response.getEntity();
if (he != null) {
respContent = EntityUtils.toString(he, "UTF-8");
String newId = JSONObject.parseObject(respContent).get("id").toString();
Map<String, Object> paramMap = Maps.newHashMap();
paramMap.put("key", "new_id");
paramMap.put("value", newId);
maps.add(paramMap);
}
} else {
throw new RuntimeException("Document文件上传失败地址或参数异常");
}
}
Map<String, Object> paramMap = Maps.newHashMap();
paramMap.put("key", "is_upload");
paramMap.put("value", isUpload);
maps.add(paramMap);
customMapper.updateById(api, maps, id);
TimeUnit.MILLISECONDS.sleep(1);
break;
} catch (Exception throwable) {
log.error("upload file error, id: {}", id, throwable);
failCount++;
if (Const.MAX_FAIL_COUNT < failCount) {
{
Map<String, Object> paramMap = Maps.newHashMap();
paramMap.put("key", "is_upload");
paramMap.put("value", 2);
maps.add(paramMap);
}
customMapper.updateById(api, maps, id);
break;
}
if (response != null) {
try {
response.close();
} catch (IOException e) {
log.error("exception message", e);
}
}
try {
TimeUnit.SECONDS.sleep(30);
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
}
}
}
salesforceExecutor.waitForFutures(futures.toArray(new Future<?>[]{}));
log.info("upload file success api:{}, field:{}", api, field);
} catch (Throwable throwable) {
log.error("upload file error", throwable);
salesforceExecutor.remove(futures.toArray(new Future<?>[]{}));
} finally {
// 把is_upload为2的重置为0
List<Map<String, Object>> maps = Lists.newArrayList();
Map<String, Object> paramMap = Maps.newHashMap();
paramMap.put("key", "is_upload");
paramMap.put("value", 0);
maps.add(paramMap);
customMapper.update(maps, api, "is_upload = 2");
}
}
} }

View File

@ -3,13 +3,13 @@ spring:
type: com.zaxxer.hikari.HikariDataSource type: com.zaxxer.hikari.HikariDataSource
driver-class-name: com.mysql.cj.jdbc.Driver driver-class-name: com.mysql.cj.jdbc.Driver
# 携科 # 携科
url: jdbc:mysql://127.0.0.1:3306/xieke?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=Asia/Shanghai # url: jdbc:mysql://127.0.0.1:3306/xieke?useUnicode=true&allowPublicKeyRetrieval=true&characterEncoding=utf8&useSSL=false&serverTimezone=Asia/Shanghai
# username: root
# password: root
# cook
url: jdbc:mysql://127.0.0.1:3306/longten?useUnicode=true&allowPublicKeyRetrieval=true&characterEncoding=utf8&useSSL=false&serverTimezone=Asia/Shanghai
username: root username: root
password: Celnet2025.QY password: Celnet2025.QY
# cook
# url: jdbc:mysql://127.0.0.1:3306/cook_1?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=Asia/Shanghai
# username: root
# password: celnet@2025.bln
mail: mail:
host: smtp.163.com host: smtp.163.com

View File

@ -3,13 +3,13 @@ spring:
type: com.zaxxer.hikari.HikariDataSource type: com.zaxxer.hikari.HikariDataSource
driver-class-name: com.mysql.cj.jdbc.Driver driver-class-name: com.mysql.cj.jdbc.Driver
# 携科 # 携科
url: jdbc:mysql://127.0.0.1:3306/xieke?useUnicode=true&allowPublicKeyRetrieval=true&characterEncoding=utf8&useSSL=false&serverTimezone=Asia/Shanghai # url: jdbc:mysql://127.0.0.1:3306/xieke?useUnicode=true&allowPublicKeyRetrieval=true&characterEncoding=utf8&useSSL=false&serverTimezone=Asia/Shanghai
username: root
password: root
# cook
# url: jdbc:mysql://127.0.0.1:3306/cook_1?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=Asia/Shanghai
# username: root # username: root
# password: celnet@2025.bln # password: root
# cook
url: jdbc:mysql://127.0.0.1:3306/longten?useUnicode=true&allowPublicKeyRetrieval=true&characterEncoding=utf8&useSSL=false&serverTimezone=Asia/Shanghai
username: root
password: Celnet2025.QY
mail: mail:
host: smtp.mxhichina.com host: smtp.mxhichina.com
port: 465 port: 465

View File

@ -3,13 +3,13 @@ spring:
type: com.zaxxer.hikari.HikariDataSource type: com.zaxxer.hikari.HikariDataSource
driver-class-name: com.mysql.cj.jdbc.Driver driver-class-name: com.mysql.cj.jdbc.Driver
# 携科 # 携科
url: jdbc:mysql://127.0.0.1:3306/xieke?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=Asia/Shanghai # url: jdbc:mysql://127.0.0.1:3306/xieke?useUnicode=true&allowPublicKeyRetrieval=true&characterEncoding=utf8&useSSL=false&serverTimezone=Asia/Shanghai
# username: root
# password: root
# cook
url: jdbc:mysql://127.0.0.1:3306/longten?useUnicode=true&allowPublicKeyRetrieval=true&characterEncoding=utf8&useSSL=false&serverTimezone=Asia/Shanghai
username: root username: root
password: Celnet2025.QY password: Celnet2025.QY
# cook
# url: jdbc:mysql://127.0.0.1:3306/cook_1?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=Asia/Shanghai
# username: root
# password: celnet@2025.bln
mail: mail:
host: smtp.163.com host: smtp.163.com