From 00cdca2f95151f37acb25c7d6f330464da354c49 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Tue, 31 Oct 2023 23:23:42 +0800
Subject: [PATCH] 房屋,人员标签查询修改,人员标签数据更新接口新增

---
 src/main/java/org/springblade/modules/house/service/impl/HouseholdLabelServiceImpl.java |   32 ++++++++++++++++++++++++++++++++
 1 files changed, 32 insertions(+), 0 deletions(-)

diff --git a/src/main/java/org/springblade/modules/house/service/impl/HouseholdLabelServiceImpl.java b/src/main/java/org/springblade/modules/house/service/impl/HouseholdLabelServiceImpl.java
index 89c8418..60c9d47 100644
--- a/src/main/java/org/springblade/modules/house/service/impl/HouseholdLabelServiceImpl.java
+++ b/src/main/java/org/springblade/modules/house/service/impl/HouseholdLabelServiceImpl.java
@@ -16,13 +16,18 @@
  */
 package org.springblade.modules.house.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.springblade.common.node.TreeNode;
+import org.springblade.modules.house.entity.HouseLabelEntity;
 import org.springblade.modules.house.entity.HouseholdLabelEntity;
 import org.springblade.modules.house.vo.HouseholdLabelVO;
 import org.springblade.modules.house.mapper.HouseholdLabelMapper;
 import org.springblade.modules.house.service.IHouseholdLabelService;
 import org.springblade.core.mp.base.BaseServiceImpl;
+import org.springblade.modules.label.entity.LabelEntity;
+import org.springblade.modules.label.service.ILabelService;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 
@@ -37,8 +42,35 @@
 @Service
 public class HouseholdLabelServiceImpl extends ServiceImpl<HouseholdLabelMapper, HouseholdLabelEntity> implements IHouseholdLabelService {
 
+
+	@Autowired
+	private ILabelService labelService;
+
 	@Override
 	public IPage<HouseholdLabelVO> selectHouseholdLabelPage(IPage<HouseholdLabelVO> page, HouseholdLabelVO householdLabel) {
 		return page.setRecords(baseMapper.selectHouseholdLabelPage(page, householdLabel));
 	}
+
+	/**
+	 * 住户-标签 自定义新增或修改
+	 * @param householdLabel
+	 * @return
+	 */
+	@Override
+	public boolean saveOrUpdateHouseholdLabel(HouseholdLabelEntity householdLabel) {
+		// 查询标签名称
+		LabelEntity labelEntity = labelService.getById(householdLabel.getLabelId());
+		householdLabel.setLabelName(labelEntity.getLabelName());
+		// 判断同一个住户同一个标签是否已存在,已存在则更新,不存在则新增
+		QueryWrapper<HouseholdLabelEntity> queryWrapper = new QueryWrapper<>();
+		queryWrapper.eq("household_id",householdLabel.getHouseholdId())
+			.eq("label_id",householdLabel.getLabelId());
+		HouseholdLabelEntity one = getOne(queryWrapper);
+		if (null != one){
+			// 更新
+			return updateById(householdLabel);
+		}
+		// 插入
+		return save(householdLabel);
+	}
 }

--
Gitblit v1.9.3