Weekend

Author: abel533

src/main/java/tk/mybatis/springboot/controller/CountryController.java

@@ -51,7 +51,7 @@ public class CountryController {
     @RequestMapping
     public ModelAndView getAll(Country country) {
         ModelAndView result = new ModelAndView("index");
-        List<Country> countryList = countryService.getAll(country);
+        List<Country> countryList = countryService.getAllByWeekend(country);
         result.addObject("pageInfo", new PageInfo<Country>(countryList));
         result.addObject("queryParam", country);
         result.addObject("page", country.getPage());

src/main/java/tk/mybatis/springboot/service/CountryService.java

@@ -28,6 +28,8 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import tk.mybatis.mapper.entity.Example;
+import tk.mybatis.mapper.weekend.Weekend;
+import tk.mybatis.mapper.weekend.WeekendCriteria;
 import tk.mybatis.springboot.mapper.CountryMapper;
 import tk.mybatis.springboot.model.Country;
 
@@ -58,6 +60,21 @@ public List<Country> getAll(Country country) {
         return countryMapper.selectByExample(example);
     }
 
+    public List<Country> getAllByWeekend(Country country) {
+        if (country.getPage() != null && country.getRows() != null) {
+            PageHelper.startPage(country.getPage(), country.getRows());
+        }
+        Weekend<Country> weekend = Weekend.of(Country.class);
+        WeekendCriteria<Country, Object> criteria = weekend.weekendCriteria();
+        if (country.getCountryname() != null && country.getCountryname().length() > 0) {
+            criteria.andLike(Country::getCountryname, "%" + country.getCountryname() + "%");
+        }
+        if (country.getCountrycode() != null && country.getCountrycode().length() > 0) {
+            criteria.andLike(Country::getCountrycode, "%" + country.getCountrycode() + "%");
+        }
+        return countryMapper.selectByExample(weekend);
+    }
+
     public Country getById(Integer id) {
         return countryMapper.selectByPrimaryKey(id);
     }