解決在ueditor中上傳圖片經過後臺java+SpringMVC添加水印方法

對於我一個菜鳥來講，之前一直對io這塊不熟悉，如今業務需求要求對富文本中的圖片添加水印，在百度上查了都不是適合個人項目，只能本身研究，研究了倆天終於寫了出來，如今我把個人方法寫出來，供你們查閱

1.由於是在SpringMVC裏面寫的，SpringMVC提供了一個MultipartFile類，直接上代碼

/**
 * ueditor上傳單文件
 * @param request
 * @return
 */
@RequestMapping(value = "ueditorUpFile")
public 
@ResponseBody
UeditorFormat ueditorUpFile(HttpServletRequest request){
   MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
   // 從config.json中取得上傳文件的ID
   MultipartFile upfile = multipartRequest.getFile("upfile");
   try {
      InputStream inputStream = upfile.getInputStream();
   } catch (IOException e) {
      e.printStackTrace();
   }
   //獲取項目根路徑
   String realpath= request.getSession().getServletContext().getRealPath("/");
   if(Objects.nonNull(upfile)){
      //這個是個人項目類，沒有關係
      SysFile file = new SysFile();
      file.setCreateDate(new Date());
      file.setName(upfile.getOriginalFilename());
      file.setRandomName(IdGen.uuid());
      file.setStatus(SysFile.TEMP_FILE);
      file.setSuffix(file.getName().substring(file.getName().lastIndexOf(".")));
      file.setSize(upfile.getSize());
      try {
         //上傳到服務器的路徑，沒有關係
         String filepath = sysFileService.genFilePath(2, file.getRandomName(), file.getSuffix());
         //獲取上傳的圖片
         File tarfile = new File(realpath+upfile.getOriginalFilename());
         try {
            //把內存圖片寫入磁盤中
            upfile.transferTo(tarfile);
            if (!tarfile.getParentFile().exists()) {
               tarfile.getParentFile().mkdir();
            }
            try {
               //添加水印
               WaterMarkGenerate.generateWithImageMark(tarfile,realpath+File.separator+"upload"+File.separator+file.getRandomName()+file.getSuffix(),realpath+File.separator+"static"+File.separator+"images"+File.separator+"watermark.png");
            } catch (Exception e) {
               e.printStackTrace();
            }
            //獲取添加水印後的圖片
            File newFile = new File(realpath+File.separator+"upload"+File.separator+file.getRandomName()+file.getSuffix());
            FileInputStream input = new FileInputStream(newFile);
            MultipartFile multipartFile = new MockMultipartFile("file",
                  file.getName(), "text/plain", IOUtils.toByteArray(input));
            upfile = multipartFile;
         } catch (IOException e) {
            e.printStackTrace();
         }
         //水印圖片保存到服務器
         file = sysFileService.saveFile(file, upfile, filepath);
         return UeditorFormat.parseSysFile(file);
      } catch (FileUploadFailException e) {
         e.printStackTrace();
         UeditorFormat uf = new UeditorFormat();
         uf.setState("文件上傳失敗");
         uf.setTitle(upfile.getOriginalFilename());
         return uf;
      }
   } else {
      UeditorFormat uf = new UeditorFormat();
      uf.setState("文件上傳失敗，上傳的文件爲空！");
      uf.setTitle(upfile.getOriginalFilename());
      return uf;
   }
}

這個添加水印的類也是我從網上找的，特別好用，親測，如今分享給你們

public class WaterMarkGenerate {
    private static final String FONT_FAMILY="微軟雅黑";//字體
    private static final int FONT_STYLE=Font.BOLD;//字體加粗
    private static final int FONT_SIZE=24;//字體大小
    private static final float ALPHA=0.7F;//水印透明度

    private static final int LOGO_WIDTH=200;//圖片水印大小

    //添加文字水印
    /*tarPath:圖片保存路徑
     *contents:文字水印內容* */
    public static void generateWithTextMark(File srcFile,
                                            String tarPath,String contents) throws Exception{
        Image srcImage=ImageIO.read(srcFile);
        int width=srcImage.getWidth(null);
        int height=srcImage.getHeight(null);

        BufferedImage tarBuffImage=new BufferedImage(width,height,BufferedImage.TYPE_INT_RGB);
        Graphics2D g=tarBuffImage.createGraphics();
        g.drawImage(srcImage, 0, 0, width,height,null);

        //計算
        int strWidth=FONT_SIZE*getTextLength(contents);
        int strHeight=FONT_SIZE;

        //水印位置
//      int x=width-strWidth;
//      int y=height-strHeight;

        int x=0,y=0;

        //設置字體和水印透明度
        g.setFont(new Font(FONT_FAMILY,FONT_STYLE,FONT_SIZE));
        g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP,ALPHA));
//      g.drawString(contents, x, y);
        //旋轉圖片
        g.rotate(Math.toRadians(-30),width/2,height/2);
        while(x < width*1.5){
            y = -height/2;
            while(y < height*1.5){
                g.drawString(contents,x,y);
                y += strHeight + 50;
            }
            x += strWidth + 50; //水印之間的間隔設爲50
        }
        g.dispose();

        JPEGImageEncoder en = JPEGCodec.createJPEGEncoder(new FileOutputStream(tarPath));
        en.encode(tarBuffImage);
    }

    //添加圖片水印
    /*
     * tarPath:圖片保存路徑
     * logoPath:logo文件路徑
     * */
    public static void generateWithImageMark(File srcFile,
                                             String tarPath,String logoPath) throws Exception{
        Image srcImage=ImageIO.read(srcFile);
        int width=srcImage.getWidth(null);
        int height=srcImage.getHeight(null);
        //建立一個不帶透明色的BufferedImage對象
        BufferedImage tarBuffImage=new BufferedImage(width,height,BufferedImage.TYPE_INT_RGB);
        Graphics2D g=tarBuffImage.createGraphics();
        g.drawImage(srcImage, 0, 0, width,height,null);

        Image logoImage= ImageIO.read(new File(logoPath));
        int logoWidth=LOGO_WIDTH;
        int logoHeight=(LOGO_WIDTH*logoImage.getHeight(null))/logoImage.getWidth(null);

        int x=width-logoWidth;
        int y=height-logoHeight;

        g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP,ALPHA));
        g.drawImage(logoImage, x, y, logoWidth, logoHeight, null);
        g.dispose();

        JPEGImageEncoder en = JPEGCodec.createJPEGEncoder(new FileOutputStream(tarPath));
        en.encode(tarBuffImage);
    }

    //文本長度的處理：文字水印的中英文字符的寬度轉換
    public static int getTextLength(String text){
        int length = text.length();
        for(int i=0;i<text.length();i++){
            String s = String.valueOf(text.charAt(i));
            if(s.getBytes().length>1){  //中文字符
                length++;
            }
        }
        length = length%2 == 0?length/2:length/2+1;  //中文和英文字符的轉換
        return length;
    }
}