修改接口，增加字段

陈浩
1 parent 40489c6a
Showing 2 changed files with 79 additions and 10 deletions
src/main/java/zzsy/controller/UserMessengeController.java
src/main/java/zzsy/entity/UserMessenge.java
@@ -9,7 +9,11 @@ import org.apache.poi.xssf.usermodel.XSSFSheet;
 import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.dao.DataIntegrityViolationException;
-import org.springframework.web.bind.annotation.*;
+import org.springframework.data.domain.Example;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
 import zzsy.dao.UserMessengeRepository;
 import zzsy.entity.UserMessenge;
 import zzsy.utils.ErrorEnum;
@@ -18,6 +22,7 @@ import zzsy.utils.ResponseBean;
 import javax.servlet.ServletOutputStream;
 import javax.servlet.http.HttpServletResponse;
 import java.io.*;
+import java.text.SimpleDateFormat;
 import java.util.List;
 import java.util.regex.Pattern;
@@ -33,17 +38,32 @@ public class UserMessengeController {
     UserMessengeRepository userMessengeService;
     @PostMapping("save")
-    public ResponseBean save(@RequestParam(name = "name") String name, @RequestParam(name = "phone")String phone, @RequestParam(name = "company")String company) {
+    public ResponseBean save(@RequestParam(name = "name") String name, @RequestParam(name = "phone")String phone, @RequestParam(name = "company")String company,@RequestParam(name = "videoId")Integer videoId,
+                             @RequestParam(name = "videoTitle") String videoTitle, @RequestParam(name = "videoUrl") String videoUrl){
         ResponseBean responseBean = new ResponseBean();
         try {
             String REGEX_PHONE = "^((13[0-9])|(14[5,7,9])|(15([0-3]|[5-9]))|(166)|(17[0,1,3,5,6,7,8])|(18[0-9])|(19[8|9]))\\d{8}$";
             boolean isPhone = Pattern.matches(REGEX_PHONE, phone);
             if(isPhone){
                 UserMessenge userMessenge = new UserMessenge();
-                userMessenge.setName(name);
-                userMessenge.setCompany(company);
                 userMessenge.setPhone(phone);
-                userMessengeService.save(userMessenge);
+                userMessenge.setVideoId(videoId);
+                Example<UserMessenge> example = Example.of(userMessenge);
+                List<UserMessenge> userMessengeList = userMessengeService.findAll(example);
+                if(userMessengeList.isEmpty()){
+                    userMessenge.setName(name);
+                    userMessenge.setCompany(company);
+                    userMessenge.setVideoTitle(videoTitle);
+                    userMessenge.setVideoUrl(videoUrl);
+                    userMessengeService.save(userMessenge);
+                }else {
+                    userMessenge.setId(userMessengeList.get(0).getId());
+                    userMessenge.setName(name);
+                    userMessenge.setCompany(company);
+                    userMessenge.setVideoTitle(videoTitle);
+                    userMessenge.setVideoUrl(videoUrl);
+                    userMessengeService.saveAndFlush(userMessenge);
+                }
             }else{
                 responseBean = new ResponseBean(ErrorEnum.PHONE_ERROR.getErrorCode(),ErrorEnum.PHONE_ERROR.getErrorMessage());
             }
@@ -74,20 +94,36 @@ public class UserMessengeController {
             XSSFSheet sheetlist = wb.createSheet(fileName);
             XSSFRow row = sheetlist.createRow(0);//行
             XSSFCell cell = row.createCell(0);//列
-            cell.setCellValue("姓名");
+            cell.setCellValue("填写时间");
             cell = row.createCell(1);//列
-            cell.setCellValue("电话");
+            cell.setCellValue("视频名称");
             cell = row.createCell(2);//列
+            cell.setCellValue("视频链接");
+            cell = row.createCell(3);//列
+            cell.setCellValue("姓名");
+            cell = row.createCell(4);//列
+            cell.setCellValue("电话");
+            cell = row.createCell(5);//列
             cell.setCellValue("公司");
-
+            cell = row.createCell(6);//列
+            cell.setCellValue("视频id");
+            SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:SS");
             for (int i = 0; i < userMessengeList.size(); i++) {
                 row = sheetlist.createRow((short) (i + 1));//行
                 cell = row.createCell(0);//列
-                cell.setCellValue(userMessengeList.get(i).getName());
+                cell.setCellValue(dateFormat.format(userMessengeList.get(i).getInsertDate()));
                 cell = row.createCell(1);//列
-                cell.setCellValue(userMessengeList.get(i).getPhone());
+                cell.setCellValue(userMessengeList.get(i).getVideoTitle());
                 cell = row.createCell(2);//列
+                cell.setCellValue(userMessengeList.get(i).getVideoUrl());
+                cell = row.createCell(3);//列
+                cell.setCellValue(userMessengeList.get(i).getName());
+                cell = row.createCell(4);//列
+                cell.setCellValue(userMessengeList.get(i).getPhone());
+                cell = row.createCell(5);//列
                 cell.setCellValue(userMessengeList.get(i).getCompany());
+                cell = row.createCell(6);//列
+                cell.setCellValue(userMessengeList.get(i).getVideoId());
             }
             ByteArrayOutputStream os = new ByteArrayOutputStream();
@@ -125,4 +161,5 @@ public class UserMessengeController {
             }
         }
     }
+
 }
@@ -6,6 +6,7 @@ package zzsy.entity;
 import javax.persistence.*;
 import java.io.Serializable;
 import java.sql.Timestamp;
+import java.text.SimpleDateFormat;
 /**
  * @Auther: Administrator
@@ -27,6 +28,12 @@ public class UserMessenge implements Serializable {
     private String phone;
     private String company;
+    @Column(name = "video_id")
+    private Integer videoId;
+    @Column(name = "video_title")
+    private String videoTitle;
+    @Column(name = "video_url")
+    private String videoUrl;
     @Column(name = "INSERT_DATE")
     private Timestamp insertDate;
@@ -63,7 +70,32 @@ public class UserMessenge implements Serializable {
         this.company = company;
     }
+    public String getVideoTitle() {
+        return videoTitle;
+    }
+
+    public void setVideoTitle(String videoTitle) {
+        this.videoTitle = videoTitle;
+    }
+
+    public Integer getVideoId() {
+        return videoId;
+    }
+
+    public void setVideoId(Integer videoId) {
+        this.videoId = videoId;
+    }
+
+    public String getVideoUrl() {
+        return videoUrl;
+    }
+
+    public void setVideoUrl(String videoUrl) {
+        this.videoUrl = videoUrl;
+    }
+
     public Timestamp getInsertDate() {
+
         return insertDate;
     }