Bug Fix

islax · Jul 26, 2019 · 6b471a4 · 6b471a4
1 parent 51d4490
commit 6b471a4
Showing 1 changed file with 6 additions and 3 deletions.
diff --git 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
 	}
 }