/* * Copyright (c) 2019 TAOS Data, Inc. * * This program is free software: you can use, redistribute, and/or modify * it under the terms of the GNU Affero General Public License, version 3 * or later ("AGPL"), as published by the Free Software Foundation. * * This program is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or * FITNESS FOR A PARTICULAR PURPOSE. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see . */ using System; using System.Collections.Generic; using System.Runtime.InteropServices; namespace TDengineDriver { public enum TDengineDataType { TSDB_DATA_TYPE_NULL = 0, // 1 bytes TSDB_DATA_TYPE_BOOL = 1, // 1 bytes TSDB_DATA_TYPE_TINYINT = 2, // 1 bytes TSDB_DATA_TYPE_SMALLINT = 3, // 2 bytes TSDB_DATA_TYPE_INT = 4, // 4 bytes TSDB_DATA_TYPE_BIGINT = 5, // 8 bytes TSDB_DATA_TYPE_FLOAT = 6, // 4 bytes TSDB_DATA_TYPE_DOUBLE = 7, // 8 bytes TSDB_DATA_TYPE_BINARY = 8, // string TSDB_DATA_TYPE_TIMESTAMP = 9,// 8 bytes TSDB_DATA_TYPE_NCHAR = 10, // unicode string TSDB_DATA_TYPE_UTINYINT = 11,// 1 byte TSDB_DATA_TYPE_USMALLINT = 12,// 2 bytes TSDB_DATA_TYPE_UINT = 13, // 4 bytes TSDB_DATA_TYPE_UBIGINT = 14 // 8 bytes } public enum TDengineInitOption { TSDB_OPTION_LOCALE = 0, TSDB_OPTION_CHARSET = 1, TSDB_OPTION_TIMEZONE = 2, TDDB_OPTION_CONFIGDIR = 3, TDDB_OPTION_SHELL_ACTIVITY_TIMER = 4 } enum TaosField { STRUCT_SIZE = 68, NAME_LENGTH = 65, TYPE_OFFSET = 65, BYTES_OFFSET = 66, } public class TDengineMeta { public string name; public short size; public byte type; public string TypeName() { switch ((TDengineDataType)type) { case TDengineDataType.TSDB_DATA_TYPE_BOOL: return "BOOL"; case TDengineDataType.TSDB_DATA_TYPE_TINYINT: return "TINYINT"; case TDengineDataType.TSDB_DATA_TYPE_SMALLINT: return "SMALLINT"; case TDengineDataType.TSDB_DATA_TYPE_INT: return "INT"; case TDengineDataType.TSDB_DATA_TYPE_BIGINT: return "BIGINT"; case TDengineDataType.TSDB_DATA_TYPE_UTINYINT: return "TINYINT UNSIGNED"; case TDengineDataType.TSDB_DATA_TYPE_USMALLINT: return "SMALLINT UNSIGNED"; case TDengineDataType.TSDB_DATA_TYPE_UINT: return "INT UNSIGNED"; case TDengineDataType.TSDB_DATA_TYPE_UBIGINT: return "BIGINT UNSIGNED"; case TDengineDataType.TSDB_DATA_TYPE_FLOAT: return "FLOAT"; case TDengineDataType.TSDB_DATA_TYPE_DOUBLE: return "DOUBLE"; case TDengineDataType.TSDB_DATA_TYPE_BINARY: return "STRING"; case TDengineDataType.TSDB_DATA_TYPE_TIMESTAMP: return "TIMESTAMP"; case TDengineDataType.TSDB_DATA_TYPE_NCHAR: return "NCHAR"; default: return "undefine"; } } } [StructLayout(LayoutKind.Sequential, CharSet = CharSet.Ansi)] public struct TAOS_BIND { // column type public int buffer_type; // one column value public IntPtr buffer; // unused public Int32 buffer_length; // actual value length in buffer public IntPtr length; // indicates the column value is null or not public IntPtr is_null; // unused public int is_unsigned; // unused public IntPtr error; public Int64 u; public uint allocated; } [StructLayout(LayoutKind.Sequential)] public struct TAOS_MULTI_BIND { // column type public int buffer_type; // array, one or more lines column value public IntPtr buffer; //length of element in TAOS_MULTI_BIND.buffer (for binary and nchar it is the longest element's length) public ulong buffer_length; //array, actual data length for each value public IntPtr length; //array, indicates each column value is null or not public IntPtr is_null; // line number, or the values number in buffer public int num; } public class TDengine { public const int TSDB_CODE_SUCCESS = 0; [DllImport("taos", EntryPoint = "taos_init", CallingConvention = CallingConvention.Cdecl)] static extern public void Init(); [DllImport("taos", EntryPoint = "taos_cleanup", CallingConvention = CallingConvention.Cdecl)] static extern public void Cleanup(); [DllImport("taos", EntryPoint = "taos_options", CallingConvention = CallingConvention.Cdecl)] static extern public void Options(int option, string value); [DllImport("taos", EntryPoint = "taos_connect", CallingConvention = CallingConvention.Cdecl)] static extern public IntPtr Connect(string ip, string user, string password, string db, short port); [DllImport("taos", EntryPoint = "taos_errstr", CallingConvention = CallingConvention.Cdecl)] static extern private IntPtr taos_errstr(IntPtr res); static public string Error(IntPtr res) { IntPtr errPtr = taos_errstr(res); return Marshal.PtrToStringAnsi(errPtr); } [DllImport("taos", EntryPoint = "taos_errno", CallingConvention = CallingConvention.Cdecl)] static extern public int ErrorNo(IntPtr res); [DllImport("taos", EntryPoint = "taos_query", CallingConvention = CallingConvention.Cdecl)] static extern public IntPtr Query(IntPtr conn, string sqlstr); [DllImport("taos", EntryPoint = "taos_affected_rows", CallingConvention = CallingConvention.Cdecl)] static extern public int AffectRows(IntPtr res); [DllImport("taos", EntryPoint = "taos_field_count", CallingConvention = CallingConvention.Cdecl)] static extern public int FieldCount(IntPtr res); [DllImport("taos", EntryPoint = "taos_fetch_fields", CallingConvention = CallingConvention.Cdecl)] static extern private IntPtr taos_fetch_fields(IntPtr res); static public List FetchFields(IntPtr res) { // const int fieldSize = 68; List metas = new List(); if (res == IntPtr.Zero) { return metas; } int fieldCount = FieldCount(res); IntPtr fieldsPtr = taos_fetch_fields(res); for (int i = 0; i < fieldCount; ++i) { int offset = i * (int)TaosField.STRUCT_SIZE; TDengineMeta meta = new TDengineMeta(); meta.name = Marshal.PtrToStringAnsi(fieldsPtr + offset); meta.type = Marshal.ReadByte(fieldsPtr + offset + (int)TaosField.TYPE_OFFSET); meta.size = Marshal.ReadInt16(fieldsPtr + offset + (int)TaosField.BYTES_OFFSET); metas.Add(meta); } return metas; } [DllImport("taos", EntryPoint = "taos_fetch_row", CallingConvention = CallingConvention.Cdecl)] static extern public IntPtr FetchRows(IntPtr res); [DllImport("taos", EntryPoint = "taos_free_result", CallingConvention = CallingConvention.Cdecl)] static extern public IntPtr FreeResult(IntPtr res); [DllImport("taos", EntryPoint = "taos_close", CallingConvention = CallingConvention.Cdecl)] static extern public int Close(IntPtr taos); //get precision of restultset [DllImport("taos", EntryPoint = "taos_result_precision", CallingConvention = CallingConvention.Cdecl)] static extern public int ResultPrecision(IntPtr taos); //stmt APIs: /// /// init a TAOS_STMT object for later use. /// /// a valid taos connection /// /// Not NULL returned for success, NULL for failure. And it should be freed with taos_stmt_close. /// [DllImport("taos", EntryPoint = "taos_stmt_init", CallingConvention = CallingConvention.Cdecl)] static extern public IntPtr StmtInit(IntPtr taos); /// /// prepare a sql statement，'sql' should be a valid INSERT/SELECT statement. /// /// could be the value returned by 'StmtInit', that may be a valid object or NULL. /// sql string,used to bind parameters with /// no used /// 0 for success, non-zero for failure. [DllImport("taos", EntryPoint = "taos_stmt_prepare", CallingConvention = CallingConvention.Cdecl)] static extern public int StmtPrepare(IntPtr stmt, string sql); /// /// For INSERT only. Used to bind table name as a parmeter for the input stmt object. /// /// could be the value returned by 'StmtInit', that may be a valid object or NULL. /// table name you want to bind /// 0 for success, non-zero for failure. [DllImport("taos", EntryPoint = "taos_stmt_set_tbname", CallingConvention = CallingConvention.Cdecl)] static extern public int StmtSetTbname(IntPtr stmt, string name); /// /// For INSERT only. /// Set a table name for binding table name as parameter. Only used for binding all tables /// in one stable, user application must call 'loadTableInfo' API to load all table /// meta before calling this API. If the table meta is not cached locally, it will return error. /// /// could be the value returned by 'StmtInit', that may be a valid object or NULL. /// table name which is belong to an stable /// 0 for success, non-zero for failure. [DllImport("taos", EntryPoint = "taos_stmt_set_sub_tbname", CallingConvention = CallingConvention.Cdecl)] static extern public int StmtSetSubTbname(IntPtr stmt, string name); /// /// For INSERT only. /// set a table name for binding table name as parameter and tag values for all tag parameters. /// /// could be the value returned by 'StmtInit', that may be a valid object or NULL. /// use to set table name /// /// is an array contains all tag values,each item in the array represents a tag column's value. /// the item number and sequence should keep consistence with that in stable tag definition. /// /// 0 for success, non-zero for failure. [DllImport("taos", EntryPoint = "taos_stmt_set_tbname_tags", CallingConvention = CallingConvention.Cdecl)] static extern public int StmtSetTbnameTags(IntPtr stmt, string name, TAOS_BIND[] tags); /// /// For both INSERT and SELECT. /// bind a whole line data. /// The usage of structure TAOS_BIND is the same with MYSQL_BIND in MySQL. /// /// could be the value returned by 'StmtInit', that may be a valid object or NULL. /// /// points to an array contains the whole line data. /// the item number and sequence should keep consistence with columns in sql statement. /// /// 0 for success, non-zero for failure. [DllImport("taos", EntryPoint = "taos_stmt_bind_param", CallingConvention = CallingConvention.Cdecl, SetLastError = true)] static extern public int StmtBindParam(IntPtr stmt, TAOS_BIND[] bind); /// /// bind a single column's data, INTERNAL used and for INSERT only. /// /// could be the value returned by 'StmtInit', that may be a valid object or NULL. /// points to a column's data which could be the one or more lines. /// the column's index in prepared sql statement, it starts from 0. /// 0 for success, non-zero for failure. [DllImport("taos", EntryPoint = "taos_stmt_bind_single_param_batch", CallingConvention = CallingConvention.Cdecl)] static extern public int StmtBindSingleParamBatch(IntPtr stmt, ref TAOS_MULTI_BIND bind, int colIdx); /// /// for INSERT only /// bind one or multiple lines data. The parameter 'bind' /// /// could be the value returned by 'StmtInit', that may be a valid object or NULL. /// /// points to an array contains one or more lines data.Each item in array represents a column's value(s), /// the item number and sequence should keep consistence with columns in sql statement. /// /// 0 for success, non-zero for failure. [DllImport("taos", EntryPoint = "taos_stmt_bind_param_batch", CallingConvention = CallingConvention.Cdecl)] static extern public int StmtBindParamBatch(IntPtr stmt, [In, Out] TAOS_MULTI_BIND[] bind); /// /// For INSERT only. /// add all current bound parameters to batch process. Must be called after each call to /// StmtBindParam/StmtBindSingleParamBatch, or all columns binds for one or more lines /// with StmtBindSingleParamBatch. User application can call any bind parameter /// API again to bind more data lines after calling to this API. /// /// could be the value returned by 'StmtInit', that may be a valid object or NULL. /// 0 for success, non-zero for failure. [DllImport("taos", EntryPoint = "taos_stmt_add_batch", CallingConvention = CallingConvention.Cdecl)] static extern public int StmtAddBatch(IntPtr stmt); /// /// actually execute the INSERT/SELECT sql statement. /// User application can continue to bind new data after calling to this API. /// /// could be the value returned by 'StmtInit', that may be a valid object or NULL. /// [DllImport("taos", EntryPoint = "taos_stmt_execute", CallingConvention = CallingConvention.Cdecl)] static extern public int StmtExecute(IntPtr stmt); /// /// For SELECT only,getting the query result. User application should free it with API 'FreeResult' at the end. /// /// could be the value returned by 'StmtInit', that may be a valid object or NULL. /// Not NULL for success, NULL for failure. [DllImport("taos", EntryPoint = "taos_stmt_use_result", CallingConvention = CallingConvention.Cdecl)] static extern public IntPtr StmtUseResult(IntPtr stmt); /// /// close STMT object and free resources. /// /// could be the value returned by 'StmtInit', that may be a valid object or NULL. /// 0 for success, non-zero for failure. [DllImport("taos", EntryPoint = "taos_stmt_close", CallingConvention = CallingConvention.Cdecl)] static extern public int StmtClose(IntPtr stmt); [DllImport("taos", EntryPoint = "taos_load_table_info", CallingConvention = CallingConvention.Cdecl)] /// /// user application must call this API to load all tables meta, /// /// taos connection /// tablelist /// static extern private int LoadTableInfoDll(IntPtr taos, string tableList); /// /// user application call this API to load all tables meta,this method call the native /// method LoadTableInfoDll. /// this method must be called before StmtSetSubTbname(IntPtr stmt, string name); /// /// taos connection /// tables need to load meta info are form in an array /// static public int LoadTableInfo(IntPtr taos, string[] tableList) { string listStr = string.Join(",", tableList); return LoadTableInfoDll(taos, listStr); } /// /// get detail error message when got failure for any stmt API call. If not failure, the result /// returned in this API is unknown. /// /// could be the value returned by 'StmtInit', that may be a valid object or NULL. /// piont the error message [DllImport("taos", EntryPoint = "taos_stmt_errstr", CallingConvention = CallingConvention.Cdecl)] static extern private IntPtr StmtErrPtr(IntPtr stmt); /// /// get detail error message when got failure for any stmt API call. If not failure, the result /// returned in this API is unknown. /// /// could be the value returned by 'StmtInit', that may be a valid object or NULL. /// error string static public string StmtErrorStr(IntPtr stmt) { IntPtr stmtErrPrt = StmtErrPtr(stmt); return Marshal.PtrToStringAnsi(stmtErrPrt); } } }