package com.qxueyou.scc.base.util;
|
|
import java.awt.BasicStroke;
|
import java.awt.Color;
|
import java.awt.Graphics2D;
|
import java.awt.image.BufferedImage;
|
import java.io.IOException;
|
import java.io.OutputStream;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
import javax.imageio.ImageIO;
|
|
import org.apache.commons.lang3.StringUtils;
|
import org.apache.http.HttpEntity;
|
import org.apache.http.HttpResponse;
|
import org.apache.http.client.methods.HttpGet;
|
import org.apache.http.impl.client.CloseableHttpClient;
|
import org.apache.http.impl.client.HttpClients;
|
import org.apache.logging.log4j.LogManager;
|
import org.apache.logging.log4j.Logger;
|
|
//import com.qxueyou.scc.base.service.IConfigService;
|
import com.qxueyou.scc.shorturl.util.ShortUrlUtils;
|
import com.google.zxing.BarcodeFormat;
|
import com.google.zxing.EncodeHintType;
|
import com.google.zxing.MultiFormatWriter;
|
import com.google.zxing.WriterException;
|
import com.google.zxing.client.j2se.MatrixToImageConfig;
|
import com.google.zxing.client.j2se.MatrixToImageWriter;
|
import com.google.zxing.common.BitMatrix;
|
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
|
|
/**
|
* 二维码生成工具类
|
*
|
* @author 德虎
|
*
|
*/
|
public final class QrCodeUtils {
|
|
private static Logger log = LogManager.getLogger(QrCodeUtils.class);
|
|
/**
|
* 生成二维码图片并输出到流
|
* @param msg 需要生成的二维码信息
|
* @param os 二维码图片输出流，输出后会自动关闭
|
* @param logoFlag 是否添加logo
|
*
|
*/
|
public static void createQRCodeImgAndSend(String msg, OutputStream os, boolean logoFlag) {
|
try {
|
//如果是网址，则转换为短网址
|
/*if(ShortUrlUtils.IsUrl(msg)){
|
IShortUrlService shortUrlService = SpringUtil.getBean(IShortUrlService.class);
|
ShortUrl oldShortUrl = shortUrlService.queryShortUrl(msg);
|
if(oldShortUrl != null){//判断是否已经存在数据
|
msg = oldShortUrl.getShortUrl();
|
}else{
|
//获取域
|
String domain = SpringUtil.getBean(IConfigService.class).getConfigByEnv("domain-name");
|
//获取短链接编码
|
String strShortUrlCode = ShortUrlUtils.shortUrl(msg);
|
//拼接成短链接
|
String shortUrl = domain+(domain.indexOf("/qxueyou", 10)>=0?"/s/":"/qxueyou/s/")+strShortUrlCode;
|
ShortUrl objShortUrl = new ShortUrl();
|
//保存短链接的值
|
objShortUrl.setShortUrl(shortUrl);
|
objShortUrl.setLongUrl(msg);
|
objShortUrl.setShortUrlCode(strShortUrlCode);
|
objShortUrl.setDeleteFlag(false);
|
TraceUtils.setCreateTrace(objShortUrl);
|
msg = shortUrl;
|
shortUrlService.saveShortUrl(objShortUrl);
|
}
|
}*/
|
MultiFormatWriter multiFormatWriter = new MultiFormatWriter();
|
|
Map<EncodeHintType, Object> hints = new HashMap<EncodeHintType, Object>(
|
2);
|
hints.put(EncodeHintType.CHARACTER_SET, "utf-8");
|
hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.M);
|
|
BitMatrix bitMatrix = multiFormatWriter.encode(
|
msg,
|
BarcodeFormat.QR_CODE, 300, 300, hints);
|
MatrixToImageConfig config = new MatrixToImageConfig(0xFF000000, -1);
|
//添加机构图标
|
if(StringUtils.isNoneBlank(ClientUtils.getOrgLogoPath()) && ShortUrlUtils.IsUrl(msg) && logoFlag){
|
int width = bitMatrix.getWidth();
|
int height = bitMatrix.getHeight();
|
BufferedImage image = new BufferedImage(width, height,
|
BufferedImage.TYPE_INT_RGB);
|
for (int x = 0; x < width; x++) {
|
for (int y = 0; y < height; y++) {
|
image.setRGB(x, y, bitMatrix.get(x, y) ? 0xFF000000
|
: 0xFFFFFFFF);
|
}
|
}
|
addLogo_QRCode(image, os);
|
}else{
|
MatrixToImageWriter.writeToStream(bitMatrix, "PNG",
|
os, config);
|
}
|
|
} catch (WriterException e) {
|
log.error(e, e);
|
} catch (IOException e) {
|
log.error(e, e);
|
} catch (Exception e) {
|
log.error(e, e);
|
}
|
}
|
|
/**
|
* 添加机构logo
|
*
|
* @param buf 生成的二维码
|
* @return
|
* @throws IOException
|
*/
|
public static void addLogo_QRCode(BufferedImage buf, OutputStream os) throws IOException{
|
CloseableHttpClient httpClient = null;
|
HttpGet httpget = null;
|
try{
|
// String domain = SpringUtil.getBean(IConfigService.class).getConfigByEnv("oss-domain");
|
String domain =null;
|
httpClient = HttpClients.createDefault();
|
//获取机构logo
|
httpget = new HttpGet(domain+ClientUtils.getOrgLogoPath());
|
HttpResponse response = httpClient.execute(httpget);
|
HttpEntity entity = response.getEntity();
|
/**
|
* 读取Logo图片
|
*/
|
BufferedImage logo = ImageIO.read(entity.getContent());
|
/**
|
* 读取二维码图片，并构建绘图对象
|
*/
|
Graphics2D g = buf.createGraphics();
|
int widthLogo = 65, heightLogo = 65;
|
|
// 计算图片放置位置
|
int x = (buf.getWidth() - widthLogo) / 2;
|
int y = (buf.getHeight() - heightLogo) / 2;
|
|
//开始绘制图片
|
g.setColor(Color.WHITE);
|
g.drawImage(logo, x, y, widthLogo, heightLogo, null);
|
g.setStroke(new BasicStroke(1));
|
g.dispose();
|
ImageIO.write(buf, "PNG", os);
|
}catch (Exception e) {
|
ImageIO.write(buf, "PNG", os);
|
log.error("二维码添加logo失败"+e);
|
}finally {
|
httpClient.close();
|
}
|
}
|
}
|
