diff --git a/src/com/hellogithub/servlet/commentServlet.java b/src/com/hellogithub/servlet/commentServlet.java index 1fe879e..0b06e5c 100644 --- a/src/com/hellogithub/servlet/commentServlet.java +++ b/src/com/hellogithub/servlet/commentServlet.java @@ -1,5 +1,6 @@ package com.hellogithub.servlet; +import com.alibaba.fastjson.JSONObject; import com.hellogithub.entity.userEntity; import com.hellogithub.service.commentService; import jakarta.servlet.ServletConfig; @@ -15,6 +16,8 @@ import java.io.PrintWriter; import java.text.DateFormat; import java.text.SimpleDateFormat; import java.util.Date; +import java.util.HashMap; + @WebServlet("/comment") public class commentServlet extends HttpServlet { private com.hellogithub.service.commentService commentService = new commentService(); @@ -52,8 +55,33 @@ public class commentServlet extends HttpServlet { PrintWriter writer = resp.getWriter(); HttpSession session =req.getSession(); userEntity userEntity=(userEntity)session.getAttribute("user"); - String username=userEntity.getUserName(); - writer.println(commentService.insertComment(username,Integer.parseInt(proId),content,Integer.parseInt(isUsed),now,Integer.parseInt(star))); + if(userEntity ==null){ + setResultError("用户未登录",writer); + }else{ + String username=userEntity.getUserName(); + int num = commentService.insertComment(username,Integer.parseInt(proId),content,Integer.parseInt(isUsed),now,Integer.parseInt(star)); + if(num == 0){ + setResultError("评论失败",writer); + }else{ + setResultOK("success",writer); + } + } + writer.close(); + } + public void setResult(Integer code, String msg, PrintWriter writer) { + HashMap result = new HashMap<>(); + result.put("code", code); + result.put("msg", msg); + String jsonString = JSONObject.toJSONString(result); + writer.println(jsonString); writer.close(); } + + public void setResultError(String msg, PrintWriter writer) { + setResult(500, msg, writer); + } + + public void setResultOK(String msg, PrintWriter writer) { + setResult(200, msg, writer); + } }