package org.jeecg.modules.base.service.impl;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.jeecg.modules.base.entity.Unit;
import org.jeecg.modules.base.mapper.UnitMapper;
import org.jeecg.modules.base.mapper.UnitCategoryMapper;
import org.jeecg.modules.base.service.IUnitService;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;

@Service
public class UnitServiceImpl extends ServiceImpl<UnitMapper, Unit> implements IUnitService {

    @Resource
    private UnitMapper unitMapper;
    @Resource
    private UnitCategoryMapper unitCategoryMapper;

    @Override
    public List<Unit> getListByUnitCategoryId(String unitCategoryId) {
        List<Unit> list = new ArrayList<>();
        list = unitMapper.getListByUnitCategoryId(unitCategoryId);
        return list;
    }

    @Override
    public List<Unit> getTreeList() {
        List<Unit> list = unitMapper.getTreeList();
        return list;
    }
}