获取实例信息的同时判断该实例是否为用户所喜爱

Author: WxrLZQ

springboot-dubbo-api/src/main/java/com/lzq/api/pojo/Content.java

@@ -64,6 +64,12 @@ public class Content implements Serializable {
     @TableField(value = "js_style")
     @JsonProperty("jsStyle")
     private String jsStyle;
+    /**
+     * 是否喜爱
+     */
+    @TableField(exist = false)
+    @JsonProperty("myFavorites")
+    private Boolean myFavorites=false;
     /**
      * 创建时间
      */

springboot-dubbo-api/src/main/java/com/lzq/api/service/ContentService.java

@@ -11,30 +11,34 @@ public interface ContentService {
 
     /**
      * 添加实例内容
+     *
      * @param content
      * @return
      */
-    public Boolean addContent(Content content);
+    Boolean addContent(Content content);
 
     /**
      * 更新实例内容
+     *
      * @param content
      * @return
      */
-    public Boolean updateContent(Content content);
+    Boolean updateContent(Content content);
 
     /**
      * 获取实例内容
+     *
      * @param exampleId
      * @param username
      * @return
      */
-    public Content getContent(String exampleId,String username);
+    Content getContent(String exampleId, String username);
 
     /**
      * 删除实例内容
+     *
      * @param exampleId 实例id
      * @return
      */
-     void deleteContent(String exampleId);
+    void deleteContent(String exampleId);
 }

springboot-dubbo-web/src/main/java/com/lzq/web/controller/ContentController.java

@@ -1,16 +1,23 @@
 package com.lzq.web.controller;
 
+import com.auth0.jwt.JWT;
 import com.lzq.api.pojo.Content;
 import com.lzq.api.service.ContentService;
+import com.lzq.web.utils.JWTUtils;
 import com.lzq.web.utils.ResultMapUtils;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang.StringUtils;
 import org.apache.dubbo.config.annotation.Reference;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
+import javax.servlet.http.HttpServletRequest;
+import java.util.List;
 import java.util.Map;
 
 /**
@@ -27,6 +34,9 @@ public class ContentController {
     @Reference
     private ContentService contentService;
 
+    @Autowired
+    private RedisTemplate redisTemplate;
+
     /**
      * 获取实例内容
      *
@@ -36,9 +46,19 @@ public class ContentController {
      */
     @GetMapping("/getContent")
     @ApiOperation("获取实例内容")
-    public Map<String, Object> getContent(String exampleId, String username) {
+    public Map<String, Object> getContent(HttpServletRequest request, String exampleId, String username) {
         log.info("获取实例内容接口"+exampleId+"------"+username);
+        String token = request.getHeader("token");
         Content content = contentService.getContent(exampleId, username);
+        //当token不为空时用户已登录
+        if (StringUtils.isNotBlank(token)){
+            String s = JWTUtils.verify(token).getClaim("username").asString();
+            List<Integer> favoriteslist = redisTemplate.opsForList().range(s + "fav", 0, -1);
+            //判断改实例是否为用户的喜爱
+            if (favoriteslist.contains(content.getExampleId())){
+                content.setMyFavorites(true);
+            }
+        }
         return ResultMapUtils.ResultMap(true, 0, content);
     }
 }

springboot-dubbo-web/src/main/java/com/lzq/web/utils/ExampleUtils.java

@@ -83,8 +83,8 @@ public static String screenshot(String username, String filename) throws IOExcep
         options.setHeadless(true);
         ChromeDriver broswer = new ChromeDriver(options);
         //设置浏览器窗口大小
-        int winHeight = 800;
-        int winWidth = 1280;
+        int winHeight = 528;
+        int winWidth = 960;
         Dimension dim = new Dimension(winWidth, winHeight);
         broswer.manage().window().setSize(dim);
         //等待1秒,