package org.springblade.modules.farmplant.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springblade.common.node.TreeNode;
import org.springblade.modules.farmplant.entity.Strain;
import org.springblade.modules.farmplant.mapper.StrainMapper;
import org.springblade.modules.farmplant.service.StrainService;
import org.springblade.modules.farmplant.vo.StrainVO;
import org.springframework.stereotype.Service;

import java.util.List;

/**
 * 种养品种服务实现类
 * @since 2022-05-12
 * @author zhongrj
 */
@Service
public class StrainServiceImpl extends ServiceImpl<StrainMapper, Strain> implements StrainService {

	/**
     * 自定义分页
	 * @param page
     * @param strain
     * @return
     */
	@Override
	public IPage<StrainVO> selectStrainPage(IPage<StrainVO> page, StrainVO strain) {
		return page.setRecords(baseMapper.selectStrainPage(page, strain));
	}

	/**
	 * 查询种类信息
	 * @param strain
	 * @return
	 */
	@Override
	public List<Strain> strainTree(StrainVO strain) {
		return list(new QueryWrapper<>(strain));
	}
}