书签导入功能

ruoyi-admin/src/main/java/com/ruoyi/web/controller/yunbookmark/BrowserController.java

@@ -0,0 +1,132 @@
+package com.ruoyi.web.controller.yunbookmark;
+
+
+
+import com.ruoyi.bookmark.domain.SqBookmark;
+import com.ruoyi.bookmark.domain.SqMenu;
+import com.ruoyi.bookmark.service.ISqBookmarkService;
+import com.ruoyi.bookmark.service.ISqMenuService;
+import com.ruoyi.common.core.controller.BaseController;
+import com.ruoyi.common.core.domain.AjaxResult;
+import com.ruoyi.common.core.domain.entity.SysUser;
+import com.ruoyi.common.utils.BookmarkHtml.ImportHtml;
+import com.ruoyi.common.utils.StringUtils;
+import org.jsoup.HttpStatusException;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.dao.DataIntegrityViolationException;
+import org.springframework.security.access.prepost.PreAuthorize;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.multipart.MultipartFile;
+import javax.net.ssl.SSLHandshakeException;
+import java.net.MalformedURLException;
+import java.net.URL;
+import java.text.SimpleDateFormat;
+import java.util.Date;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+
+
+/**
+ * @Auther: Wang
+ * @Date:
+ * 功能描述: 导入书签 导出书签
+ */
+
+@RestController
+@RequestMapping("/IO")
+public class BrowserController extends BaseController {
+    public static Logger logger =  LoggerFactory.getLogger(BrowserController.class);
+
+    @Autowired
+    private ISqMenuService iSqMenuService;
+    @Autowired
+    private ISqBookmarkService iSqBookmarkService;
+
+
+
+    @RequestMapping("/import")
+    @PreAuthorize("@ss.hasPermi('bookmark:bookmark:list')")
+    public AjaxResult importCollect(@RequestParam("htmlFile") MultipartFile htmlFile){
+        logger.debug("开始上传状态是:");
+
+        SysUser sysUser=getAuthUser();
+        Long userID= sysUser.getUserId();
+
+
+        try {
+             Map<String, Map<String, String>> map = ImportHtml.parseHtml(htmlFile.getInputStream());
+          //  Map<String, List<Map>> map = ImportHtml.importHtmlMore(htmlFile.getInputStream());
+                if(null == map || map.isEmpty()){
+                    logger.info("未获取到url连接");
+                    return AjaxResult.error("未获取到url连接,空书签");
+                }
+
+
+                for (Entry<String, Map<String, String>> entry : map.entrySet()) {
+                   String favoritesName = entry.getKey();
+                   //添加目录
+                    SqMenu sqMenu=new SqMenu();
+                    sqMenu.setUserId(userID);
+                    sqMenu.setCreateTime(new Date());
+                    sqMenu.setMenuName(favoritesName);
+                    sqMenu.setParentId(0L);
+                    iSqMenuService.insertSqMenu(sqMenu);
+
+
+                    importHtml(entry.getValue(), sqMenu.getMenuId(), userID);
+                }
+
+        }catch (SSLHandshakeException e){
+            logger.error("文章解析出错:",e);
+        }
+        catch (DataIntegrityViolationException e){
+            logger.error("导入存储异常:",e);
+        }
+        catch (HttpStatusException a){
+            logger.error("文档解析错误:",a);
+        }
+        catch (Exception e) {
+            logger.error("导入html异常:",e);
+        }
+
+        return AjaxResult.success("导入成功");
+    }
+
+
+
+    /**
+     * 导入收藏文章
+     */
+    public void importHtml(Map<String, String> map,Long menuID,Long userId){
+        for(Entry<String, String> entry : map.entrySet()){
+            try {
+                //获取URL后查询最新的URL信息
+                Map<String, String> result = ImportHtml.getCollectFromUrl(entry.getKey());
+                SqBookmark sqBookmark =new SqBookmark();
+                sqBookmark.setUserid(userId);
+                sqBookmark.setTitle(entry.getValue());
+                sqBookmark.setUrl(entry.getKey());
+                sqBookmark.setUrls(ImportHtml.Urlutils(new URL(entry.getKey())));
+                if(StringUtils.isBlank(result.get("description"))){
+                    sqBookmark.setDescription(entry.getValue());
+                }else{
+                    sqBookmark.setDescription(result.get("description"));
+                }
+                sqBookmark.setMenuId(menuID);
+                sqBookmark.setCreateTime(new Date());
+                iSqBookmarkService.insertSqBookmark(sqBookmark);
+            } catch (Exception e) {
+                logger.error("导入存储异常：",e);
+            }
+        }
+
+    }
+
+
+
+}

ruoyi-admin/src/main/resources/logback.xml

@@ -11,7 +11,7 @@
 			<pattern>${log.pattern}</pattern>
 		</encoder>
 	</appender>
-	
+
 	<!-- 系统日志输出 -->
 	<appender name="file_info" class="ch.qos.logback.core.rolling.RollingFileAppender">
 	    <file>${log.path}/sys-info.log</file>
@@ -34,7 +34,7 @@
             <onMismatch>DENY</onMismatch>
         </filter>
 	</appender>
-	
+
 	<appender name="file_error" class="ch.qos.logback.core.rolling.RollingFileAppender">
 	    <file>${log.path}/sys-error.log</file>
         <!-- 循环政策：基于时间创建日志文件 -->
@@ -56,7 +56,7 @@
             <onMismatch>DENY</onMismatch>
         </filter>
     </appender>
-	
+
 	<!-- 用户访问日志输出  -->
     <appender name="sys-user" class="ch.qos.logback.core.rolling.RollingFileAppender">
 		<file>${log.path}/sys-user.log</file>
@@ -70,7 +70,7 @@
             <pattern>${log.pattern}</pattern>
         </encoder>
     </appender>
-	
+
 	<!-- 系统模块日志级别控制  -->
 	<logger name="com.ruoyi" level="info" />
 	<!-- Spring日志级别控制  -->
@@ -79,15 +79,15 @@
 	<root level="info">
 		<appender-ref ref="console" />
 	</root>
-	
+
 	<!--系统操作日志-->
     <root level="info">
         <appender-ref ref="file_info" />
         <appender-ref ref="file_error" />
     </root>
-	
+
 	<!--系统用户操作日志-->
     <logger name="sys-user" level="info">
         <appender-ref ref="sys-user"/>
     </logger>
-</configuration> 
+</configuration>