diff --git a/repository/Repository.go b/repository/Repository.go index 1bd7707..fe6a44e 100644 --- a/repository/Repository.go +++ b/repository/Repository.go @@ -119,10 +119,13 @@ func PaginateForWeb(w http.ResponseWriter, r *http.Request) QueryProcessor { db = db.Offset(offset) } - var totalRecords int - db.Model(out).Count(&totalRecords) + if out != nil { + var totalRecords int + db.Model(out).Count(&totalRecords) + + w.Header().Set("X-Total-Count", strconv.Itoa(totalRecords)) + } - w.Header().Set("X-Total-Count", strconv.Itoa(totalRecords)) return db, nil } }