ak/zn-cloud
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

新增 小程序码生成 、 上传接口

Browse Source
This commit is contained in:
furongxin 2024-06-27 19:28:58 +08:00
parent 17934098e1
commit e27451aa23
9 changed files with 225 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
yudao-module-infra/yudao-module-infra-api/src/main/java/cn/iocoder/yudao/module/infra/enums/ErrorCodeConstants.java
View File

@ -69,5 +69,5 @@ public interface ErrorCodeConstants {
ErrorCode DEMO03_STUDENT_NOT_EXISTS = new ErrorCode(1_001_201_007, "学生不存在");
ErrorCode DEMO03_GRADE_NOT_EXISTS = new ErrorCode(1_001_201_008, "学生班级不存在");
ErrorCode DEMO03_GRADE_EXISTS = new ErrorCode(1_001_201_009, "学生班级已存在");
ErrorCode OA_QRCODE_ERROR = new ErrorCode(1_009_301_001, "小程序码生成错误！");
}

60
yudao-module-infra/yudao-module-infra-biz/src/main/java/cn/iocoder/yudao/module/infra/controller/admin/file/FileController.java
View File

@ -3,17 +3,16 @@ package cn.iocoder.yudao.module.infra.controller.admin.file;
import cn.hutool.core.io.IoUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.core.util.URLUtil;
import cn.hutool.json.JSONObject;
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.framework.common.util.servlet.ServletUtils;
import cn.iocoder.yudao.framework.operatelog.core.annotations.OperateLog;
import cn.iocoder.yudao.module.infra.controller.admin.file.vo.file.*;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.BpmFileDO;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.BusinessFileDO;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.FileDO;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.UserFileDO;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.*;
import cn.iocoder.yudao.module.infra.service.file.FileService;
import cn.iocoder.yudao.module.system.api.social.SocialClientApi;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
@ -29,8 +28,13 @@ import javax.annotation.security.PermitAll;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.validation.Valid;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.error;
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
import static cn.iocoder.yudao.module.infra.enums.ErrorCodeConstants.OA_QRCODE_ERROR;
@Tag(name = "管理后台 - 文件存储")
@RestController
@ -42,6 +46,9 @@ public class FileController {
@Resource
private FileService fileService;
@Resource
private SocialClientApi socialClientApi;
@PostMapping("/uploadBpmFileProcessInstanceId")
@Operation(summary = "更新文件的流程实例ID")
@OperateLog(logArgs = false) // 上传文件，没有记录操作日志的必要
@ -121,6 +128,7 @@ public class FileController {
@PostMapping("/businessUpload")
@Operation(summary = "上传业务类型附件【如：工作日/周报附件】")
@OperateLog(logArgs = false) // 上传文件，没有记录操作日志的必要
@PermitAll
public CommonResult<BusinessFileDO> businessUpload(@RequestParam("uploadFiles") MultipartFile file,@RequestParam("businessType")Long businessType) throws Exception {
return success(fileService.createBusinessReturnFile(file,businessType));
}
@ -128,6 +136,7 @@ public class FileController {
@DeleteMapping("/deleteBusinessFile")
@Operation(summary = "删除业务流附件【如：工作日/周报附件】")
@Parameter(name = "url", description = "附件URL地址", required = true)
@PermitAll
public CommonResult<Boolean> deleteBusinessFile(@RequestParam("url") String url) throws Exception {
fileService.deleteBusinessFile(url);
return success(true);
@ -145,6 +154,7 @@ public class FileController {
@PostMapping("/userFileUpload")
@Operation(summary = "上传用户文件")
@OperateLog(logArgs = false) // 上传文件，没有记录操作日志的必要
@PermitAll
public CommonResult<UserFileDO> userFileUpload(@RequestParam("uploadFiles") MultipartFile file) {
return success(fileService.createUserReturnFile(file));
}
@ -156,4 +166,46 @@ public class FileController {
fileService.deleteUserFile(url);
return success(true);
}
@GetMapping("/create-QRCode")
@Operation(summary = "获得入职申请小程序码")
@Parameter(name = "userId", description = "发起人用户编号", required = true, example = "1024")
@Parameter(name = "deptId", description = "部门编号", required = true, example = "1024")
@Parameter(name = "postId", description = "岗位编号", example = "1024")
public CommonResult<QRCodeDO> getQRCode(@RequestParam("userId") Long userId,
@RequestParam("deptId") Long deptId,
@RequestParam(value = "postId", required = false) Long postId) {
// 查询当前部门编号下 是否存在小程序码
QRCodeDO qrCodeDO = fileService.getQRCode(deptId);
// 小程序码不存在，则 生成新的小程序码
if (qrCodeDO == null) {
// 生成小程序码
File QRCode = socialClientApi.getQRCode("subPages/task/myTask", "id=").getCheckedData();
JSONObject object = new JSONObject();
object.set("userId", userId);
object.set("deptId", deptId);
object.set("postId", postId);
try {
FileInputStream fis = new FileInputStream(QRCode);
byte[] content = IoUtil.readBytes(fis);
// 上传小程序码 获得url
qrCodeDO = fileService.createQRCodeFile(deptId, QRCode.getName(), content, object.toString());
} catch (IOException e) {
return error(OA_QRCODE_ERROR);
}
}else { // 存在的时候， 则更新小程序码的生成时间
fileService.updateQRCodeFile(qrCodeDO.getId());
}
return success(qrCodeDO);
}
}

55
yudao-module-infra/yudao-module-infra-biz/src/main/java/cn/iocoder/yudao/module/infra/dal/dataobject/file/QRCodeDO.java Normal file
View File

@ -0,0 +1,55 @@
package cn.iocoder.yudao.module.infra.dal.dataobject.file;
import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableName;
import com.baomidou.mybatisplus.extension.handlers.JacksonTypeHandler;
import lombok.*;
/**
* 文件表
* 每次文件上传，都会记录一条记录到该表中
*
*/
@TableName("system_qr_code")
@Data
@EqualsAndHashCode(callSuper = true)
@ToString(callSuper = true)
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class QRCodeDO extends BaseDO {
/**
* 编号，数据库自增
*/
private Long id;
/**
* 业务编号
*/
private String businessId;
/**
* 业务类型
* 1 入职申请小程序码
*/
private Integer businessType;
/**
* 配置编号
* 关联 {@link FileConfigDO#getId()}
*/
private Long configId;
/**
* 文件路径
*/
private String url;
/**
* 参数值
*/
@TableField(typeHandler = JacksonTypeHandler.class)
private String scene;
}

17
yudao-module-infra/yudao-module-infra-biz/src/main/java/cn/iocoder/yudao/module/infra/dal/mysql/file/QRCodeDOMapper.java Normal file
View File

@ -0,0 +1,17 @@
package cn.iocoder.yudao.module.infra.dal.mysql.file;
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.QRCodeDO;
import org.apache.ibatis.annotations.Mapper;
@Mapper
public interface QRCodeDOMapper extends BaseMapperX<QRCodeDO> {
default QRCodeDO selectByDeptId(Long deptId) {
return selectOne(new LambdaQueryWrapperX<QRCodeDO>()
.eq(QRCodeDO::getBusinessId, deptId)
.eq(QRCodeDO::getBusinessType, 1));
}
}

3
yudao-module-infra/yudao-module-infra-biz/src/main/java/cn/iocoder/yudao/module/infra/framework/rpc/config/RpcConfiguration.java
View File

@ -1,11 +1,12 @@
package cn.iocoder.yudao.module.infra.framework.rpc.config;
import cn.iocoder.yudao.module.system.api.equipment.AttendanceMachineApi;
import cn.iocoder.yudao.module.system.api.social.SocialClientApi;
import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
import org.springframework.cloud.openfeign.EnableFeignClients;
import org.springframework.context.annotation.Configuration;
@Configuration(proxyBeanMethods = false)
@EnableFeignClients(clients = { AdminUserApi.class, AttendanceMachineApi.class })
@EnableFeignClients(clients = { AdminUserApi.class, AttendanceMachineApi.class, SocialClientApi.class })
public class RpcConfiguration {
}

26
yudao-module-infra/yudao-module-infra-biz/src/main/java/cn/iocoder/yudao/module/infra/service/file/FileService.java
View File

@ -5,10 +5,7 @@ import cn.iocoder.yudao.module.infra.controller.admin.file.vo.file.BusinessFileU
import cn.iocoder.yudao.module.infra.controller.admin.file.vo.file.FilePageReqVO;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.module.infra.controller.admin.file.vo.file.UserFileUpdateReqVO;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.BpmFileDO;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.BusinessFileDO;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.FileDO;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.UserFileDO;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.*;
import org.springframework.web.multipart.MultipartFile;
/**
@ -149,4 +146,25 @@ public interface FileService {
* @param updateReqVO 更新信息
*/
void updateUserFileUserId(UserFileUpdateReqVO updateReqVO);
/**
* 保存业务类型的附件
* @param name 文件名
* @param content 字节流
* @return url
*/
QRCodeDO createQRCodeFile(Long deptId, String name, byte[] content, String scene) ;
/**
* 获得小程序码信息
* @param deptId 部门编号
* @return 小程序码信息
*/
QRCodeDO getQRCode(Long deptId);
/**
* 更新 小程序码生成时间
* @param id 编号
*/
void updateQRCodeFile(Long id);
}

55
yudao-module-infra/yudao-module-infra-biz/src/main/java/cn/iocoder/yudao/module/infra/service/file/FileServiceImpl.java
View File

@ -13,10 +13,7 @@ import cn.iocoder.yudao.module.infra.controller.admin.file.vo.file.BpmFileUpload
import cn.iocoder.yudao.module.infra.controller.admin.file.vo.file.BusinessFileUploadReqVO;
import cn.iocoder.yudao.module.infra.controller.admin.file.vo.file.FilePageReqVO;
import cn.iocoder.yudao.module.infra.controller.admin.file.vo.file.UserFileUpdateReqVO;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.BpmFileDO;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.BusinessFileDO;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.FileDO;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.UserFileDO;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.*;
import cn.iocoder.yudao.module.infra.dal.mysql.file.*;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import lombok.SneakyThrows;
@ -24,6 +21,7 @@ import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
import java.time.LocalDateTime;
import java.util.List;
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
@ -52,6 +50,9 @@ public class FileServiceImpl implements FileService {
@Resource
private UserFileMapper userFileMapper;
@Resource
private QRCodeDOMapper qrCodeDOMapper;
@Override
public PageResult<FileDO> getFilePage(FilePageReqVO pageReqVO) {
return fileMapper.selectPage(pageReqVO);
@ -468,4 +469,50 @@ public class FileServiceImpl implements FileService {
// 调用 MyBatis Plus 的 update 方法执行批量更新
userFileMapper.update(null, lambdaUpdateWrapper);
}
@Override
@SneakyThrows
public QRCodeDO createQRCodeFile(Long deptId, String name, byte[] content, String scene) {
long timestamp = System.currentTimeMillis();
// 计算默认的 path 名
String type = FileTypeUtils.getMineType(content, name);
String path = deptId + "_" + timestamp;
String beginPath = name.replace(".", "_");
path = beginPath + "_" + path + "." + FileNameUtil.extName(name);
// 如果 name 为空，则使用 path 填充
if (StrUtil.isEmpty(name)) {
name = path;
}
// 上传到文件存储器
FileClient client = fileConfigService.getMasterFileClient();
Assert.notNull(client, "客户端(master) 不能为空");
String url = client.upload(content, path, type);
// 插入 system_qr_code
QRCodeDO fileDo = new QRCodeDO();
fileDo.setBusinessId(deptId.toString());
fileDo.setBusinessType(1);
fileDo.setConfigId(client.getId());
fileDo.setUrl(url);
fileDo.setScene(scene);
qrCodeDOMapper.insert(fileDo);
return fileDo;
}
@Override
public QRCodeDO getQRCode(Long deptId) {
return qrCodeDOMapper.selectByDeptId(deptId);
}
@Override
public void updateQRCodeFile(Long id) {
QRCodeDO updateDO = new QRCodeDO();
updateDO.setId(id);
updateDO.setUpdateTime(LocalDateTime.now());
qrCodeDOMapper.updateById(updateDO);
}
}

8
yudao-module-system/yudao-module-system-api/src/main/java/cn/iocoder/yudao/module/system/api/social/SocialClientApi.java
View File

@ -12,6 +12,8 @@ import org.springframework.cloud.openfeign.FeignClient;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import java.io.File;
@FeignClient(name = ApiConstants.NAME) // TODO 芋艿：fallbackFactory =
@Tag(name = "RPC 服务 - 社交应用")
public interface SocialClientApi {
@ -47,4 +49,10 @@ public interface SocialClientApi {
CommonResult<SocialWxPhoneNumberInfoRespDTO> getWxMaPhoneNumberInfo(@RequestParam("userType") Integer userType,
@RequestParam("phoneCode") String phoneCode);
@GetMapping(PREFIX + "/getQRCode")
@Operation(summary = "获得微信小程序码")
@Parameter(name = "path", description = "地址", required = true, example = "/subPages/register/register")
@Parameter(name = "scene", description = "参数", required = true, example = "a=1")
CommonResult<File> getQRCode(@RequestParam("path") String path,
@RequestParam("scene") String scene);
}

13
yudao-module-system/yudao-module-system-biz/src/main/java/cn/iocoder/yudao/module/system/api/social/SocialClientApiImpl.java
View File

@ -1,17 +1,21 @@
package cn.iocoder.yudao.module.system.api.social;
import cn.binarywang.wx.miniapp.api.WxMaService;
import cn.binarywang.wx.miniapp.bean.WxMaPhoneNumberInfo;
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.system.api.social.dto.SocialWxJsapiSignatureRespDTO;
import cn.iocoder.yudao.module.system.api.social.dto.SocialWxPhoneNumberInfoRespDTO;
import cn.iocoder.yudao.module.system.service.social.SocialClientService;
import lombok.SneakyThrows;
import me.chanjar.weixin.common.bean.WxJsapiSignature;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.RestController;
import javax.annotation.Resource;
import java.io.File;
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
/**
@ -43,4 +47,13 @@ public class SocialClientApiImpl implements SocialClientApi {
return success(BeanUtils.toBean(info, SocialWxPhoneNumberInfoRespDTO.class));
}
@Override
@SneakyThrows
public CommonResult<File> getQRCode(String path, String scene) {
WxMaService wxService = socialClientService.getWxMaService();
File QRCode = wxService.getQrcodeService().createWxaCodeUnlimit(scene, path);
return success(QRCode);
}
}