齊
6 days ago
14 changed files with 840 additions and 71 deletions
@ -0,0 +1,109 @@
@@ -0,0 +1,109 @@
|
||||
package apelet.association.controller; |
||||
|
||||
import apelet.association.utils.DataTransformationUtil; |
||||
import apelet.common.core.object.ObjectCollection; |
||||
import apelet.common.core.object.ObjectValue; |
||||
import apelet.common.core.object.ResponseResult; |
||||
import apelet.common.generator.utils.OrmGenDataSourceUtil; |
||||
import apelet.common.orm.impl.Filter; |
||||
import apelet.common.orm.impl.FilterItem; |
||||
import apelet.common.orm.impl.Selector; |
||||
import cn.hutool.core.util.ObjectUtil; |
||||
import com.alibaba.fastjson.JSONObject; |
||||
import org.apache.commons.lang3.StringUtils; |
||||
import org.springframework.beans.factory.annotation.Autowired; |
||||
import org.springframework.web.bind.annotation.*; |
||||
|
||||
import java.util.*; |
||||
import java.util.stream.Collectors; |
||||
|
||||
@RestController |
||||
@RequestMapping("/tenantadmin/activity") |
||||
public class AssociationActivitiesController { |
||||
|
||||
@Autowired |
||||
private OrmGenDataSourceUtil ormGenDataSourceUtil; |
||||
|
||||
/** |
||||
* 签到/签退 |
||||
* |
||||
* @param jsonObject |
||||
* @return |
||||
*/ |
||||
@PostMapping("/checkInOut") |
||||
public ResponseResult<?> activityCheckIn(@RequestBody JSONObject jsonObject) throws Exception { |
||||
String activityId = jsonObject.getString("activityId"); |
||||
String activityName = jsonObject.getString("activityName"); |
||||
if (StringUtils.isEmpty(activityId) || StringUtils.isEmpty(activityName)) { |
||||
return ResponseResult.error("500", "活动信息为空!!!"); |
||||
} |
||||
String signInPerson = jsonObject.getString("signInPerson"); |
||||
if (StringUtils.isEmpty(signInPerson)) { |
||||
return ResponseResult.error("500", "签到人为空!!!"); |
||||
} |
||||
String phoneNumber = jsonObject.getString("phoneNumber"); |
||||
if (StringUtils.isEmpty(phoneNumber)) { |
||||
return ResponseResult.error("500", "用户手机号为空!!!"); |
||||
} |
||||
|
||||
String signInUnit = jsonObject.getString("signInUnit"); |
||||
if (StringUtils.isEmpty(signInUnit)) { |
||||
return ResponseResult.error("500", "签到单位为空!!!"); |
||||
} |
||||
Integer type = jsonObject.getInteger("type"); |
||||
|
||||
Filter filter = new Filter(); |
||||
filter.add(new FilterItem("unit_name", FilterItem.equals, signInUnit)); |
||||
ObjectCollection collection = ormGenDataSourceUtil.query("membership_apply", filter, new Selector()); |
||||
|
||||
if (collection == null || collection.isEmpty()) { |
||||
return ResponseResult.error("500", "签到单位不存在, 请检查签到单位!!!"); |
||||
} |
||||
|
||||
ObjectCollection managerCollection = ormGenDataSourceUtil.query("membership_manager", new Filter(), new Selector()); |
||||
List<Map<String, Object>> managerList = DataTransformationUtil.objectCollectionToList(managerCollection); |
||||
Map<String, Map<String, Object>> mapMap = managerList.stream().collect(Collectors.toMap(map -> String.valueOf(map.get("id")), map -> map, (oldValue, newValue) -> oldValue)); |
||||
|
||||
ObjectValue member = null; |
||||
for (int i = 0; i < collection.size(); i++) { |
||||
ObjectValue object = collection.getObject(i); |
||||
ObjectValue manger = object.getObjectValue("membership_manger_id"); |
||||
Map<String, Object> map = mapMap.get(manger != null && manger.get("id") != null ? manger.get("id") : ""); |
||||
Object phone = map.get("phone"); |
||||
if (ObjectUtil.equal(phone, phoneNumber)) { |
||||
member = object; |
||||
} |
||||
} |
||||
if (member == null) { |
||||
return ResponseResult.error("500", signInUnit + " 签到单位下签到人不存在!!!"); |
||||
} |
||||
if (type == 1) { |
||||
//签到
|
||||
ObjectValue value = new ObjectValue("check_in_out"); |
||||
value.put("name", signInPerson); |
||||
value.put("phone", phoneNumber); |
||||
value.put("membership_id", member); |
||||
value.put("create_time", new Date()); |
||||
value.put("create_time", new Date()); |
||||
value.put("parent_id", activityId); |
||||
value.put("check_in", 1); |
||||
ormGenDataSourceUtil.addNew(value.getTableName(), value); |
||||
} |
||||
if (type == 2) { |
||||
//签退
|
||||
filter = new Filter(); |
||||
filter.add(new FilterItem("membership_id", FilterItem.equals, member.get("id"))); |
||||
filter.add(new FilterItem("parent_id", FilterItem.equals, activityId)); |
||||
filter.add(new FilterItem("check_in", FilterItem.equals, 1)); |
||||
collection = ormGenDataSourceUtil.query("check_in_out", filter, new Selector()); |
||||
if (collection != null && !collection.isEmpty()) { |
||||
ObjectValue object = collection.getObject(0); |
||||
object.put("check_out", 1); |
||||
ormGenDataSourceUtil.update("check_in_out", object, new Selector()); |
||||
} |
||||
} |
||||
return ResponseResult.success("签到成功!!!"); |
||||
} |
||||
|
||||
|
||||
} |
||||
@ -0,0 +1,49 @@
@@ -0,0 +1,49 @@
|
||||
package apelet.association.plugin.active; |
||||
|
||||
import apelet.common.core.exception.MyRuntimeException; |
||||
import apelet.common.core.object.ObjectValue; |
||||
import apelet.common.core.util.ApplicationContextHolder; |
||||
import apelet.common.generator.utils.OrmGenDataSourceUtil; |
||||
import apelet.common.online.abstractplugin.ListPlugin; |
||||
import apelet.common.online.dto.OnlineEventPluginExecuteDto; |
||||
import apelet.common.online.model.constant.AttributeEnum; |
||||
import apelet.common.orm.impl.Selector; |
||||
|
||||
/** |
||||
* 活动日程插件 |
||||
*/ |
||||
public class ActivitySchedulePlugin extends ListPlugin { |
||||
|
||||
private OrmGenDataSourceUtil ormGenDataSourceUtil; |
||||
|
||||
public ActivitySchedulePlugin() { |
||||
ormGenDataSourceUtil = ApplicationContextHolder.getBean(OrmGenDataSourceUtil.class); |
||||
} |
||||
|
||||
|
||||
@Override |
||||
public void formCreated(String widgetVariableName, ObjectValue objectValue) { |
||||
OnlineEventPluginExecuteDto dto = getDto(); |
||||
this.setWidgetAttribute("id", AttributeEnum.SHOW, false); |
||||
|
||||
} |
||||
|
||||
@Override |
||||
public void change(String widgetVariableName, ObjectValue objectValue) { |
||||
|
||||
} |
||||
|
||||
@Override |
||||
public void buttonTriggered(String widgetVariableName, ObjectValue objectValue) { |
||||
if ("保存".equals(widgetVariableName)) { |
||||
try { |
||||
ObjectValue value = ormGenDataSourceUtil.queryOne("association_activity", objectValue.get("id")); |
||||
value.put("activity_schedule", objectValue.getObjectCollection("activity_schedule")); |
||||
ormGenDataSourceUtil.update("association_activity", value, new Selector()); |
||||
} catch (Exception e) { |
||||
throw new MyRuntimeException(e.getMessage()); |
||||
} |
||||
this.cancelOperate(); |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,153 @@
@@ -0,0 +1,153 @@
|
||||
package apelet.association.plugin.active; |
||||
|
||||
import apelet.association.utils.MyFileUtil; |
||||
import apelet.association.utils.QRCodeUtil; |
||||
import apelet.common.core.exception.MyRuntimeException; |
||||
import apelet.common.core.object.*; |
||||
import apelet.common.core.upload.*; |
||||
import apelet.common.core.util.ApplicationContextHolder; |
||||
import apelet.common.generator.utils.OrmGenDataSourceUtil; |
||||
import apelet.common.online.abstractplugin.ListPlugin; |
||||
import apelet.common.online.dto.OnlineEventPluginExecuteDto; |
||||
import apelet.common.online.model.OnlineDatasource; |
||||
import apelet.common.online.model.OnlineTable; |
||||
import apelet.common.online.model.constant.AttributeEnum; |
||||
import apelet.common.online.service.OnlineColumnService; |
||||
import apelet.common.online.service.OnlineDatasourceService; |
||||
import apelet.common.online.service.OnlineTableService; |
||||
import apelet.common.orm.impl.*; |
||||
import com.alibaba.fastjson.JSON; |
||||
import com.alibaba.fastjson.JSONObject; |
||||
import lombok.extern.slf4j.Slf4j; |
||||
import org.apache.commons.io.FileUtils; |
||||
import org.apache.commons.lang3.StringUtils; |
||||
import org.springframework.web.multipart.MultipartFile; |
||||
|
||||
import java.io.*; |
||||
import java.util.HashMap; |
||||
import java.util.Map; |
||||
|
||||
/** |
||||
* 活动签到配置插件 |
||||
*/ |
||||
|
||||
@Slf4j |
||||
public class ActivitySignConfigPlugin extends ListPlugin { |
||||
|
||||
private MyFileUtil myFileUtil; |
||||
private QRCodeUtil qrCodeUtil; |
||||
|
||||
private OnlineTableService onlineTableService; |
||||
private OnlineColumnService onlineColumnService; |
||||
private OrmGenDataSourceUtil ormGenDataSourceUtil; |
||||
|
||||
|
||||
private OnlineDatasourceService onlineDatasourceService; |
||||
|
||||
|
||||
public ActivitySignConfigPlugin() { |
||||
myFileUtil = ApplicationContextHolder.getBean(MyFileUtil.class); |
||||
qrCodeUtil = ApplicationContextHolder.getBean(QRCodeUtil.class); |
||||
onlineTableService = ApplicationContextHolder.getBean(OnlineTableService.class); |
||||
onlineColumnService = ApplicationContextHolder.getBean(OnlineColumnService.class); |
||||
ormGenDataSourceUtil = ApplicationContextHolder.getBean(OrmGenDataSourceUtil.class); |
||||
onlineDatasourceService = ApplicationContextHolder.getBean(OnlineDatasourceService.class); |
||||
} |
||||
|
||||
|
||||
@Override |
||||
public void formCreated(String widgetVariableName, ObjectValue objectValue) { |
||||
OnlineEventPluginExecuteDto dto = getDto(); |
||||
Map parentParams = dto.getParentParams(); |
||||
Object activityId = parentParams.get("id"); |
||||
this.setWidgetAttribute("activityId", AttributeEnum.VALUE_CHANGE, parentParams); |
||||
Filter filter = new Filter(); |
||||
filter.add(new FilterItem("activity_id", FilterItem.equals, activityId)); |
||||
Object data = ormGenDataSourceUtil.queryData("orm", "activity_sign_config", filter, new Selector(), new Sorter(), new HashMap<>()); |
||||
if (data instanceof ObjectCollection) { |
||||
ObjectCollection collection = (ObjectCollection) data; |
||||
ObjectValue object = collection.getObject(0); |
||||
this.setWidgetAttribute("id", AttributeEnum.VALUE_CHANGE, object.get("id")); |
||||
|
||||
this.setWidgetAttribute("checkAction", AttributeEnum.VALUE_CHANGE, object.get("check_action")); |
||||
|
||||
this.setWidgetAttribute("scanCheck", AttributeEnum.VALUE_CHANGE, object.getInt("scan_check")); |
||||
this.setWidgetAttribute("positionLimit", AttributeEnum.VALUE_CHANGE, object.getInt("position_limit")); |
||||
this.setWidgetAttribute("duplicateCheck", AttributeEnum.VALUE_CHANGE, object.getInt("duplicate_check")); |
||||
this.setWidgetAttribute("scanCodeReview", AttributeEnum.VALUE_CHANGE, object.getInt("scan_code_review")); |
||||
this.setWidgetAttribute("checkCertificate", AttributeEnum.VALUE_CHANGE, object.getInt("id")); |
||||
|
||||
try { |
||||
JSONObject signInCode = JSON.parseObject(object.getString("sign_in_code")); |
||||
this.setWidgetAttribute("signInCode", AttributeEnum.VALUE_CHANGE, signInCode); |
||||
|
||||
JSONObject signOutCode = JSON.parseObject(object.getString("sign_out_code")); |
||||
this.setWidgetAttribute("signOutCode", AttributeEnum.VALUE_CHANGE, signOutCode); |
||||
} catch (Exception ee) { |
||||
log.error("二维码数据转换失败!!!"); |
||||
ee.printStackTrace(); |
||||
} |
||||
|
||||
} |
||||
this.setWidgetAttribute("id", AttributeEnum.SHOW, false); |
||||
this.setWidgetAttribute("activityId", AttributeEnum.SHOW, false); |
||||
} |
||||
|
||||
|
||||
@Override |
||||
public void buttonTriggered(String widgetVariableName, ObjectValue objectValue) { |
||||
OnlineEventPluginExecuteDto dto = getDto(); |
||||
Model model = dto.getModel(); |
||||
if ("button1778742711763".equals(widgetVariableName)) { |
||||
|
||||
if (StringUtils.isNotBlank(objectValue.getString("sign_in_code")) && |
||||
StringUtils.isNotBlank(objectValue.getString("sign_out_code"))) { |
||||
showWarningMessage("当前已存在二维码, 请勿重复生产!!!"); |
||||
return; |
||||
} |
||||
// 生成二维码
|
||||
String savePath = System.getProperty("user.dir") + "\\zz-resource\\qrcode"; |
||||
|
||||
File checkOutFile; |
||||
File checkInFile; |
||||
try { |
||||
Object activityId = objectValue.get("id"); |
||||
Object activityName = objectValue.get("name"); |
||||
|
||||
OnlineDatasource onlineDatasource = onlineDatasourceService.getById(model.getDatasourceId()); |
||||
OnlineTable table = onlineTableService.getOnlineTableFromCache(onlineDatasource.getMasterTableId()); |
||||
|
||||
String checkIn = "https://localhost:8080/#/pages/login/routerHdView?activityId=" + activityId + "&activityName=" + activityName + "&type=" + 1; |
||||
byte[] checkInQr = qrCodeUtil.generateQrCode(checkIn, 350, 350); |
||||
checkInFile = MyFileUtil.writeToFile(checkInQr, savePath, "qrcode-" + System.currentTimeMillis() + ".png"); |
||||
MultipartFile checkInMultipart = MyFileUtil.readFileAsMultipartFile(checkInFile, "image/png"); |
||||
UploadResponseInfo checkInUpload = myFileUtil.uploadMyFile(table, "sign_in_code", true, checkInMultipart); |
||||
this.setWidgetAttribute("signInCode", AttributeEnum.VALUE_CHANGE, checkInUpload); |
||||
|
||||
String checkOut = "https://localhost:8080/#/pages/login/routerHdView?activityId=" + activityId + "&activityName=" + activityName + "&type=" + 2; |
||||
byte[] checkOutQr = qrCodeUtil.generateQrCode(checkOut, 300, 300); |
||||
checkOutFile = MyFileUtil.writeToFile(checkOutQr, savePath, "qrcode-" + System.currentTimeMillis() + ".png"); |
||||
MultipartFile checkOutMultipart = MyFileUtil.readFileAsMultipartFile(checkOutFile, "image/png"); |
||||
UploadResponseInfo checkOutUpload = myFileUtil.uploadMyFile(table, "sign_out_code", true, checkOutMultipart); |
||||
this.setWidgetAttribute("signOutCode", AttributeEnum.VALUE_CHANGE, checkOutUpload); |
||||
} catch (Exception e) { |
||||
e.printStackTrace(); |
||||
throw new MyRuntimeException("二维码生成失败, 请联系管理员!!!"); |
||||
} finally { |
||||
try { |
||||
FileUtils.cleanDirectory(new File(savePath)); |
||||
} catch (IOException e) { |
||||
e.printStackTrace(); |
||||
System.err.println("文件刪除失敗!!!"); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public void change(String widgetVariableName, ObjectValue objectValue) { |
||||
super.formCreated(widgetVariableName, objectValue); |
||||
} |
||||
|
||||
|
||||
} |
||||
@ -0,0 +1,73 @@
@@ -0,0 +1,73 @@
|
||||
package apelet.association.task; |
||||
|
||||
|
||||
import apelet.common.core.object.ObjectCollection; |
||||
import apelet.common.core.object.ObjectValue; |
||||
import apelet.common.generator.utils.OrmGenDataSourceUtil; |
||||
import apelet.common.orm.impl.Filter; |
||||
import apelet.common.orm.impl.Selector; |
||||
import org.springframework.beans.factory.annotation.Autowired; |
||||
import org.springframework.scheduling.annotation.Scheduled; |
||||
import org.springframework.stereotype.Component; |
||||
|
||||
import java.time.LocalDate; |
||||
import java.time.ZoneId; |
||||
import java.time.temporal.ChronoUnit; |
||||
import java.util.Date; |
||||
|
||||
@Component |
||||
public class UserRenewalReminderTask { |
||||
|
||||
@Autowired |
||||
OrmGenDataSourceUtil ormGenDataSourceUtil; |
||||
|
||||
// 提前 30 天提醒
|
||||
private static final int REMIND_BEFORE_DAYS = 30; |
||||
// 会员年度天数（固定 365）
|
||||
private static final int YEAR_DAYS = 365; |
||||
|
||||
@Scheduled(cron = "0 0 1 * * ?") |
||||
public void sendRemind(){ |
||||
ObjectCollection collection = ormGenDataSourceUtil.query("membership_apply", new Filter(), new Selector()); |
||||
if(collection == null){ |
||||
return; |
||||
} |
||||
for (int i = 0; i < collection.size(); i++) { |
||||
ObjectValue objectValue = collection.getObject(i); |
||||
Date registTime = objectValue.getDate("regist_time"); |
||||
if (needRemind(registTime)) { |
||||
// TODO 发送消息提示续费
|
||||
|
||||
} |
||||
|
||||
} |
||||
|
||||
|
||||
} |
||||
|
||||
|
||||
public boolean needRemind(Date date) { |
||||
LocalDate today = LocalDate.now(); |
||||
LocalDate join = date.toInstant().atZone(ZoneId.systemDefault()).toLocalDate(); |
||||
|
||||
if (today.isBefore(join)) { |
||||
return false; |
||||
} |
||||
// 入会到今天总天数
|
||||
long totalDays = ChronoUnit.DAYS.between(join, today); |
||||
|
||||
// 第几个年度（从 1 开始）
|
||||
int yearIndex = (int) (totalDays / YEAR_DAYS) + 1; |
||||
|
||||
// 该年度到期日
|
||||
LocalDate expireDate = join.plusDays((long) YEAR_DAYS * yearIndex); |
||||
|
||||
// 提醒开始日：到期前 30 天
|
||||
LocalDate remindStart = expireDate.minusDays(REMIND_BEFORE_DAYS); |
||||
|
||||
// 今天在 [remindStart, expireDate] 区间内，则提醒
|
||||
return !today.isBefore(remindStart) && !today.isAfter(expireDate); |
||||
} |
||||
|
||||
|
||||
} |
||||
@ -0,0 +1,27 @@
@@ -0,0 +1,27 @@
|
||||
package apelet.association.utils; |
||||
|
||||
import apelet.common.core.object.ObjectCollection; |
||||
import apelet.common.core.object.ObjectValue; |
||||
import com.alibaba.fastjson.JSONObject; |
||||
|
||||
import java.util.ArrayList; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
|
||||
public class DataTransformationUtil { |
||||
|
||||
public static List<Map<String, Object>> objectCollectionToList(ObjectCollection oc) { |
||||
|
||||
List<Map<String, Object>> list = new ArrayList<>(); |
||||
if(oc == null || oc.isEmpty()){ |
||||
return list; |
||||
} |
||||
for (int i = 0; i < oc.size(); i++) { |
||||
ObjectValue object = oc.getObject(i); |
||||
JSONObject jsonObject = new JSONObject(object.getValues()); |
||||
list.add(jsonObject); |
||||
} |
||||
return list; |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,211 @@
@@ -0,0 +1,211 @@
|
||||
package apelet.association.utils; |
||||
|
||||
import apelet.common.core.constant.ErrorCodeEnum; |
||||
import apelet.common.core.object.ResponseResult; |
||||
import apelet.common.core.upload.*; |
||||
import apelet.common.online.config.OnlineProperties; |
||||
import apelet.common.online.model.OnlineColumn; |
||||
import apelet.common.online.model.OnlineTable; |
||||
import apelet.common.online.model.constant.FieldKind; |
||||
|
||||
import liquibase.util.BooleanUtil; |
||||
import org.springframework.beans.factory.annotation.Autowired; |
||||
import org.springframework.http.MediaType; |
||||
import org.springframework.stereotype.Component; |
||||
import org.springframework.web.multipart.MultipartFile; |
||||
|
||||
import java.io.*; |
||||
import java.nio.file.*; |
||||
|
||||
@Component |
||||
public class MyFileUtil { |
||||
|
||||
@Autowired |
||||
private UpDownloaderFactory upDownloaderFactory; |
||||
|
||||
@Autowired |
||||
private OnlineProperties onlineProperties; |
||||
|
||||
|
||||
public UploadResponseInfo uploadMyFile(OnlineTable table, String fieldName, Boolean asImg, MultipartFile multipartFile) throws IOException { |
||||
ResponseResult<OnlineColumn> verifyResult = this.doVerifyUpDownloadFileColumn(table, fieldName, asImg); |
||||
OnlineColumn uploadColumn = verifyResult.getData(); |
||||
UploadStoreTypeEnum uploadStoreType = UploadStoreTypeEnum.values()[uploadColumn.getUploadFileSystemType()]; |
||||
BaseUpDownloader upDownloader = upDownloaderFactory.get(uploadStoreType); |
||||
return upDownloader.doUpload(null, onlineProperties.getUploadFileBaseDir(), table.getModelName(), fieldName, asImg, multipartFile); |
||||
} |
||||
|
||||
|
||||
public static MultipartFile createMultipartFile(String filePath, String fileName) { |
||||
Path path = Paths.get(filePath); |
||||
return new MultipartFile() { |
||||
@Override |
||||
public String getName() { |
||||
return fileName; |
||||
} |
||||
|
||||
@Override |
||||
public String getOriginalFilename() { |
||||
return fileName; |
||||
} |
||||
|
||||
@Override |
||||
public String getContentType() { |
||||
try { |
||||
return Files.probeContentType(path); |
||||
} catch (IOException e) { |
||||
return MediaType.APPLICATION_OCTET_STREAM_VALUE; |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public boolean isEmpty() { |
||||
return false; |
||||
} |
||||
|
||||
@Override |
||||
public long getSize() { |
||||
try { |
||||
return Files.size(path); |
||||
} catch (IOException e) { |
||||
return 0; |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public byte[] getBytes() throws IOException { |
||||
return Files.readAllBytes(path); |
||||
} |
||||
|
||||
@Override |
||||
public InputStream getInputStream() throws IOException { |
||||
return Files.newInputStream(path); |
||||
} |
||||
|
||||
@Override |
||||
public void transferTo(File dest) throws IOException { |
||||
Files.copy(path, dest.toPath(), StandardCopyOption.REPLACE_EXISTING); |
||||
} |
||||
}; |
||||
} |
||||
|
||||
|
||||
public static void saveFileToLocal(byte[] qrCodeByte, String savePath, String fileName) throws IOException { |
||||
// 1. 创建文件夹
|
||||
File folder = new File(savePath); |
||||
if (!folder.exists()) { |
||||
folder.mkdirs(); |
||||
} |
||||
// 2. 最终文件
|
||||
File file = new File(folder, fileName); |
||||
// 3. 写入文件（try-with-resources 会自动关闭流，最安全）
|
||||
try (FileOutputStream fos = new FileOutputStream(file); |
||||
ByteArrayInputStream bais = new ByteArrayInputStream(qrCodeByte)) { |
||||
byte[] buffer = new byte[1024]; |
||||
int len; |
||||
while ((len = bais.read(buffer)) != -1) { |
||||
fos.write(buffer, 0, len); |
||||
} |
||||
fos.flush(); |
||||
} |
||||
} |
||||
|
||||
private ResponseResult<OnlineColumn> doVerifyUpDownloadFileColumn( |
||||
OnlineTable table, String fieldName, Boolean asImage) { |
||||
OnlineColumn column = null; |
||||
for (OnlineColumn c : table.getColumnMap().values()) { |
||||
if (c.getColumnName().equals(fieldName)) { |
||||
column = c; |
||||
break; |
||||
} |
||||
} |
||||
if (column == null) { |
||||
return ResponseResult.error(ErrorCodeEnum.INVALID_DATA_FIELD); |
||||
} |
||||
if (BooleanUtil.isTrue(asImage)) { |
||||
if (column.getFieldKind() != FieldKind.UPLOAD_IMAGE) { |
||||
return ResponseResult.error(ErrorCodeEnum.INVALID_UPLOAD_FIELD); |
||||
} |
||||
} else { |
||||
if (column.getFieldKind() != FieldKind.UPLOAD) { |
||||
return ResponseResult.error(ErrorCodeEnum.INVALID_UPLOAD_FIELD); |
||||
} |
||||
} |
||||
return ResponseResult.success(column); |
||||
} |
||||
|
||||
|
||||
public static File writeToFile(byte[] qrBytes, String dirPath, String fileName) throws IOException { |
||||
Path dir = Paths.get(dirPath); |
||||
if (!Files.exists(dir)) { |
||||
Files.createDirectories(dir); |
||||
} |
||||
Path filePath = dir.resolve(fileName); |
||||
try (FileOutputStream fos = new FileOutputStream(filePath.toFile())) { |
||||
fos.write(qrBytes); |
||||
} |
||||
return filePath.toFile(); |
||||
} |
||||
|
||||
|
||||
public static MultipartFile readFileAsMultipartFile(File file, String contentType) throws IOException { |
||||
return new CustomMultipartFile(file, contentType); |
||||
} |
||||
|
||||
// ==================== 步骤3：获取 InputStream ====================
|
||||
public static InputStream getInputStream(MultipartFile multipartFile) throws IOException { |
||||
return multipartFile.getInputStream(); |
||||
} |
||||
|
||||
// ==================== 自定义 MultipartFile 实现 ====================
|
||||
public static class CustomMultipartFile implements MultipartFile { |
||||
|
||||
private final File file; |
||||
private final String contentType; |
||||
|
||||
public CustomMultipartFile(File file, String contentType) { |
||||
this.file = file; |
||||
this.contentType = contentType; |
||||
} |
||||
|
||||
@Override |
||||
public String getName() { |
||||
return file.getName(); |
||||
} |
||||
|
||||
@Override |
||||
public String getOriginalFilename() { |
||||
return file.getName(); |
||||
} |
||||
|
||||
@Override |
||||
public String getContentType() { |
||||
return contentType; |
||||
} |
||||
|
||||
@Override |
||||
public boolean isEmpty() { |
||||
return file.length() == 0; |
||||
} |
||||
|
||||
@Override |
||||
public long getSize() { |
||||
return file.length(); |
||||
} |
||||
|
||||
@Override |
||||
public byte[] getBytes() throws IOException { |
||||
return Files.readAllBytes(file.toPath()); |
||||
} |
||||
|
||||
@Override |
||||
public InputStream getInputStream() throws IOException { |
||||
return new FileInputStream(file); |
||||
} |
||||
|
||||
@Override |
||||
public void transferTo(File dest) throws IOException { |
||||
Files.copy(file.toPath(), dest.toPath()); |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,78 @@
@@ -0,0 +1,78 @@
|
||||
package apelet.association.utils; |
||||
|
||||
import com.google.zxing.BarcodeFormat; |
||||
import com.google.zxing.EncodeHintType; |
||||
import com.google.zxing.MultiFormatWriter; |
||||
import com.google.zxing.client.j2se.MatrixToImageWriter; |
||||
import com.google.zxing.common.BitMatrix; |
||||
import com.google.zxing.qrcode.QRCodeWriter; |
||||
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel; |
||||
import lombok.var; |
||||
import org.springframework.stereotype.Component; |
||||
|
||||
import javax.imageio.ImageIO; |
||||
import javax.servlet.http.HttpServletResponse; |
||||
import java.awt.image.BufferedImage; |
||||
import java.io.ByteArrayOutputStream; |
||||
import java.io.File; |
||||
import java.io.OutputStream; |
||||
import java.util.HashMap; |
||||
import java.util.Map; |
||||
|
||||
@Component |
||||
public class QRCodeUtil { |
||||
|
||||
|
||||
/** |
||||
* 生成二维码并直接输出到响应流中 |
||||
* |
||||
* @param url 扫码后要跳转的网址 |
||||
* @param response HTTP响应 |
||||
*/ |
||||
public void generateQRCode(String url, HttpServletResponse response) { |
||||
try { |
||||
// 设置响应类型为图片
|
||||
response.setContentType("image/png"); |
||||
OutputStream os = response.getOutputStream(); |
||||
|
||||
// 二维码参数配置
|
||||
Map<EncodeHintType, Object> hints = new HashMap<>(); |
||||
hints.put(EncodeHintType.CHARACTER_SET, "UTF-8"); // 防止中文乱码
|
||||
hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H); // 高容错率(30%)
|
||||
hints.put(EncodeHintType.MARGIN, 1); // 白边宽度
|
||||
|
||||
// 生成二维码矩阵 (宽350，高350)
|
||||
BitMatrix bitMatrix = new MultiFormatWriter().encode(url, BarcodeFormat.QR_CODE, 350, 350, hints); |
||||
|
||||
// 将矩阵转换成图片并输出
|
||||
MatrixToImageWriter.writeToStream(bitMatrix, "PNG", os); |
||||
|
||||
} catch (Exception e) { |
||||
e.printStackTrace(); |
||||
throw new RuntimeException("生成二维码失败"); |
||||
} |
||||
} |
||||
|
||||
|
||||
|
||||
/** |
||||
* 生成二维码字节流（可直接返回给前端） |
||||
* @param content 二维码内容：即你的网页 URL |
||||
* @param width 宽度 |
||||
* @param height 高度 |
||||
* @return 图片字节数组 |
||||
*/ |
||||
public byte[] generateQrCode(String content, int width, int height) throws Exception { |
||||
Map<EncodeHintType, Object> hints = new HashMap<>(); |
||||
hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.L); |
||||
hints.put(EncodeHintType.CHARACTER_SET, "UTF-8"); |
||||
// 缩小白边（可选）
|
||||
hints.put(EncodeHintType.MARGIN, 1); |
||||
QRCodeWriter writer = new QRCodeWriter(); |
||||
var bitMatrix = writer.encode(content, BarcodeFormat.QR_CODE, width, height, hints); |
||||
BufferedImage image = MatrixToImageWriter.toBufferedImage(bitMatrix); |
||||
ByteArrayOutputStream baos = new ByteArrayOutputStream(); |
||||
ImageIO.write(image, "PNG", baos); |
||||
return baos.toByteArray(); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue