From 90d92b4a169c336793d0c0065c70e19276f23805 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E8=B6=85=E7=84=B6?= <tjpu626@163.com>
Date: Thu, 2 Dec 2021 10:18:04 +0800
Subject: [PATCH] =?UTF-8?q?zcr:=E9=93=BE=E6=8E=A5=E5=9B=BE=E6=A0=87?=
 =?UTF-8?q?=E6=96=87=E4=BB=B6=E6=B5=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../sarUrl/controller/TsUrlController.java    | 57 +++++++++++++++++++
 .../com/adc/da/slrs/sarUrl/entity/TsUrl.java  |  6 ++
 2 files changed, 63 insertions(+)

diff --git a/adc-da-slrs/src/main/java/com/adc/da/slrs/sarUrl/controller/TsUrlController.java b/adc-da-slrs/src/main/java/com/adc/da/slrs/sarUrl/controller/TsUrlController.java
index 12edcf60..f4a48ad6 100644
--- a/adc-da-slrs/src/main/java/com/adc/da/slrs/sarUrl/controller/TsUrlController.java
+++ b/adc-da-slrs/src/main/java/com/adc/da/slrs/sarUrl/controller/TsUrlController.java
@@ -1,6 +1,8 @@
 package com.adc.da.slrs.sarUrl.controller;
 
 
+import com.adc.da.att.entity.AttFileEO;
+import com.adc.da.att.service.IAttFileEOService;
 import com.adc.da.http.ResponseMessage;
 import com.adc.da.http.Result;
 import com.adc.da.slrs.sarUrl.entity.OrderUrlVO;
@@ -12,12 +14,21 @@ import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.servlet.ModelAndView;
 
+import javax.imageio.ImageIO;
+import javax.servlet.ServletOutputStream;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
+import javax.servlet.http.HttpSession;
+import javax.swing.*;
+import java.awt.*;
+import java.awt.image.BufferedImage;
+import java.io.*;
 import java.util.Arrays;
+import java.util.Base64;
 import java.util.List;
 
 /**
@@ -34,6 +45,52 @@ public class TsUrlController{
 	@Autowired
 	private ITsUrlService tsUrlService;
 
+	@Autowired
+	private IAttFileEOService iAttFileEOService;
+
+	@Value("${file.path}")
+	private String filePath;
+
+	@GetMapping("getIconImg")
+	public String getIconImg(HttpServletRequest request, HttpServletResponse response){
+		String url = "";
+		String path = "api/att/attFile/upload";
+		ServletOutputStream outputStream = null;
+		FileInputStream inputStream = null;
+		//通过id获取路径
+		String id = request.getParameter("id");
+		String fid = tsUrlService.getById(id).getIcon();
+		List<AttFileEO> attFileEOS = iAttFileEOService.getMultiFileInfos(fid);
+		if(attFileEOS.isEmpty()){
+			return null;
+		}
+		AttFileEO attFileEO = attFileEOS.get(0);
+		url = filePath + attFileEO.getFilePath() + attFileEO.getFileName();
+
+		try {
+			System.out.println(id);
+			TsUrl tsUrl = tsUrlService.getById(id);
+			System.out.println(tsUrl);
+
+			File file = new File(url);
+			if(!file.exists()) return null;
+			inputStream = new FileInputStream(file);
+			response.setContentType("image/jpeg");
+			outputStream = response.getOutputStream();
+			int len = 0;
+			byte[] buffer = new byte[1024 * 10];
+			while ((len = inputStream.read(buffer)) != -1){
+				outputStream.write(buffer,0,len);
+			}
+			outputStream.flush();
+			outputStream.close();
+			inputStream.close();
+		} catch (IOException e) {
+			System.out.println("---------图标获取异常--------------");
+		}
+		return null;
+	}
+
 	@ApiOperation("查询")
 	@GetMapping
 	public ResponseMessage<List<TsUrl>> getUrls(){
diff --git a/adc-da-slrs/src/main/java/com/adc/da/slrs/sarUrl/entity/TsUrl.java b/adc-da-slrs/src/main/java/com/adc/da/slrs/sarUrl/entity/TsUrl.java
index 6798937c..12d6c292 100644
--- a/adc-da-slrs/src/main/java/com/adc/da/slrs/sarUrl/entity/TsUrl.java
+++ b/adc-da-slrs/src/main/java/com/adc/da/slrs/sarUrl/entity/TsUrl.java
@@ -61,4 +61,10 @@ public class TsUrl implements Serializable {
 	/**排序*/
     @ApiModelProperty(value = "排序")
     private Integer orderBy;
+
+    /**新增*/
+    @ApiModelProperty(value = "图标信息")
+    private String icon;
+    @ApiModelProperty(value = "位置")
+    private Integer type;
 }