feats

src/main/java/com/huiju/estateagents/controller/TaExtendContentController.java

@@ -6,7 +6,9 @@ import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.huiju.estateagents.base.BaseController;
 import com.huiju.estateagents.base.ResponseBean;
 import com.huiju.estateagents.entity.TaExtendContent;
+import com.huiju.estateagents.service.ITaBuildingService;
 import com.huiju.estateagents.service.ITaExtendContentService;
+import org.omg.CORBA.PUBLIC_MEMBER;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -35,6 +37,8 @@ public class TaExtendContentController extends BaseController {
     @Autowired
     public ITaExtendContentService iTaExtendContentService;
 
+    @Autowired
+    public ITaBuildingService iTaBuildingService;
 
     /**
      * 分页查询列表
@@ -71,6 +75,9 @@ public class TaExtendContentController extends BaseController {
     public ResponseBean taExtendContentAdd(@RequestBody TaExtendContent taExtendContent){
         ResponseBean responseBean = new ResponseBean();
         try {
+            Integer cityId = iTaBuildingService.getCityById(taExtendContent.getBuildingId());
+            taExtendContent.setCityId(cityId);
+
             if (iTaExtendContentService.save(taExtendContent)){
                 responseBean.addSuccess(taExtendContent);
             }else {
@@ -117,6 +124,9 @@ public class TaExtendContentController extends BaseController {
                                         @RequestBody TaExtendContent taExtendContent){
         ResponseBean responseBean = new ResponseBean();
         try {
+            Integer cityId = iTaBuildingService.getCityById(taExtendContent.getBuildingId());
+            taExtendContent.setCityId(cityId);
+
             if (iTaExtendContentService.updateById(taExtendContent)){
                 responseBean.addSuccess(taExtendContent);
             }else {

src/main/java/com/huiju/estateagents/controller/TaGoodsController.java

@@ -9,6 +9,7 @@ import com.huiju.estateagents.common.CommConstant;
 import com.huiju.estateagents.common.JWTUtils;
 import com.huiju.estateagents.common.StringUtils;
 import com.huiju.estateagents.entity.TaGoods;
+import com.huiju.estateagents.service.ITaBuildingService;
 import com.huiju.estateagents.service.ITaGoodsService;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -41,6 +42,8 @@ public class TaGoodsController extends BaseController {
     @Autowired
     public ITaGoodsService iTaGoodsService;
 
+    @Autowired
+    public ITaBuildingService iTaBuildingService;
 
     /**
      * 分页查询列表
@@ -79,6 +82,10 @@ public class TaGoodsController extends BaseController {
     public ResponseBean taGoodsAdd(@RequestBody TaGoods taGoods){
         ResponseBean responseBean = new ResponseBean();
         try {
+
+            Integer cityId = iTaBuildingService.getCityById(taGoods.getBuildingId());
+            taGoods.setCityId(cityId);
+
             if (iTaGoodsService.save(taGoods)){
                 responseBean.addSuccess(taGoods);
             }else {
@@ -123,6 +130,10 @@ public class TaGoodsController extends BaseController {
     public ResponseBean taGoodsUpdate(@RequestBody TaGoods taGoods){
         ResponseBean responseBean = new ResponseBean();
         try {
+
+            Integer cityId = iTaBuildingService.getCityById(taGoods.getBuildingId());
+            taGoods.setCityId(cityId);
+
             if (iTaGoodsService.updateById(taGoods)){
                 responseBean.addSuccess(taGoods);
             }else {

src/main/java/com/huiju/estateagents/entity/TaExtendContent.java

@@ -85,5 +85,5 @@ public class TaExtendContent implements Serializable {
      */
     private String buildingId;
 
-
+    private Integer cityId;
 }

src/main/java/com/huiju/estateagents/entity/TaGoods.java

@@ -107,6 +107,8 @@ public class TaGoods implements Serializable {
      */
     private String address;
 
+    private Integer cityId;
+
     /**
      * 兑换人
      */

src/main/java/com/huiju/estateagents/service/ITaBuildingService.java

@@ -136,4 +136,6 @@ public interface ITaBuildingService extends IService<TaBuilding> {
      * @return
      */
     ResponseBean buildingAll();
+
+    Integer getCityById(String buildingId);
 }

src/main/java/com/huiju/estateagents/service/impl/TaBuildingServiceImpl.java

@@ -477,6 +477,16 @@ public class TaBuildingServiceImpl extends ServiceImpl<TaBuildingMapper, TaBuild
         return ResponseBean.success(taBuilding);
     }
 
+    @Override
+    public Integer getCityById(String buildingId) {
+        if (StringUtils.isEmpty(buildingId)) return null;
+
+        TaBuilding taBuilding = taBuildingMapper.selectById(buildingId);
+        if (null == taBuilding) return null;
+
+        return taBuilding.getCityId();
+    }
+
 
     int insertImgBatch(List<TaBuildingImg> imgs, String buildingId) {
         int rows = 0;