Spring Data Rest 将自定义端点添加到特定的存储库

use*_*401 6 java spring spring-data spring-data-rest

我想将自定义搜索端点添加到我现有的用户存储库。

我的用户存储库如下所示:

@RepositoryRestResource(collectionResourceRel="users", path="users")
public interface UserRepository extends PagingAndSortingRepository<User, Long>{

    User findByUsername(String username);
}
Run Code Online (Sandbox Code Playgroud)

自定义控制器:

@BasePathAwareController 
@RequestMapping("users/search")
public class CustomController implements ResourceProcessor<RepositorySearchesResource>, ResourceAssembler<User, Resource<User>> {
    @Autowired
    UserRepository userReposiotry;
    @Autowired
    private EntityLinks entityLinks;

    @RequestMapping(value = "findFirst", produces = "application/json")
    @ResponseBody
    public ResponseEntity<Resource<User>> findFirstUser() {
          Resource<User> resource = toResource(userReposiotry.findOne(1L));
          return new ResponseEntity<Resource<User>>(resource, HttpStatus.OK);
    }

    @Override
    public RepositorySearchesResource process(RepositorySearchesResource resource) {
        LinkBuilder lb = entityLinks.linkFor(User.class, "username");
        resource.add(new Link(lb.toString() + "/search/findFirst", "findFirst"));
        return resource;
    }

    @Override
    public Resource<User> toResource(User user) {
        Resource<User> resource = new Resource<User>(user);
        return resource;
    }
}
Run Code Online (Sandbox Code Playgroud)

这将为用户返回正确的搜索端点:

{
  "_links": {
    "findByUsername": {
      "href": "http://localhost:8080/api/users/search/findByUsername"
    },
    "self": {
      "href": "http://localhost:8080/api/users/search"
    },
    "findFirst": {
      "href": "http://localhost:8080/api/users/search/findFirst",
      "templated": true
    }
  }
}
Run Code Online (Sandbox Code Playgroud)

但也适用于其他端点,如邀请:

{
  "_links": {
    "findUserByInvite": {
      "href": "http://localhost:8080/api/invites/search/findUserByInvite"
    },
    "self": {
      "href": "http://localhost:8080/api/invites/search"
    },
    "findFirst": {
      "href": "http://localhost:8080/api/invites/search/findFirst",
      "templated": true
    }
  }
}
Run Code Online (Sandbox Code Playgroud)

这怎么能仅限于用户?谢谢

Mat*_*nkt 4

我假设您的邀请端点也返回一个RepositorySearchesResource?!每当ResourceProcessorspring-data-rest 序列化RepositorySearchesResource. 如果您想要为用户和邀请提供不同的链接,您有一些替代方案:

  • 为您的搜索端点使用不同的返回类型,以便您可以有不同的ResourceProcessor实现
  • 在您的内部添加更多逻辑,ResourceProcessor以区分您是否处于邀请或用户用例中