Expose column names in row struct

mysql-d · Dec 15, 2018 · bb272f0 · bb272f0
1 parent d28a180
commit bb272f0
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/source/mysql/protocol/comms.d b/source/mysql/protocol/comms.d
@@ -636,7 +636,7 @@ body
 
 // Moved here from `struct Row.this`
 package(mysql) void ctorRow(Connection conn, ref ubyte[] packet, ResultSetHeaders rh, bool binary,
-	out Variant[] _values, out bool[] _nulls)
+	out Variant[] _values, out bool[] _nulls, out string[] _names)
 in
 {
 	assert(rh.fieldCount <= uint.max);
@@ -646,7 +646,7 @@ body
 	scope(failure) conn.kill();
 
 	uint fieldCount = cast(uint)rh.fieldCount;
-	_values.length = _nulls.length = fieldCount;
+	_values.length = _nulls.length = _names.length = fieldCount;
 
 	if(binary)
 	{
@@ -669,6 +669,7 @@ body
 		do
 		{
 			FieldDescription fd = rh[i];
+			_names[i] = fd.name;
 			sqlValue = packet.consumeIfComplete(fd.type, binary, fd.unsigned, fd.charSet);
 			// TODO: Support chunk delegate
 			if(sqlValue.isIncomplete)

diff --git a/source/mysql/result.d b/source/mysql/result.d
@@ -35,6 +35,7 @@ package:
 	Variant[]   _values; // Temporarily "package" instead of "private"
 private:
 	bool[]      _nulls;
+	string[]    _names;
 
 public:
 
@@ -51,7 +52,7 @@ public:
 	+/
 	this(Connection con, ref ubyte[] packet, ResultSetHeaders rh, bool binary)
 	{
-		ctorRow(con, packet, rh, binary, _values, _nulls);
+		ctorRow(con, packet, rh, binary, _values, _nulls, _names);
 	}
 
 	/++
@@ -72,6 +73,14 @@ public:
 		return _values[i];
 	}
 
+	/++
+	Get the name of the column with specified index.
+	+/
+	string getName(size_t index)
+	{
+		return _names[index];
+	}
+
 	/++
 	Check if a column in the result row was NULL