bjang03
7 months ago
7 changed files with 218 additions and 1 deletions
@ -0,0 +1,59 @@ |
|||
package com.win.module.eam.controller.licences; |
|||
|
|||
import cn.hutool.extra.qrcode.QrCodeUtil; |
|||
import com.win.framework.common.pojo.CommonResult; |
|||
import com.win.module.eam.controller.licences.vo.GenerateLicenceReqVO; |
|||
import com.win.module.eam.service.licences.LicencesService; |
|||
import io.swagger.v3.oas.annotations.Operation; |
|||
import io.swagger.v3.oas.annotations.Parameter; |
|||
import io.swagger.v3.oas.annotations.Parameters; |
|||
import io.swagger.v3.oas.annotations.tags.Tag; |
|||
import org.springframework.security.access.prepost.PreAuthorize; |
|||
import org.springframework.validation.annotation.Validated; |
|||
import org.springframework.web.bind.annotation.*; |
|||
import org.springframework.web.multipart.MultipartFile; |
|||
|
|||
import javax.annotation.Resource; |
|||
import javax.servlet.http.HttpServletResponse; |
|||
import javax.validation.Valid; |
|||
import javax.validation.constraints.NotNull; |
|||
import java.io.IOException; |
|||
|
|||
import static com.win.framework.common.pojo.CommonResult.success; |
|||
|
|||
@Tag(name = "接口-续期许可") |
|||
@RestController |
|||
@RequestMapping("/eam/basic/licences") |
|||
@Validated |
|||
public class LicencesController { |
|||
@Resource |
|||
private LicencesService licencesService; |
|||
|
|||
@PostMapping("/licenceDiscernByCodeFile") |
|||
@Operation(summary = "识别续期许可(二维码图片)") |
|||
@Parameters({ |
|||
@Parameter(name = "file", description = "续期二维码图片", required = true) |
|||
}) |
|||
@PreAuthorize("@ss.hasPermission('basic:licences:licenceDiscern')") |
|||
public CommonResult<Boolean> licenceDiscernByCodeFile(@Valid @NotNull @RequestParam("file") MultipartFile file) throws IOException { |
|||
licencesService.licencesDiscern(QrCodeUtil.decode(file.getInputStream())); |
|||
return success(true); |
|||
} |
|||
|
|||
@PostMapping("/licenceDiscernByCodeStr") |
|||
@Operation(summary = "识别续期许可(二维码字符串)") |
|||
@Parameters({ |
|||
@Parameter(name = "code", description = "续期二维码识别出来的字符串", required = true) |
|||
}) |
|||
@PreAuthorize("@ss.hasPermission('basic:licences:licenceDiscern')") |
|||
public CommonResult<Boolean> licenceDiscernByCodeStr(@RequestBody String code) { |
|||
licencesService.licencesDiscern(code); |
|||
return success(true); |
|||
} |
|||
@PostMapping("/generateLicence") |
|||
@Operation(summary = "生成续期二维码") |
|||
@PreAuthorize("@ss.hasPermission('basic:licences:licenceDiscern')") |
|||
public void generateLicence(@Valid @RequestBody GenerateLicenceReqVO req,HttpServletResponse response) throws Exception { |
|||
licencesService.generateLicence(req,response); |
|||
} |
|||
} |
|||
@ -0,0 +1,20 @@ |
|||
package com.win.module.eam.controller.licences.vo; |
|||
|
|||
import io.swagger.v3.oas.annotations.media.Schema; |
|||
import lombok.Data; |
|||
import lombok.ToString; |
|||
|
|||
import javax.validation.constraints.NotNull; |
|||
|
|||
@Schema(description = "管理后台 - 生成续期码 Request VO") |
|||
@Data |
|||
@ToString(callSuper = true) |
|||
public class GenerateLicenceReqVO { |
|||
@NotNull(message = "公司编码不能为空") |
|||
@Schema(description = "公司编码") |
|||
private String companyCode; |
|||
|
|||
@NotNull(message = "到期时间不能为空") |
|||
@Schema(description = "到期时间") |
|||
private String endTime; |
|||
} |
|||
@ -0,0 +1,13 @@ |
|||
package com.win.module.eam.mq.message; |
|||
|
|||
import lombok.Data; |
|||
import lombok.ToString; |
|||
|
|||
import java.util.List; |
|||
|
|||
@Data |
|||
@ToString(callSuper = true) |
|||
public class LicencesMessage { |
|||
private String secretKey; |
|||
private String data; |
|||
} |
|||
@ -0,0 +1,11 @@ |
|||
package com.win.module.eam.service.licences; |
|||
|
|||
import com.win.module.eam.controller.licences.vo.GenerateLicenceReqVO; |
|||
|
|||
import javax.servlet.http.HttpServletResponse; |
|||
import javax.validation.Valid; |
|||
|
|||
public interface LicencesService { |
|||
void licencesDiscern(String encryptionStr); |
|||
void generateLicence(@Valid GenerateLicenceReqVO req, HttpServletResponse response) throws Exception; |
|||
} |
|||
@ -0,0 +1,108 @@ |
|||
package com.win.module.eam.service.licences; |
|||
|
|||
import cn.hutool.core.codec.Base64; |
|||
import cn.hutool.crypto.digest.MD5; |
|||
import cn.hutool.extra.qrcode.QrCodeUtil; |
|||
import cn.hutool.extra.qrcode.QrConfig; |
|||
import com.alibaba.excel.metadata.data.ImageData; |
|||
import com.alibaba.fastjson.JSONArray; |
|||
import com.alibaba.fastjson.JSONObject; |
|||
import com.win.module.eam.controller.licences.vo.GenerateLicenceReqVO; |
|||
import com.win.module.eam.mq.message.LicencesMessage; |
|||
import org.apache.calcite.util.Util; |
|||
import org.springframework.data.redis.core.RedisTemplate; |
|||
import org.springframework.stereotype.Service; |
|||
|
|||
import javax.annotation.Resource; |
|||
import javax.crypto.Cipher; |
|||
import javax.crypto.KeyGenerator; |
|||
import javax.crypto.SecretKey; |
|||
import javax.crypto.spec.SecretKeySpec; |
|||
import javax.servlet.http.HttpServletResponse; |
|||
import java.nio.charset.StandardCharsets; |
|||
import java.util.ArrayList; |
|||
import java.util.List; |
|||
|
|||
import static com.win.framework.common.exception.util.ServiceExceptionUtil.exception; |
|||
import static com.win.module.eam.enums.ErrorCodeConstants.ENCRYPTION_STR_FORMAT_IS_ERROR; |
|||
|
|||
@Service |
|||
public class LicencesServiceImpl implements LicencesService { |
|||
private static final String ALGORITHM = "AES"; |
|||
private static final int KEY_SIZE = 128; |
|||
private static final String LICENCES_UPDATE = "LICENCES"; |
|||
@Resource |
|||
private RedisTemplate<String, String> redisTemplate; |
|||
|
|||
private static String generateKey() throws Exception { |
|||
KeyGenerator keyGenerator = KeyGenerator.getInstance(ALGORITHM); |
|||
keyGenerator.init(KEY_SIZE); |
|||
SecretKey secretKey = keyGenerator.generateKey(); |
|||
return Base64.encode(secretKey.getEncoded()); |
|||
} |
|||
|
|||
private static String encrypt(String data, String key) throws Exception { |
|||
SecretKeySpec secretKeySpec = new SecretKeySpec(Base64.decode(key), ALGORITHM); |
|||
Cipher cipher = Cipher.getInstance(ALGORITHM); |
|||
cipher.init(Cipher.ENCRYPT_MODE, secretKeySpec); |
|||
byte[] encryptedBytes = cipher.doFinal(data.getBytes(StandardCharsets.UTF_8)); |
|||
return Base64.encode(encryptedBytes); |
|||
} |
|||
|
|||
private static String decrypt(String encryptedData, String key) throws Exception { |
|||
SecretKeySpec secretKeySpec = new SecretKeySpec(Base64.decode(key), ALGORITHM); |
|||
Cipher cipher = Cipher.getInstance(ALGORITHM); |
|||
cipher.init(Cipher.DECRYPT_MODE, secretKeySpec); |
|||
byte[] decryptedBytes = cipher.doFinal(Base64.decode(encryptedData)); |
|||
return new String(decryptedBytes, StandardCharsets.UTF_8); |
|||
} |
|||
|
|||
@Override |
|||
public void licencesDiscern(String encryptionStr) { |
|||
if (Util.isNullOrEmpty(encryptionStr)) { |
|||
throw exception(ENCRYPTION_STR_FORMAT_IS_ERROR); |
|||
} |
|||
try { |
|||
//todo 数据校验
|
|||
JSONObject codeJson = JSONObject.parseObject(encryptionStr); |
|||
String secretKey = codeJson.getString("secretKey"); |
|||
if (Util.isNullOrEmpty(secretKey)) { |
|||
throw exception(ENCRYPTION_STR_FORMAT_IS_ERROR); |
|||
} |
|||
String dataStr = codeJson.getString("data"); |
|||
if (Util.isNullOrEmpty(dataStr)) { |
|||
throw exception(ENCRYPTION_STR_FORMAT_IS_ERROR); |
|||
} |
|||
JSONArray dataJson = JSONArray.parseArray(decrypt(dataStr, secretKey)); |
|||
if (dataJson.isEmpty()) { |
|||
throw exception(ENCRYPTION_STR_FORMAT_IS_ERROR); |
|||
} |
|||
//todo 更新到redis并使用发布订阅通知其他pods
|
|||
redisTemplate.opsForValue().set(LICENCES_UPDATE, dataJson.toJSONString()); |
|||
redisTemplate.convertAndSend(LICENCES_UPDATE, 1); |
|||
} catch (Exception e) { |
|||
throw exception(500, e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public void generateLicence(GenerateLicenceReqVO req, HttpServletResponse response) throws Exception { |
|||
LicencesMessage licencesMessage = new LicencesMessage() {{ |
|||
String secretKey = generateKey(); |
|||
setSecretKey(secretKey); |
|||
List data = new ArrayList(); |
|||
MD5 md5 = MD5.create(); |
|||
data.add(new JSONObject(){{ |
|||
put(md5.digestHex("/MES"),encrypt("2025-10-10 00:00:00",secretKey)); |
|||
}}.toJSONString()); |
|||
data.add(new JSONObject(){{ |
|||
put(md5.digestHex("/WMS"),encrypt("2025-10-10 00:00:00",secretKey)); |
|||
}}.toJSONString()); |
|||
setData(encrypt(JSONObject.toJSONString(data),secretKey)); |
|||
}}; |
|||
QrConfig config = new QrConfig(300, 300); |
|||
config.setErrorCorrection(com.google.zxing.qrcode.decoder.ErrorCorrectionLevel.M); // 设置纠错级别
|
|||
// 或者直接输出到流
|
|||
QrCodeUtil.generate(JSONObject.toJSONString(licencesMessage), config, ImageData.ImageType.PICTURE_TYPE_DIB.name(), response.getOutputStream()); |
|||
} |
|||
} |
|||
Loading…
Reference in new issue