添加接口

src/main/java/com/fdage/followheartplay/controller/RequirementController.java

@@ -0,0 +1,28 @@
+package com.fdage.followheartplay.controller;
+
+import com.fdage.followheartplay.entity.RequirementEntity;
+import com.fdage.followheartplay.service.RequirementService;
+import com.fdkankan.common.response.ResultData;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.*;
+
+@RestController
+@RequestMapping("/requirement")
+public class RequirementController {
+
+    @Autowired
+    private RequirementService requirementService;
+
+
+    /**
+     * 查询场景信息
+     * @param token
+     * @param requirementEntity
+     * @return
+     */
+    @PostMapping("/save")
+    public ResultData saveScene(@RequestHeader String token, @RequestBody RequirementEntity requirementEntity){
+        requirementService.save(requirementEntity);
+        return ResultData.ok();
+    }
+}

src/main/java/com/fdage/followheartplay/controller/SceneController.java

@@ -1,16 +1,14 @@
 package com.fdage.followheartplay.controller;
 
 import com.fdage.followheartplay.dto.SceneDto;
+import com.fdage.followheartplay.entity.SceneEntity;
 import com.fdage.followheartplay.service.SceneService;
 import com.fdkankan.common.constant.ServerCode;
 import com.fdkankan.common.response.ResultData;
 import com.fdkankan.common.util.JwtUtil;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.util.ObjectUtils;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestHeader;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
 
@@ -37,6 +35,18 @@ public class SceneController {
     /**
      * 查询场景信息
      * @param token
+     * @param sceneEntity
+     * @return
+     */
+    @PostMapping("/save")
+    public ResultData saveScene(@RequestHeader String token, @RequestBody SceneEntity sceneEntity){
+        sceneService.save(sceneEntity);
+        return ResultData.ok();
+    }
+
+    /**
+     * 查询场景信息
+     * @param token
      * @param sceneIds 场景码集合
      * @return
      */

src/main/java/com/fdage/followheartplay/entity/RequirementEntity.java

@@ -0,0 +1,53 @@
+package com.fdage.followheartplay.entity;
+
+import javax.persistence.*;
+
+/**
+ * 场景实体
+ */
+@Entity
+@Table(name="requirement",schema = "follow_heart_play")
+public class RequirementEntity extends BaseEntity{
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    @Column(name = "id")
+    private Long id;
+    @Column(name = "industry_name")
+    private String  industryName;
+    @Column(name = "requirement_desc")
+    private String requirementDesc;
+    @Column(name = "phone")
+    private String phone;
+
+    public Long getId() {
+        return id;
+    }
+
+    public void setId(Long id) {
+        this.id = id;
+    }
+
+    public String getIndustryName() {
+        return industryName;
+    }
+
+    public void setIndustryName(String industryName) {
+        this.industryName = industryName;
+    }
+
+    public String getRequirementDesc() {
+        return requirementDesc;
+    }
+
+    public void setRequirementDesc(String requirementDesc) {
+        this.requirementDesc = requirementDesc;
+    }
+
+    public String getPhone() {
+        return phone;
+    }
+
+    public void setPhone(String phone) {
+        this.phone = phone;
+    }
+}

src/main/java/com/fdage/followheartplay/repository/RequirementRepository.java

@@ -0,0 +1,10 @@
+package com.fdage.followheartplay.repository;
+
+import com.fdage.followheartplay.entity.RequirementEntity;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public interface RequirementRepository extends JpaRepository<RequirementEntity,Long> {
+
+}

src/main/java/com/fdage/followheartplay/service/RequirementService.java

@@ -0,0 +1,18 @@
+package com.fdage.followheartplay.service;
+
+import com.fdage.followheartplay.entity.RequirementEntity;
+import com.fdage.followheartplay.repository.RequirementRepository;
+import com.fdage.followheartplay.repository.SceneBuyRecordRepository;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+@Service
+public class RequirementService {
+
+    @Autowired
+    private RequirementRepository requirementRepository;
+
+    public void save(RequirementEntity requirementEntity) {
+        requirementRepository.save(requirementEntity);
+    }
+}

src/main/java/com/fdage/followheartplay/service/SceneService.java

@@ -2,6 +2,7 @@ package com.fdage.followheartplay.service;
 
 import com.fdage.followheartplay.dto.SceneDto;
 import com.fdage.followheartplay.entity.SceneBuyRecordEntity;
+import com.fdage.followheartplay.entity.SceneEntity;
 import com.fdage.followheartplay.repository.SceneBuyRecordRepository;
 import com.fdage.followheartplay.repository.SceneRepository;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -52,4 +53,8 @@ public class SceneService {
         }
         repository.updateTbStatus(username, Arrays.stream(sceneIds.split(",")).map(Long::valueOf).collect(Collectors.toList()), 1);
     }
+
+    public void save(SceneEntity sceneEntity) {
+        sceneRepository.save(sceneEntity);
+    }
 }