diff --git a/src/main/java/com/yupi/springbootinit/controller/HostInfoController.java b/src/main/java/com/yupi/springbootinit/controller/HostInfoController.java index 456c403..3f88561 100644 --- a/src/main/java/com/yupi/springbootinit/controller/HostInfoController.java +++ b/src/main/java/com/yupi/springbootinit/controller/HostInfoController.java @@ -14,6 +14,7 @@ import org.apache.ibatis.annotations.Param; import org.springframework.web.bind.annotation.*; import javax.annotation.Resource; +import java.util.ArrayList; import java.util.List; /* @@ -39,9 +40,13 @@ public class HostInfoController { } @GetMapping("country_info") - public BaseResponse> getCountryInfo(@RequestParam(name = "countryName") String countryName){ - - return ResultUtils.success(hostInfoService.getCountryInfo(countryName)); + public BaseResponse> getCountryInfo(@RequestParam(name = "countryName") String countryName){ + List countryInfo = hostInfoService.getCountryInfo(countryName); + ListcountryNameList = new ArrayList<>(); + for (CountryInfoVO countryInfoVO : countryInfo) { + countryNameList.add(countryInfoVO.getCountryName()); + } + return ResultUtils.success(countryNameList); } @PostMapping("query_count")