![](/img/trans.png)
[英]How to make a custom method in JpaRepository from Spring Data JPA?
[英]Spring Data JPA - Custom Sort in JpaRepository
我正在使用 Spring Data JPA 和 Spring Data REST,並且我為我的Thing
實體創建了一個 JpaRepository。
@Repository
public interface ThingRepository extends JpaRepository<Thing, Long> {
@RestResource(path = "findByName", rel = "findByName")
Page findByName(@Param("name") String name, Pageable page);
}
我想應用排序。 我想根據自定義評級算法對Thing
列表進行排序。
List<Thing> sortByRating(List<Thing> things){
// custom logic
return things;
};
如果可能的話,我想使用自定義函數來對結果進行排序。 Spring JPA 中這種事情的最佳方法是什么? 如何讓我的存儲庫使用我的函數對結果集進行排序?
Spring Data 提供了基本的排序,如此處所示。 您還可以使用@Query
注釋您的查詢並使用order by
,但我相信您需要的是更復雜的排序邏輯。
如果您使用的是 Java 8,您可以使用它的功能對列表進行排序, 本文展示了很好的示例,這需要更多的工作,但看起來就像您所追求的那樣。 請注意,為了從stream
轉換為list
,您必須包含.collect(Collectors.toList());
在stream
的末尾。
示例(取自文章):
employees.stream().sorted((Employee e1, Employee e2) -> e1.getHireDate()
.compareTo(e2.getHireDate())).collect(Collectors.toList());
編輯:我沒有注意到您正在使用分頁。 我有一個類似的問題,我必須在返回Page
對象之前對頁面內容做一些事情,所以我最終創建了PageImpl<T>
類的擴展:
public class PageImplBean<T> extends PageImpl<T> {
private static final long serialVersionUID = 1L;
private int number;
private int size;
private int totalPages;
private int numberOfElements;
private long totalElements;
private boolean previousPage;
private boolean first;
private boolean nextPage;
private boolean last;
private List<T> content;
@JsonIgnore
private Sort sort;
public PageImplBean() {
super(new ArrayList<T>());
}
public PageImplBean(Page pagina){
super(new ArrayList<T>());
this.number = pagina.getNumber();
this.size = pagina.getSize();
this.totalPages = pagina.getTotalPages();
this.numberOfElements = pagina.getNumberOfElements();
this.totalElements = pagina.getTotalElements();
this.previousPage = pagina.hasPrevious();
this.first = pagina.isFirst();
this.nextPage = pagina.hasNext();
this.last = pagina.isLast();
}
public int getNumber() {
return number;
}
public void setNumber(int number) {
this.number = number;
}
public int getSize() {
return size;
}
public void setSize(int size) {
this.size = size;
}
public int getTotalPages() {
return totalPages;
}
public void setTotalPages(int totalPages) {
this.totalPages = totalPages;
}
public int getNumberOfElements() {
return numberOfElements;
}
public void setNumberOfElements(int numberOfElements) {
this.numberOfElements = numberOfElements;
}
public long getTotalElements() {
return totalElements;
}
public void setTotalElements(long totalElements) {
this.totalElements = totalElements;
}
public boolean isPreviousPage() {
return previousPage;
}
public void setPreviousPage(boolean previousPage) {
this.previousPage = previousPage;
}
public boolean isFirst() {
return first;
}
public void setFirst(boolean first) {
this.first = first;
}
public boolean isNextPage() {
return nextPage;
}
public void setNextPage(boolean nextPage) {
this.nextPage = nextPage;
}
public boolean isLast() {
return last;
}
public void setLast(boolean last) {
this.last = last;
}
public List<T> getContent() {
return content;
}
public void setContent(List<T> content) {
this.content = content;
}
public Sort getSort() {
return sort;
}
public void setSort(Sort sort) {
this.sort = sort;
}
public PageImpl<T> pageImpl() {
return new PageImpl<T>(getContent(), new PageRequest(getNumber(),
getSize(), getSort()), getTotalElements());
}
}
在您的情況下,重要的是基類中不存在的setContent
方法。 您可以將其與您的排序方法一起使用:
PageImplBean<Thing> page = //call your rest repository
List<Thing> pageContent = page.getContent();
page.setContent(sortByRating(pageContent));
您可以自定義您的對象頁面(Pageable)。 如果您獲得請求數據:
public PageRequest customSort(Args...) {
String[] newParams = {"newParam"};
Sort sort = page.getSort();
Sort.Order dataSort =
sort.iterator().next();
Sort newSort = Sort.by(dataSort.getDirection(), newParams);
return PageRequest.of(page.getPageNumber(), this.getPageSize(), newSort);
}
之前的 SQL:
order by oldParam ASC/DESC;
SQL之后:
order by newParam ASC/DESC;
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.