MF-1061 - Fix cassandra-reader count for json format (absmach#1327)

Signed-off-by: Manuel Imperiale <manuel.imperiale@gmail.com>
fbugarski · Mar 8, 2021 · 26bb8d7 · 26bb8d7
1 parent 2b7fe6f
commit 26bb8d7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/readers/cassandra/messages.go b/readers/cassandra/messages.go
@@ -50,13 +50,14 @@ func (cr cassandraRepository) ReadAll(chanID string, offset, limit uint64, query
 		value, string_value, bool_value, data_value, sum, time,
 		update_time FROM messages WHERE channel = ? %s LIMIT ?
 		ALLOW FILTERING`, q)
+	countCQL := fmt.Sprintf(`SELECT COUNT(*) FROM %s WHERE channel = ? %s ALLOW FILTERING`, defTable, q)
+
 	if table != defTable {
 		selectCQL = fmt.Sprintf(`SELECT channel, subtopic, publisher, protocol, created, payload FROM %s WHERE channel = ? %s LIMIT ?
 			ALLOW FILTERING`, table, q)
+		countCQL = fmt.Sprintf(`SELECT COUNT(*) FROM %s WHERE channel = ? %s ALLOW FILTERING`, table, q)
 	}
 
-	countCQL := fmt.Sprintf(`SELECT COUNT(*) FROM %s WHERE channel = ? %s ALLOW FILTERING`, defTable, q)
-
 	iter := cr.session.Query(selectCQL, vals...).Iter()
 	defer iter.Close()
 	scanner := iter.Scanner()