adapted to IoTDBDataReader

Author: fpddmw

src/Apache.IoTDB.Data/IoTDBDataReader.cs

@@ -45,7 +45,7 @@ public class IoTDBDataReader : DbDataReader
         private bool _closeConnection;
 
         private int _fieldCount;
-
+        
         RowRecord rowdata = null;
 
 
@@ -56,6 +56,9 @@ internal IoTDBDataReader(IoTDBCommand IoTDBCommand, SessionDataSet dataSet, bool
             _command = IoTDBCommand;
             _closeConnection = closeConnection;
             _fieldCount = dataSet.GetColumnNames().Count;
+            _hasRows = dataSet.RowCount() > 0;
+            _recordsAffected = dataSet.RowCount();
+
             _closed = _closeConnection;
             _metas = dataSet.GetColumnNames();
             _dataSet = dataSet;
@@ -134,7 +137,7 @@ public override bool Read()
             {
                 throw new InvalidOperationException($"DataReaderClosed{nameof(Read)}");
             }
-            if (_dataSet.HasNext())
+            if (_dataSet.Next())
             {
                 rowdata = _dataSet.GetRow();
             }
@@ -452,7 +455,7 @@ public override int GetValues(object[] values)
         /// <returns>A System.Data.DataTable that describes the column metadata.</returns>
         public override DataTable GetSchemaTable()
         {
-            if (_dataSet.HasNext())
+            if (_dataSet.Next())
             {
                 rowdata = _dataSet.GetRow();
             }

src/Apache.IoTDB/DataStructure/RpcDataSet.cs

@@ -38,7 +38,7 @@ public class RpcDataSet : System.IDisposable
         private TsBlock _curTsBlock;
         private int _queryResultSize;
         private int _queryResultIndex;
-        private int _tsBlockSize;
+        public int _tsBlockSize;
         private int _tsBlockIndex;
         private TimeZoneInfo _zoneId;
         private int _timeFactor;

src/Apache.IoTDB/DataStructure/SessionDataSet.cs

@@ -120,6 +120,7 @@ public bool HasNext()
         public IReadOnlyList<string> GetColumnTypes() => _rpcDataSet._columnTypeList;
 
         public RowRecord GetRow() => _rpcDataSet.GetRow();
+        public int RowCount() => _rpcDataSet._tsBlockSize;
 
         public void ShowTableNames()
         {