#

Junjie

2024-01-11 901b2ab3e0e7c592602848cfa256f34c5b0c7bb8

 zy-asrs-common/src/main/java/com/zy/asrs/common/wms/service/impl/TagServiceImpl.java

@@ -5,8 +5,11 @@
import com.zy.asrs.common.wms.mapper.TagMapper;
import com.zy.asrs.common.wms.service.TagService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.zy.asrs.framework.common.Cools;
import com.zy.asrs.framework.exception.CoolException;
import org.springframework.stereotype.Service;

import java.util.Date;

@Service("manTagService")
public class TagServiceImpl extends ServiceImpl<TagMapper, Tag> implements TagService {
@@ -30,4 +33,61 @@
        return top;
    }

    @Override
    public Tag getTag(String firstTagName, String secondTagName, Long hostId) {
        Date now = new Date();
        Tag tag = this.getOne(new LambdaQueryWrapper<Tag>()
                .eq(Tag::getName, "全部")
                .eq(Tag::getHostId, hostId)
                .isNull(Tag::getParentId));
        if (tag == null) {
            throw new CoolException("分类系统异常");
        }

        if (Cools.isEmpty(firstTagName)) {
            //一级分类，直接返回根分类(全部)
            return tag;
        }

        if (!Cools.isEmpty(firstTagName)) {
            //存在一级分类，查询一级分类
            Tag firstTag = this.getOne(new LambdaQueryWrapper<Tag>()
                    .eq(Tag::getName, firstTagName)
                    .eq(Tag::getHostId, hostId));
            if (firstTag == null) {
                //一级分类为空，创建一级分类
                firstTag = new Tag();
                firstTag.setName(firstTagName);
                firstTag.setParentId(tag.getId());
                firstTag.setStatus(1);
                firstTag.setHostId(hostId);
                firstTag.setCreateTime(now);
                firstTag.setUpdateTime(now);
                this.save(firstTag);
            }

            if (!Cools.isEmpty(secondTagName)) {
                //存在二级分类，查询二级分类
                Tag secondTag = this.getOne(new LambdaQueryWrapper<Tag>()
                        .eq(Tag::getName, secondTagName)
                        .eq(Tag::getHostId, hostId));
                if (secondTag == null) {
                    //二级分类为空，创建二级分类
                    secondTag = new Tag();
                    secondTag.setName(secondTagName);
                    secondTag.setParentId(firstTag.getId());
                    secondTag.setStatus(1);
                    secondTag.setHostId(hostId);
                    secondTag.setCreateTime(now);
                    secondTag.setUpdateTime(now);
                    this.save(secondTag);
                    return secondTag;
                }
                return secondTag;
            }else {
                return firstTag;//不存在二级分类，直接返回一级分类
            }
        }
        return tag;
    }
}