最近公司要做一个用户邀请的功能,生成二维码邀请用户就能获得积分奖励。于是就开发了一个小功能,二维码可以保存在本地,也可以利用FastFileStorage上传到文件服务器。
pom文件用到的包
<!-- zxing生成二维码 -->
<dependency>
<groupId>com.google.zxing</groupId>
<artifactId>core</artifactId>
<version>${zxing-core.version}</version>
</dependency>
<dependency>
<groupId>com.google.zxing</groupId>
<artifactId>javase</artifactId>
<version>${zxing-javase.version}</version>
</dependency>
<!-- 上传文件服务器 -->
<dependency>
<groupId>com.github.tobato</groupId>
<artifactId>fastdfs-client</artifactId>
<version>1.26.5</version>
</dependency>
二维码生成代码 QRCodeUtil 如下
import com.github.tobato.fastdfs.domain.fdfs.StorePath;
import com.github.tobato.fastdfs.service.FastFileStorageClient;
import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.WriterException;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import javax.annotation.PostConstruct;
import javax.imageio.ImageIO;
import javax.swing.filechooser.FileSystemView;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
/**
* 二维码生成器
* @Author ztz
* @Date 2022年12月12日 18:04
* @Version 1.0
*/
@Component
public class QRCodeUtil {
private static final String FileFormat=".png";
/** 本地目录,这里是配置在yml或者xml文件里的地址,可以是服务器本身的目录(需要权限放行)或者oos服务器地址*/
@Value("${accessFile.location}")
private String location;
private static String staticLocation;
@Autowired
private FastFileStorageClient storageClient;
@Autowired
private static FastFileStorageClient staticStorageClient;
/**静态方法不能直接使用 @Value,所以采用 @PostConstruct方式引用*/
@PostConstruct
public void init() {
staticStorageClient = storageClient;
staticLocation =location;
}
public static String httpUrl = "你的文件服务器地址";
//CODE_WIDTH:二维码宽度,单位像素
private static final int CODE_WIDTH = 100;
//CODE_HEIGHT:二维码高度,单位像素
private static final int CODE_HEIGHT = 100;
//FRONT_COLOR:二维码前景色,0x000000 表示黑色
private static final int FRONT_COLOR = 0x000000;
//BACKGROUND_COLOR:二维码背景色,0xFFFFFF 表示白色
//演示用 16 进制表示,和前端页面 CSS 的取色是一样的,注意前后景颜色应该对比明显,如常见的黑白
private static final int BACKGROUND_COLOR = 0xFFFFFF;
/**
* 生成二维码到本地
* @param content 二维码内容。一般是一个链接
@param folderName 生成的文件目录
* @return
*/
public static String createCodeToFile(String content,String folderName) {
try {
File codeImgFileSaveDir = new File(staticLocation+folderName);
//1. 使用UUID重新生成文件名,防止文件名称重复造成文件覆盖
String fileName = UUID.randomUUID() + "." +FileFormat;
content = content.trim();
if (codeImgFileSaveDir==null || codeImgFileSaveDir.isFile()) {
//二维码图片存在目录为空,默认放在桌面...
codeImgFileSaveDir = FileSystemView.getFileSystemView().getHomeDirectory();
}
if (!codeImgFileSaveDir.exists()) {
//二维码图片存在目录不存在,开始创建...
codeImgFileSaveDir.mkdirs();
}
//核心代码-生成二维码
BufferedImage bufferedImage = getBufferedImage(content);
File codeImgFile = new File(codeImgFileSaveDir, fileName);
ImageIO.write(bufferedImage, "png", codeImgFile);
return folderName+fileName;
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
/**
* 生成二维码的流文件并保存到文件服务器
* @param content :二维码内容
*/
public static String createCodeToOutputStream(String content) {
String url = null;
try {
if (StringUtils.isBlank(content)) {
return url;
}
content = content.trim();
//核心代码-生成二维码
BufferedImage bufferedImage = getBufferedImage(content);
//以流的方式将图片上传到fastdfs上:
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
ImageIO.write(bufferedImage, "png", outputStream);
InputStream inputStream = new ByteArrayInputStream(outputStream.toByteArray());
StorePath storePath = staticStorageClient.uploadFile(inputStream,inputStream.available(),"png",null);
if(storePath!=null) {
url = httpUrl+storePath.getFullPath();
}
} catch (Exception e) {
e.printStackTrace();
}
return url;
}
//核心代码-生成二维码
private static BufferedImage getBufferedImage(String content) throws WriterException {
//com.google.zxing.EncodeHintType:编码提示类型,枚举类型
Map<EncodeHintType, Object> hints = new HashMap();
//EncodeHintType.CHARACTER_SET:设置字符编码类型
hints.put(EncodeHintType.CHARACTER_SET, "UTF-8");
//EncodeHintType.ERROR_CORRECTION:设置误差校正
//ErrorCorrectionLevel:误差校正等级,L = ~7% correction、M = ~15% correction、Q = ~25% correction、H = ~30% correction
//不设置时,默认为 L 等级,等级不一样,生成的图案不同,但扫描的结果是一样的
hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.M);
//EncodeHintType.MARGIN:设置二维码边距,单位像素,值越小,二维码距离四周越近
hints.put(EncodeHintType.MARGIN, 1);
MultiFormatWriter multiFormatWriter = new MultiFormatWriter();
BitMatrix bitMatrix = multiFormatWriter.encode(content, BarcodeFormat.QR_CODE, CODE_WIDTH, CODE_HEIGHT, hints);
BufferedImage bufferedImage = new BufferedImage(CODE_WIDTH, CODE_HEIGHT, BufferedImage.TYPE_INT_BGR);
for (int x = 0; x < CODE_WIDTH; x++) {
for (int y = 0; y < CODE_HEIGHT; y++) {
bufferedImage.setRGB(x, y, bitMatrix.get(x, y) ? FRONT_COLOR : BACKGROUND_COLOR);
}
}
return bufferedImage;
}
利用redis的incr生成唯一邀请码序列,也可以配合时间和uuid等参数生成唯一订单
。
/**
* 递增
*
* @param key 键
* @param delta 要增加几(大于0)
* @return
*/
public long incr(String key, long delta) {
if (delta < 0) {
throw new RuntimeException("递增因子必须大于0");
}
return redisTemplate.opsForValue().increment(key, delta);
}
文件存放在本地的话记得别忘了配置静态资源访问路径,记得shiro里也要放行
@Configuration
public class WebMvcConfig implements WebMvcConfigurer {
//匹配url 中的资源映射
@Value("${accessFile.resourceHandler}")
private String resourceHandler;
//上传文件保存的本地目录
@Value("${accessFile.location}")
private String location;
/**
* 配置静态资源映射
*
* @param registry
*/
@Override
public void addResourceHandlers(ResourceHandlerRegistry registry) {
//匹配到resourceHandler,将URL映射至location,也就是本地文件夹
registry.addResourceHandler(resourceHandler).addResourceLocations("file:" + location);
}
/**
* 跨域
* @param registry
*/
@Override
public void addCorsMappings(CorsRegistry registry) {
registry.addMapping("/**")
.allowedOriginPatterns("*")
.allowCredentials(true)
.allowedMethods("GET", "POST", "PUT", "DELETE", "OPTIONS")
.maxAge(3600);
}
}
/**
* 递增
*
* @param key 键
* @param delta 要增加几(大于0)
* @return
*/
public long incr(String key, long delta) {
if (delta < 0) {
throw new RuntimeException("递增因子必须大于0");
}
return redisTemplate.opsForValue().increment(key, delta);
}