implemented get_autoincrement_value

Author: Reggie Burnett

lib/mysqlx/protocol.py

@@ -35,7 +35,7 @@
 from .protobuf import mysqlx_expr_pb2 as MySQLxExpr
 from .result import ColumnMetaData
 from .dbdoc import DbDoc
-from .expr import (Expr, ExprParser, Find, UpdateOperation, build_int_scalar,
+from .expr import (Expr, ExprParser, Find, UpdateOperation, build_null_scalar, build_int_scalar,
                    build_string_scalar, build_bool_scalar, build_double_scalar)
 from .errors import InterfaceError, OperationalError, ProgrammingError
 
@@ -260,8 +260,11 @@ def _process_frame(self, msg, rs):
             sessStateMsg = MySQLxNotice.SessionStateChanged()
             sessStateMsg.ParseFromString(msg.payload)
             if sessStateMsg.param == \
-               MySQLxNotice.SessionStateChanged.ROWS_AFFECTED:
+                    MySQLxNotice.SessionStateChanged.ROWS_AFFECTED:
                 rs._rows_affected = sessStateMsg.value.v_unsigned_int
+            elif sessStateMsg.param == \
+                    MySQLxNotice.SessionStateChanged.GENERATED_INSERT_ID:
+                rs._generated_id = sessStateMsg.value.v_unsigned_int
 
     def _read_message(self, rs):
         while True:
@@ -302,7 +305,8 @@ def get_column_metadata(self, rs):
 
     def arg_object_to_expr(self, value, allow_relational):
         if value is None:
-            return Expr.build_null_scalar()
+            return MySQLxExpr.Expr(type=MySQLxExpr.Expr.LITERAL,
+                                   literal=build_null_scalar())
         if isinstance(value, bool):
             return MySQLxExpr.Expr(type=MySQLxExpr.Expr.LITERAL,
                                    literal=build_bool_scalar(value))

lib/mysqlx/result.py

@@ -518,6 +518,7 @@ def __init__(self, connection):
         self._protocol = self._connection.protocol
         self._closed = False
         self._rows_affected = 0
+        self._generated_id = -1
         self._warnings = []
         if connection._active_result is not None:
             connection._active_result.fetch_all()
@@ -540,9 +541,8 @@ def __init__(self, connection):
     def get_affected_items_count(self):
         return self._rows_affected
 
-    @property
     def get_autoincrement_value(self):
-        pass
+        return self._generated_id
 
 
 class BufferingResult(BaseResult):
@@ -618,6 +618,9 @@ def __init__(self, connection):
         super(SqlResult, self).__init__(connection)
         self._has_more_results = False
 
+    def get_autoincrement_value(self):
+        return self._generated_id
+
     def next_result(self):
         if self._closed:
             return False

tests/test_mysqlx_crud.py

@@ -479,6 +479,17 @@ def test_multiple_resultsets(self):
         self.assertEqual(False, result.next_result())
 
 
+    def test_auto_inc_value(self):
+        table_name = "{0}.test".format(self.schema_name)
+
+        self.node_session.sql(
+            "CREATE TABLE {0}(id INT KEY AUTO_INCREMENT, name VARCHAR(50))".format(table_name)).execute()
+        result = self.node_session.sql("INSERT INTO {0} VALUES (NULL, 'Fred')".format(table_name)).execute()
+        self.assertEqual(1, result.get_autoincrement_value())
+        table = self.schema.get_table("test")
+        result2 = table.insert("id", "name").values(None, "Boo").execute()
+        self.assertEqual(2, result2.get_autoincrement_value())
+
     def test_column_metadata(self):
         table_name = "{0}.test".format(self.schema_name)