java 根据网络图片url获取InputStream

package com.awj.mall.restfulapi.modular.user.util;

import java.awt.AlphaComposite;
import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.RenderingHints;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Map;

import javax.imageio.ImageIO;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

import com.awj.mall.core.base.service.IUploadService;

@RunWith(SpringRunner.class)
@SpringBootTest
public class ImageWatermarkUtil {

	@Autowired
	private IUploadService iUploadService;

	// 水印透明度
	private static float alpha = 0.2f;
	// 水印文字大小
	public static final int FONT_SIZE = 28;
	// 水印文字字体
	private static Font font = new Font("微软雅黑", Font.BOLD, FONT_SIZE);
	// 水印文字颜色
	private static Color color = Color.white;
	// 水印之间的间隔
	private static final int XMOVE = 80;
	// 水印之间的间隔
	private static final int YMOVE = 80;

	/**
	 * 获取文本长度。汉字为1:1,英文和数字为2:1
	 */
	private 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;
	}

	/**
	 * 给图片添加水印文字
	 * 
	 * @param logoText
	 *            水印文字
	 * @param srcImgPath
	 *            源图片路径
	 * @param targerPath
	 *            目标图片路径
	 */
	public InputStream imageByText(String logoText, String srcImgPath) {
		return imageByText(logoText, srcImgPath, null);
	}

	/**
	 * 给图片添加水印文字、可设置水印文字的旋转角度
	 * 
	 * @param logoText
	 * @param srcImgPath
	 * @param targerPath
	 * @param degree
	 */
	public InputStream imageByText(String logoText, String srcImgPath, Integer degree) {

		ByteArrayOutputStream os = new ByteArrayOutputStream();
		try {
			// 源图片
			Image srcImg = ImageIO.read(this.getImageStream(srcImgPath));
			int width = srcImg.getWidth(null);// 原图宽度
			int height = srcImg.getHeight(null);// 原图高度
			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);
			}
			// 设置水印文字颜色
			g.setColor(color);
			// 设置水印文字Font
			g.setFont(font);
			// 设置水印文字透明度
			g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, alpha));
			g.drawString(logoText, width/2, height);

//			int x = -width / 2;
//			int y = -height / 2;
//			int markWidth = FONT_SIZE * getTextLength(logoText);// 字体长度
//			int markHeight = FONT_SIZE;// 字体高度
//
//			// 循环添加水印
//			while (x < width * 1.5) {
//				y = -height / 2;
//				while (y < height * 1.5) {
//					g.drawString(logoText, x, y);
//
//					y += markHeight + YMOVE;
//				}
//				x += markWidth + XMOVE;
//			}
			// 释放资源
			g.dispose();
			// 生成图片
			// 10、生成图片
			ImageIO.write(buffImg, "JPG", os);
			InputStream in = new ByteArrayInputStream(os.toByteArray());
			return in;
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			try {
				if (null != os)
					os.close();
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
		return null;
	}

	/**
	 * 获取网络图片流
	 * 
	 * @param url
	 * @return
	 */
	private InputStream getImageStream(String url) {
		try {
			HttpURLConnection connection = (HttpURLConnection) new URL(url).openConnection();
			connection.setReadTimeout(5000);
			connection.setConnectTimeout(5000);
			connection.setRequestMethod("GET");
			if (connection.getResponseCode() == HttpURLConnection.HTTP_OK) {
				InputStream inputStream = connection.getInputStream();
				return inputStream;
			}
		} catch (IOException e) {
			System.out.println("获取网络图片出现异常,图片路径为:" + url);
			e.printStackTrace();
		}
		return null;
	}

	@Test
	public void testImage() {
		String srcImgPath = "https://anwanjia.oss-cn-hangzhou.aliyuncs.com/user-res/_2019731105443_8402.jpg";
		Date date=new Date();
		// 水印文字
		String logoText = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(date);
		System.out.println("给图片添加水印文字开始...");
		// 给图片添加正水印文字
		InputStream in=imageByText(logoText, srcImgPath, null);
		Map result = this.iUploadService.uploadToOss("jpg", "service-res", in,
				srcImgPath.substring(srcImgPath.lastIndexOf("/") + 1));
		System.out.println(result.get("ossUrl").toString());
		// 给图片添加斜水印文字
		// imageByText (logoText, srcImgPath, -40);
		System.out.println("给图片添加水印文字结束...");

		// markImageByText("2019-09-03","https://anwanjia.oss-cn-hangzhou.aliyuncs.com/service-res/_2019718192132_9528.jpg",45,new
		// Color(0,0,0),"jpg");
	}
}


你可能感兴趣的:(java 根据网络图片url获取InputStream)