|
|
@@ -1,6 +1,7 @@
|
|
|
package com.izouma.nineth.web;
|
|
|
|
|
|
import com.izouma.nineth.domain.Collection;
|
|
|
+import com.izouma.nineth.domain.FileObject;
|
|
|
import com.izouma.nineth.dto.CollectionDTO;
|
|
|
import com.izouma.nineth.dto.CreateBlindBox;
|
|
|
import com.izouma.nineth.dto.PageQuery;
|
|
|
@@ -13,13 +14,16 @@ import com.izouma.nineth.utils.SecurityUtils;
|
|
|
import com.izouma.nineth.utils.excel.ExcelUtils;
|
|
|
import io.swagger.annotations.ApiOperation;
|
|
|
import lombok.AllArgsConstructor;
|
|
|
+import org.apache.commons.lang3.StringUtils;
|
|
|
import org.springframework.cache.annotation.CacheEvict;
|
|
|
import org.springframework.data.domain.Page;
|
|
|
import org.springframework.web.bind.annotation.*;
|
|
|
|
|
|
import javax.servlet.http.HttpServletResponse;
|
|
|
import java.io.IOException;
|
|
|
+import java.util.Collections;
|
|
|
import java.util.List;
|
|
|
+import java.util.stream.Collectors;
|
|
|
|
|
|
@RestController
|
|
|
@RequestMapping("/collection")
|
|
|
@@ -97,7 +101,13 @@ public class CollectionController extends BaseController {
|
|
|
|
|
|
@GetMapping("/recommend")
|
|
|
public List<CollectionDTO> recommend(@RequestParam String type) {
|
|
|
- return collectionService.toDTO(collectionRepo.recommend(type));
|
|
|
+ return collectionService.toDTO(collectionRepo.recommend(type).stream().map(rc -> {
|
|
|
+ if (StringUtils.isNotBlank(rc.getRecommend().getPic())) {
|
|
|
+ rc.getCollection().setPic(Collections.singletonList(new FileObject(null, rc.getRecommend()
|
|
|
+ .getPic(), null, null)));
|
|
|
+ }
|
|
|
+ return rc.getCollection();
|
|
|
+ }).collect(Collectors.toList()));
|
|
|
}
|
|
|
}
|
|
|
|