生成指定下载地址的二维码

公司新开发的app， 需要进行推广， 所以开发了一个邀请功能的二维码 并邀请统计是哪个业务员推广的，
在这里记录一下
先来看代码

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.Map;
import javax.imageio.ImageIO;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import com.cloopen.rest.sdk.utils.encoder.BASE64Encoder;
import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
import cn.com.bonc.redis.RedisUtil;


@RequestMapping({"QrCode"})
@RestController
public class QrCodeController {	
	@Value("${rootpath}")
	String rootpath;

 
private static final int QRCOLOR = 0xFF000000; // 默认是黑色
private static final int BGWHITE = 0xFFFFFFFF; // 背景颜色
private static final int WIDTH = 400; // 二维码宽
private static final int HEIGHT = 400; // 二维码高

@Autowired
RedisUtil redis;

// 用于设置QR二维码参数
 private static Map<EncodeHintType, Object> hints = new HashMap<EncodeHintType, Object>() {
     private static final long serialVersionUID = 1L;
     {
         put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H);// 设置QR二维码的纠错级别（H为最高级别）具体级别信息
         put(EncodeHintType.CHARACTER_SET, "utf-8");// 设置编码方式
         put(EncodeHintType.MARGIN, 0);
     }
 };
 
 /**
  * 邀请码
  * @param request
  * @param response
  */
 @RequestMapping({"getQrCodeBuyer"})
 public  void getQrCodeBuyer(HttpServletRequest request,HttpServletResponse response) {
	 /*String token = request.getParameter("token");
     Map<String, Object> user =SecurityUtils.getSubject().getSession()!=null? (Map)SecurityUtils.getSubject().getSession().getAttribute("user"):null;   
     String loginName = "";
     if(user!=null){
    	 loginName =  user.get("userName").toString();
     }else if(token!=null&&!token.isEmpty()){
    	 loginName = redis.getString(token) ;
     }*/
	 String loginName =request.getParameter("referrer");
	 //logo路径
	 File logoFile = new File(rootpath+"/ROOT/WEB-INF/classes/static/images/feibao.jpg");
     String url = "https://www.zhongbaochuang.com/pc/download/buyer.html?referrer="+loginName;
     String note = "";
     OutputStream toClient=null;
	try {
		toClient = response.getOutputStream();
	} catch (IOException e) {
		// TODO Auto-generated catch block
		e.printStackTrace();
	}
     drawLogoQRCode(logoFile, toClient, url, note);        
    
 }


 // 生成带logo的二维码图片
 public void drawLogoQRCode(File logoFile, OutputStream out, String qrUrl, String note) {
	
     try {
         MultiFormatWriter multiFormatWriter = new MultiFormatWriter();
         // 参数顺序分别为：编码内容，编码类型，生成图片宽度，生成图片高度，设置参数
         BitMatrix bm = multiFormatWriter.encode(qrUrl, BarcodeFormat.QR_CODE, WIDTH, HEIGHT, hints);
         BufferedImage image = new BufferedImage(WIDTH, HEIGHT, BufferedImage.TYPE_INT_RGB);

         // 开始利用二维码数据创建Bitmap图片，分别设为黑（0xFFFFFFFF）白（0xFF000000）两色
         for (int x = 0; x < WIDTH; x++) {
             for (int y = 0; y < HEIGHT; y++) {
                 image.setRGB(x, y, bm.get(x, y) ? QRCOLOR : BGWHITE);
            }
         }

         int width = image.getWidth();
         int height = image.getHeight();
         if (logoFile.exists()) {
             // 构建绘图对象
             Graphics2D g = image.createGraphics();
             // 读取Logo图片
             BufferedImage logo = ImageIO.read(logoFile);
             // 开始绘制logo图片
             g.drawImage(logo, width * 2 / 5, height * 2 / 5, width * 2 / 10, height * 2 / 10, null);
             g.dispose();
             logo.flush();
         }

         // 自定义文本描述
         if (StringUtils.isNotEmpty(note)) {
             // 新的图片，把带logo的二维码下面加上文字
             BufferedImage outImage = new BufferedImage(400, 445, BufferedImage.TYPE_4BYTE_ABGR);
             Graphics2D outg = outImage.createGraphics();
             // 画二维码到新的面板
             outg.drawImage(image, 0, 0, image.getWidth(), image.getHeight(), null);
             // 画文字到新的面板
             outg.setColor(Color.BLACK);
             outg.setFont(new Font("楷体", Font.BOLD, 30)); // 字体、字型、字号
             int strWidth = outg.getFontMetrics().stringWidth(note);
             if (strWidth > 399) {
                 //长度过长就截取前面部分
                 // 长度过长就换行
                 String note1 = note.substring(0, note.length() / 2);
                 String note2 = note.substring(note.length() / 2, note.length());
                 int strWidth1 = outg.getFontMetrics().stringWidth(note1);
                 int strWidth2 = outg.getFontMetrics().stringWidth(note2);
                outg.drawString(note1, 200 - strWidth1 / 2, height + (outImage.getHeight() - height) / 2 + 12);
                 BufferedImage outImage2 = new BufferedImage(400, 485, BufferedImage.TYPE_4BYTE_ABGR);
                 Graphics2D outg2 = outImage2.createGraphics();
                 outg2.drawImage(outImage, 0, 0, outImage.getWidth(), outImage.getHeight(), null);
                 outg2.setColor(Color.BLACK);
                 outg2.setFont(new Font("宋体", Font.BOLD, 30)); // 字体、字型、字号
                 outg2.drawString(note2, 200 - strWidth2 / 2,outImage.getHeight() + (outImage2.getHeight() - outImage.getHeight()) / 2 + 5);
                 outg2.dispose();
                 outImage2.flush();
                 outImage = outImage2;
             } else {
                 outg.drawString(note, 200 - strWidth / 2, height + (outImage.getHeight() - height) / 2 + 12); // 画文字
             }
             outg.dispose();
             outImage.flush();
             image = outImage;
         }

         image.flush();

        // ImageIO.write(image, "png", codeFile); // TODO
        ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
 		ImageIO.write(image, "png", byteArrayOutputStream);
        BASE64Encoder base64Encoder = new BASE64Encoder();		
        byte[] byteArray = byteArrayOutputStream.toByteArray();
        //String b= "data:image/png;base64," +  base64Encoder.encodeBuffer(byteArray).trim();
        out.write(byteArray);
        out.flush();
        out.close();
         //System.out.println();

     } catch (Exception e) {
         e.printStackTrace();
     }
	
 }

}