add size to the result of .INTO

Author: siscia

redisql_lib/src/redis.rs

@@ -697,7 +697,7 @@ pub fn stream_query_result_array(
 ) -> Result<QueryResult, err::RediSQLError> {
     let row_last_index = array.len() - 1;
 
-    let mut result = Vec::with_capacity(3);
+    let mut result = Vec::with_capacity(4);
     result.push(sql::Entity::Text {
         text: stream_name.to_string(),
     });
@@ -780,17 +780,22 @@ pub fn stream_query_result_array(
             }
         }
     }
+    context.release(lock);
+
     if result.len() == 2 {
         let start_and_end = result[1].clone();
         result.push(start_and_end);
     }
+    result.push(sql::Entity::Integer {
+        int: array.len() as i64,
+    });
 
-    context.release(lock);
     Ok(QueryResult::Array {
         names: vec![
             String::from("stream"),
             String::from("first_id"),
             String::from("last_id"),
+            String::from("size"),
         ],
         array: vec![result],
     })

test/correctness/test.py

@@ -652,6 +652,7 @@ def test_stream_query(self):
 
             result = self.exec_naked("REDISQL.QUERY.INTO", "{A}:1", "A", "SELECT * FROM foo")
             self.assertEquals(result[0][0], "{A}:1")
+            self.assertEquals(result[0][3], 513)
 
             result = self.exec_naked("XRANGE", "{A}:1", "-", "+")
             self.assertEquals(len(result), total_len)
@@ -676,6 +677,7 @@ def test_stream_query_statement(self):
             result = self.exec_naked("REDISQL.QUERY_STATEMENT.INTO",
                     "{B}:1", "B", "select_all")
             self.assertEquals(result[0][0], "{B}:1")
+            self.assertEquals(result[0][3], 513)
 
             result = self.exec_naked("XRANGE", "{B}:1", "-", "+")
             self.assertEquals(len(result), total_len)
@@ -697,6 +699,7 @@ def test_stream_query(self):
 
             result = self.exec_naked("REDISQL.QUERY.INTO.NOW", "{A}:1", "A", "SELECT * FROM foo")
             self.assertEquals(result[0][0], "{A}:1")
+            self.assertEquals(result[0][3], 513)
 
             result = self.exec_naked("XRANGE", "{A}:1", "-", "+")
             self.assertEquals(len(result), total_len)
@@ -721,6 +724,7 @@ def test_stream_query_statement(self):
             result = self.exec_naked("REDISQL.QUERY_STATEMENT.INTO.NOW",
                     "{B}:1", "B", "select_all")
             self.assertEquals(result[0][0], "{B}:1")
+            self.assertEquals(result[0][3], 513)
 
             result = self.exec_naked("XRANGE", "{B}:1", "-", "+")
             self.assertEquals(len(result), total_len)