图片添加二维码水印教程

写个图片添加图片水印的工具类，代码仅供参考
注意：二维码要清晰度要求比较高，就不要设置透明效果了

import java.awt.AlphaComposite;
import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.RenderingHints;
import java.awt.image.BufferedImage;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.text.SimpleDateFormat;
import java.util.Date;

import javax.imageio.ImageIO;
import javax.swing.ImageIcon;

import com.itextpdf.text.BadElementException;
import com.itextpdf.text.DocumentException;
import com.itextpdf.text.Rectangle;
import com.itextpdf.text.pdf.PdfContentByte;
import com.itextpdf.text.pdf.PdfGState;
import com.itextpdf.text.pdf.PdfReader;
import com.itextpdf.text.pdf.PdfStamper;
import com.itextpdf.text.pdf.PdfStructTreeController.returnType;
import com.itextpdf.text.pdf.parser.PdfImageObject.ImageBytesType;
import com.stuff.stuffmanage.model.CommonStuffModel;
/**
 * 
 * 
 *  水印处理工具类. 

 * 
 *
 * @author nicky.ma
 * @date 2019/06/11
 */
public class WaterMarkUtils {
	

	/**
	 *  生成二维码
	 * @author nicky.ma
	 * @date   2019年6月12日下午2:15:51
	 * @param commonStuffModel
	 * @return
	 */
	private static BufferedImage createQrCodeImg(CommonStuffModel commonStuffModel){
		StringBuffer strBuf = new StringBuffer();
		strBuf.append("材料入库时间：").append(new SimpleDateFormat("yyyy-MM-dd").format(new Date())).append("\n");
		strBuf.append("材料有效期：").append(commonStuffModel.getValidEndDateStr()).append("\n");
		strBuf.append("材料名称：").append(commonStuffModel.getStuffName()).append("\n");
		strBuf.append("材料目录：").append(commonStuffModel.getDirName()).append("\n");
		strBuf.append("材料版本：").append(commonStuffModel.getVersion()).append("\n");
		strBuf.append("出具单位：").append(commonStuffModel.getIssueUnit()).append("\n");
		// 生成二维码
		BufferedImage bufferedImage = QrCodeUtils.createQrCodeBufferdImage(strBuf.toString(), 175, 175);
		return bufferedImage;
	}
	
	/**
	 * 图片附件添加二维码水印
	 * @author  nicky.ma
	 * @date   2019年6月12日下午2:13:29
	 * @param bos
	 * 			文件保存 IO stream
	 * @param input
	 * 			获取文件上传IO stream
	 * @param commonStuffModel
	 * 			公共材料库信息实体
	 * @param imgType
	 * 			附件类型
	 */
	public static void setQrCodeForIMG(BufferedOutputStream bos, InputStream input, 
			ApprCommonStuffModel apprCommonStuffModel, String imgType){
		BufferedImage bufferedImage = createQrCodeImg(apprCommonStuffModel);
		//图片附件加上二维码水印
		setWaterMarkForIMG(bufferedImage, input, bos, imgType);
	}
	
	/**
     * 给图片添加水印
     * @param bufferedImage
     * 			水印图片
     * @param in
     * 			获取附件上传stream
     * @param os
     * 			附件保存stream
     * @param imgType
     * 			上传附件的类型
     */
    public static void setWaterMarkForIMG(BufferedImage bufferedImage, InputStream in, OutputStream os, String imgType) {
    	setWaterMarkForIMG(bufferedImage, in, os, null, imgType);
    }

    /**
     * 给图片添加水印、可设置水印图片旋转角度
     * @author  nicky.ma
     * @date   2019年6月12日下午2:11:11
     * @param bufferedImage
     * 			水印图片
     * @param in
     * 			获取附件上传stream
     * @param os
     * 			附件保存stream
     * @param degree
     * 			选择角度
     * @param imgType
     * 			上传附件的类型
     */
    public static void setWaterMarkForIMG(BufferedImage bufferedImage, InputStream in, OutputStream os, Integer degree,String imgType) {
        //OutputStream os = null;
        long startTime = System.currentTimeMillis();
        System.out.println("图片附件添加图片水印 start");
        try {
            Image srcImg = ImageIO.read(in);
            BufferedImage buffImg = new BufferedImage(srcImg.getWidth(null), srcImg.getHeight(null),
                    BufferedImage.TYPE_INT_RGB);
            // 得到画笔对象
            Graphics2D g = buffImg.createGraphics();
            // 设置对线段的锯齿状边缘处理
            g.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BILINEAR);
            g.drawImage(srcImg.getScaledInstance(srcImg.getWidth(null), srcImg.getHeight(null), Image.SCALE_SMOOTH), 0,
                    0, null);
            if (null != degree) {
                // 设置水印旋转
                g.rotate(Math.toRadians(degree), (double) buffImg.getWidth() / 2, (double) buffImg.getHeight() / 2);
            }
            //float alpha = 0.2f; // 透明度
            //g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, alpha));
            
            // 水印图象的路径 水印一般为gif或者png的，这样可设置透明度
            //ImageIcon imgIcon = new ImageIcon(iconPath);
            // 得到Image对象。
            //Image img = imgIcon.getImage();
            // 表示水印图片的位置
            g.drawImage(bufferedImage, buffImg.getWidth()-180, 0, null);

            g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER));
            g.dispose();

            //os = new FileOutputStream(targerPath);
            // 生成图片
            ImageIO.write(buffImg, imgType, os);

            long endTime = System.currentTimeMillis();
            System.out.println("图片附件添加图片水印 ok 所用时间:"+(endTime-startTime)+"s");
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if (null != os)
                    os.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
 }

对于上传的文件，我们怎么知道类型？如果用Spring提供的MultipartFile，这里可以获取ContentType来判断，这里只提供思路

	/**文件类型集合*/
	private static Map FILE_TYPES =new HashMap();
	/**图片类型集合*/
	private static Map IMG_TYPES = new HashMap();

	static{
		FILE_TYPES.put("xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");//xlsx
		FILE_TYPES.put("xls", "application/vnd.ms-excel");//xls
		FILE_TYPES.put("docx", "application/vnd.openxmlformats-officedocument.wordprocessingml.document");//docx
		FILE_TYPES.put("doc", "application/msword");//doc
		FILE_TYPES.put("jpg", "image/jpeg");//jpg
		FILE_TYPES.put("png", "image/png");//png
		FILE_TYPES.put("gif", "image/gif");//gif
		FILE_TYPES.put("bmp", "image/bmp");//bmp
		FILE_TYPES.put("txt", "text/plain");//txt
		FILE_TYPES.put("pdf", "application/pdf");//pdf
		FILE_TYPES.put("zip", "application/x-zip-compressed");//zip
		FILE_TYPES.put("rar", "application/octet-stream");//rar
	}
	static{
		IMG_TYPES.put("jpg", "image/jpeg");//jpg
		IMG_TYPES.put("png", "image/png");//png
		IMG_TYPES.put("gif", "image/gif");//gif
		IMG_TYPES.put("bmp", "image/bmp");//bmp
	}

	/**
	 * 校验上传附件是否为图片类型的
	 * @author nicky.ma
	 * @date   2019年6月12日上午11:15:30
	 * @param fileContentType 
	 * 			response格式ContentType
	 */
	public static boolean checkContainImgType(String fileContentType){
		if(!StringUtils.isEmpty(fileContentType)){
			return IMG_TYPES.containsValue(fileContentType);
		}
		return false;
	}

	public static String loadFileType(String fileContentType){
		for(Map.Entry entry : FILE_TYPES.entrySet()){
			if(entry.getValue().equals(fileContentType)){
				return entry.getKey();
			}
		}
		return null;
	}

有了工具类之后，我们需要获取文件上传的inputStream

public void upload(MultipartFile myfiles,String url,String rootPath,CommonStuffModel commonStuffModel)throws Exception{
	
        if(!myfiles.isEmpty()){   
            File localFile = new File(rootPath+url);  
            File parentFile = localFile.getParentFile();
            if(!parentFile.exists()){
            	parentFile.mkdirs();
			}
           
            String cOntentType= myfiles.getContentType();
           if(checkContainImgType(contentType)){//上传了图片类型附件
	            	InputStream inputStream = myfiles.getInputStream();
					BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(localFile));
	            	WaterMarkUtils.setQrCodeForIMG(bos, inputStream, 		commonStuffModel,CommonFileUtil.loadFileType(contentType));
	         }else{
	            	myfiles.transferTo(localFile);
	          }
		}
}

ok，生成二维码水印