使用 Spring 数据 REST 设置 X-Total-Count

Question

我正在测试 Spring 数据 REST。我一直在关注 this tutorial 以了解其工作原理。

到目前为止的代码很简单：

@Entity
@SequenceGenerator(name="my_seq", sequenceName="user_id_seq")
@Table(name = "employees")
data class Employee (@Id @GeneratedValue(strategy = GenerationType.SEQUENCE, generator = "my_seq") val id: Long? = null,
                     val name: String = "defConstructorHell")

当我用 GET 询问时，我得到以下 json:

{
  "_embedded" : {
    "employees" : [ {
      "name" : "ciao",
      "_links" : {
        "self" : {
          "href" : "http://localhost:5000/api/employees/1"
        },
        "employee" : {
          "href" : "http://localhost:5000/api/employees/1"
        }
      }
    } ]
  },
  "_links" : {
    "self" : {
      "href" : "http://localhost:5000/api/employees{?page,size,sort}",
      "templated" : true
    },
    "profile" : {
      "href" : "http://localhost:5000/api/profile/employees"
    }
  },
  "page" : {
    "size" : 20,
    "totalElements" : 1,
    "totalPages" : 1,
    "number" : 0
  }
}

完全没问题，但我使用的 js 框架需要 header X-Total-Count.

您知道是否有办法通过 Spring Data REST 实现它？

Answer 1

最后我添加了一个控制器来添加所请求的 header:

@RepositoryRestController
class EmployeeController @Autowired constructor(val repo: EmployeeRepository) {

    @RequestMapping(method = arrayOf(GET),
            path = arrayOf("employees"))
    @ResponseBody
    fun getEmployees(@RequestParam("_sort", required = false, defaultValue = "id") _sort: String,
                     @RequestParam("_order", required = false, defaultValue = "DESC") _order: String,
                     @RequestParam("_start", required = false, defaultValue = "0") _start: Int,
                     @RequestParam("_end", required = false, defaultValue = "20") _end: Int): ResponseEntity<MutableIterable<Employee>> {
        val pr = PageRequest(_start, 20, Sort.Direction.valueOf(_order), _sort)
        val result = repo.findAll(pr)
        val headers = HttpHeaders()
        headers.add("X-Total-Count", result.count().toString())
        return ResponseEntity(result.content, headers, OK)
    }
}

使用 Spring 数据 REST 设置 X-Total-Count

Setup X-Total-Count with Spring Data REST

rest

spring

spring-data-rest