package com.dsh.course.service.impl; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.dsh.course.entity.OpenCityBusiness; import com.dsh.course.mapper.OpenCityBusinessMapper; import com.dsh.course.model.vo.BaseWarpper; import com.dsh.course.service.IOpenCityBusinessService; import org.springframework.stereotype.Service; import javax.annotation.Resource; import java.util.ArrayList; import java.util.List; @Service public class OpenCityBusinessServiceImpl extends ServiceImpl implements IOpenCityBusinessService { @Resource private OpenCityBusinessMapper openCityBusinessMapper; /** * 获取业务类型 * @param province 省名 * @param city 市名称 * @param district 区县名称 * @return * @throws Exception */ @Override public List queryBusiness(String province, String city, String district) throws Exception { List openCityBusinesses = openCityBusinessMapper.queryBusiness(province, city, district); if(openCityBusinesses.size() == 0){ openCityBusinesses = openCityBusinessMapper.queryBusiness(province, city, null); } if(openCityBusinesses.size() == 0){ openCityBusinesses = openCityBusinessMapper.queryBusiness(province, null, null); } List list = new ArrayList<>(); for(OpenCityBusiness b : openCityBusinesses){ BaseWarpper baseWarpper = new BaseWarpper(); switch (b.getBusinessType()){ case 1: baseWarpper.setName("专车"); baseWarpper.setId(1); break; case 4: baseWarpper.setName("同城小件物流"); baseWarpper.setId(4); break; } list.add(baseWarpper); } return list; } /** * 根据开通的城市id获取业务类型 * @param id * @return * @throws Exception */ @Override public List queryBusinessById(Integer id) throws Exception { List openCityBusinesses = openCityBusinessMapper.queryBusinessById(id); List list = new ArrayList<>(); for(OpenCityBusiness b : openCityBusinesses){ BaseWarpper baseWarpper = new BaseWarpper(); switch (b.getBusinessType()){ case 1: baseWarpper.setName("专车"); baseWarpper.setId(1); break; case 4: baseWarpper.setName("同城小件物流"); baseWarpper.setId(4); break; } list.add(baseWarpper); } return list; } }