| using System; |
| using System.Collections.Generic; |
| using System.Threading; |
| using System.Threading.Tasks; |
| using Apache.IoTDB.DataStructure; |
| |
| namespace Apache.IoTDB.Samples |
| { |
| public partial class SessionPoolTest |
| { |
| public async Task TestInsertTablet() |
| { |
| var session_pool = new SessionPool(host, port, pool_size); |
| var status = 0; |
| await session_pool.Open(false); |
| if (debug) session_pool.OpenDebugMode(); |
| |
| System.Diagnostics.Debug.Assert(session_pool.IsOpen()); |
| await session_pool.DeleteStorageGroupAsync(test_group_name); |
| var device_id = string.Format("{0}.{1}", test_group_name, test_device); |
| var measurement_lst = new List<string> |
| { |
| test_measurements[1], |
| test_measurements[2], |
| test_measurements[3] |
| }; |
| var value_lst = new List<List<object>> |
| { |
| new() {"iotdb", true, (int) 12}, new() {"c#", false, (int) 13}, |
| new() {"client", true, (int) 14} |
| }; |
| var timestamp_lst = new List<long> { 1, 2, 3 }; |
| var datatype_lst = new List<TSDataType> { TSDataType.TEXT, TSDataType.BOOLEAN, TSDataType.INT32 }; |
| var tablet = new Tablet(device_id, measurement_lst, datatype_lst, value_lst, timestamp_lst); |
| status = await session_pool.InsertTabletAsync(tablet); |
| System.Diagnostics.Debug.Assert(status == 0); |
| var res = await session_pool.ExecuteQueryStatementAsync( |
| "select * from " + string.Format("{0}.{1}", test_group_name, test_device) + " where time<15"); |
| res.ShowTableNames(); |
| while (res.HasNext()) Console.WriteLine(res.Next()); |
| |
| await res.Close(); |
| // large data test |
| value_lst = new List<List<object>>() { }; |
| timestamp_lst = new List<long>() { }; |
| var tasks = new List<Task<int>>(); |
| var start_ms = DateTime.Now.Ticks / 10000; |
| for (var timestamp = 4; timestamp <= fetch_size * processed_size; timestamp++) |
| { |
| timestamp_lst.Add(timestamp); |
| value_lst.Add(new List<object>() { "iotdb", true, (int)timestamp }); |
| if (timestamp % fetch_size == 0) |
| { |
| tablet = new Tablet(device_id, measurement_lst, datatype_lst, value_lst, timestamp_lst); |
| tasks.Add(session_pool.InsertTabletAsync(tablet)); |
| value_lst = new List<List<object>>() { }; |
| timestamp_lst = new List<long>() { }; |
| } |
| } |
| |
| Task.WaitAll(tasks.ToArray()); |
| var end_ms = DateTime.Now.Ticks / 10000; |
| Console.WriteLine(string.Format("total tablet insert time is {0}", end_ms - start_ms)); |
| res = await session_pool.ExecuteQueryStatementAsync( |
| "select * from " + string.Format("{0}.{1}", test_group_name, test_device)); |
| res.ShowTableNames(); |
| var res_count = 0; |
| while (res.HasNext()) |
| { |
| res.Next(); |
| res_count += 1; |
| } |
| |
| await res.Close(); |
| Console.WriteLine(res_count + " " + fetch_size * processed_size); |
| System.Diagnostics.Debug.Assert(res_count == fetch_size * processed_size); |
| status = await session_pool.DeleteStorageGroupAsync(test_group_name); |
| System.Diagnostics.Debug.Assert(status == 0); |
| await session_pool.Close(); |
| Console.WriteLine("TestInsertTablet Passed!"); |
| } |
| |
| public async Task TestInsertTablets() |
| { |
| var session_pool = new SessionPool(host, port, pool_size); |
| var status = 0; |
| await session_pool.Open(false); |
| if (debug) session_pool.OpenDebugMode(); |
| |
| System.Diagnostics.Debug.Assert(session_pool.IsOpen()); |
| await session_pool.DeleteStorageGroupAsync(test_group_name); |
| var device_id = new List<string>() |
| { |
| string.Format("{0}.{1}", test_group_name, test_devices[1]), |
| string.Format("{0}.{1}", test_group_name, test_devices[2]) |
| }; |
| var measurements_lst = new List<List<string>>() |
| { |
| new() {test_measurements[1], test_measurements[2], test_measurements[3]}, |
| new() {test_measurements[1], test_measurements[2], test_measurements[3]} |
| }; |
| var values_lst = new List<List<List<object>>>() |
| { |
| new() |
| { |
| new List<object>() {"iotdb", true, (int) 12}, new List<object>() {"c#", false, (int) 13}, |
| new List<object>() {"client", true, (int) 14} |
| }, |
| new() |
| { |
| new List<object>() {"iotdb_2", true, (int) 1}, new List<object>() {"c#_2", false, (int) 2}, |
| new List<object>() {"client_2", true, (int) 3} |
| } |
| }; |
| var datatype_lst = new List<List<TSDataType>>() |
| { |
| new() {TSDataType.TEXT, TSDataType.BOOLEAN, TSDataType.INT32}, |
| new() {TSDataType.TEXT, TSDataType.BOOLEAN, TSDataType.INT32} |
| }; |
| var timestamp_lst = new List<List<long>>() |
| {new() {2, 1, 3}, new() {3, 1, 2}}; |
| var tablets = new List<Tablet>() { }; |
| for (var i = 0; i < device_id.Count; i++) |
| { |
| var tablet = new Tablet(device_id[i], measurements_lst[i], datatype_lst[i], values_lst[i], timestamp_lst[i]); |
| tablets.Add(tablet); |
| } |
| |
| status = await session_pool.InsertTabletsAsync(tablets); |
| // System.Diagnostics.Debug.Assert(status == 0); |
| var res = await session_pool.ExecuteQueryStatementAsync( |
| "select * from " + string.Format("{0}.{1}", test_group_name, test_devices[1]) + " where time<15"); |
| res.ShowTableNames(); |
| while (res.HasNext()) Console.WriteLine(res.Next()); |
| |
| // large data test |
| |
| var tasks = new List<Task<int>>(); |
| // tablets = new List<Tablet>(); |
| for (var timestamp = 4; timestamp <= processed_size * fetch_size; timestamp++) |
| { |
| var local_device_id = string.Format("{0}.{1}", test_group_name, test_devices[1]); |
| var local_measurements = new List<string>() |
| {test_measurements[1], test_measurements[2], test_measurements[3]}; |
| var local_value = new List<List<object>>() { new() { "iotdb", true, (int)timestamp } }; |
| var local_timestamp = new List<long> { timestamp }; |
| var local_data_type = new List<TSDataType> { TSDataType.TEXT, TSDataType.BOOLEAN, TSDataType.INT32 }; |
| var tablet = new Tablet(local_device_id, local_measurements, local_data_type, local_value, local_timestamp); |
| tablets.Add(tablet); |
| if (timestamp % fetch_size == 0) |
| { |
| tasks.Add(session_pool.InsertTabletsAsync(tablets)); |
| tablets = new List<Tablet>() { }; |
| } |
| } |
| |
| Task.WaitAll(tasks.ToArray()); |
| res = await session_pool.ExecuteQueryStatementAsync( |
| "select * from " + string.Format("{0}.{1}", test_group_name, test_devices[1])); |
| res.ShowTableNames(); |
| var res_count = 0; |
| while (res.HasNext()) |
| { |
| res.Next(); |
| res_count += 1; |
| } |
| |
| await res.Close(); |
| Console.WriteLine(res_count + " " + fetch_size * processed_size); |
| System.Diagnostics.Debug.Assert(res_count == fetch_size * processed_size); |
| status = await session_pool.DeleteStorageGroupAsync(test_group_name); |
| System.Diagnostics.Debug.Assert(status == 0); |
| await session_pool.Close(); |
| Console.WriteLine("TestInsertTablets Passed!"); |
| } |
| public async Task TestInsertTabletWithNullValue() |
| { |
| var session_pool = new SessionPool(host, port, pool_size); |
| var status = 0; |
| await session_pool.Open(false); |
| if (debug) session_pool.OpenDebugMode(); |
| |
| System.Diagnostics.Debug.Assert(session_pool.IsOpen()); |
| await session_pool.DeleteStorageGroupAsync(test_group_name); |
| var device_id = string.Format("{0}.{1}", test_group_name, test_device); |
| var measurements = new List<string>() { test_measurements[1], test_measurements[2], test_measurements[3] }; |
| var values = new List<List<object>>() |
| { |
| new List<object>() {null, true, (int) 12}, new List<object>() {"c#", null, (int) 13}, |
| new List<object>() {"client", true, null} |
| }; |
| var datatype = new List<TSDataType>() { TSDataType.TEXT, TSDataType.BOOLEAN, TSDataType.INT32 }; |
| var timestamp = new List<long>() { 2, 1, 3 }; |
| var tablet = new Tablet(device_id, measurements, datatype, values, timestamp); |
| status = await session_pool.InsertTabletAsync(tablet); |
| System.Diagnostics.Debug.Assert(status == 0); |
| var res = await session_pool.ExecuteQueryStatementAsync( |
| "select * from " + string.Format("{0}.{1}", test_group_name, test_device)); |
| res.ShowTableNames(); |
| var res_count = 0; |
| while (res.HasNext()) |
| { |
| Console.WriteLine(res.Next()); |
| res_count += 1; |
| } |
| |
| await res.Close(); |
| status = await session_pool.DeleteStorageGroupAsync(test_group_name); |
| System.Diagnostics.Debug.Assert(status == 0); |
| await session_pool.Close(); |
| Console.WriteLine("TestInsertTabletsWithNullValue Passed!"); |
| } |
| } |
| } |