diff options
Diffstat (limited to 'database/sqlite/sqlite_functions.h')
-rw-r--r-- | database/sqlite/sqlite_functions.h | 83 |
1 files changed, 83 insertions, 0 deletions
diff --git a/database/sqlite/sqlite_functions.h b/database/sqlite/sqlite_functions.h new file mode 100644 index 00000000..40b5af46 --- /dev/null +++ b/database/sqlite/sqlite_functions.h @@ -0,0 +1,83 @@ +// SPDX-License-Identifier: GPL-3.0-or-later + +#ifndef NETDATA_SQLITE_FUNCTIONS_H +#define NETDATA_SQLITE_FUNCTIONS_H + +#include "daemon/common.h" +#include "sqlite3.h" + +void analytics_set_data_str(char **name, const char *value); + +// return a node list +struct node_instance_list { + uuid_t node_id; + uuid_t host_id; + char *hostname; + int live; + int queryable; + int hops; +}; + +typedef enum db_check_action_type { + DB_CHECK_NONE = (1 << 0), + DB_CHECK_RECLAIM_SPACE = (1 << 1), + DB_CHECK_ANALYZE = (1 << 2), + DB_CHECK_CONT = (1 << 3), + DB_CHECK_RECOVER = (1 << 4), +} db_check_action_type_t; + +#define SQL_MAX_RETRY (100) +#define SQLITE_INSERT_DELAY (10) // Insert delay in case of lock + +#define CHECK_SQLITE_CONNECTION(db_meta) \ + if (unlikely(!db_meta)) { \ + if (default_rrd_memory_mode != RRD_MEMORY_MODE_DBENGINE) { \ + return 1; \ + } \ + error_report("Database has not been initialized"); \ + return 1; \ + } + +SQLITE_API int sqlite3_step_monitored(sqlite3_stmt *stmt); +SQLITE_API int sqlite3_exec_monitored( + sqlite3 *db, /* An open database */ + const char *sql, /* SQL to be evaluated */ + int (*callback)(void*,int,char**,char**), /* Callback function */ + void *data, /* 1st argument to callback */ + char **errmsg /* Error msg written here */ + ); + +// Initialization and shutdown +int init_database_batch(sqlite3 *database, const char *batch[]); +int sql_init_database(db_check_action_type_t rebuild, int memory); +void sql_close_database(void); +int configure_sqlite_database(sqlite3 *database, int target_version); + +// Helpers +int bind_text_null(sqlite3_stmt *res, int position, const char *text, bool can_be_null); +int prepare_statement(sqlite3 *database, const char *query, sqlite3_stmt **statement); +int execute_insert(sqlite3_stmt *res); +int exec_statement_with_uuid(const char *sql, uuid_t *uuid); +int db_execute(sqlite3 *database, const char *cmd); +void initialize_thread_key_pool(void); + +// Look up functions +int get_node_id(uuid_t *host_id, uuid_t *node_id); +struct node_instance_list *get_node_list(void); +void sql_load_node_id(RRDHOST *host); + +// Help build archived hosts in memory when agent starts +void sql_build_host_system_info(uuid_t *host_id, struct rrdhost_system_info *system_info); +RRDLABELS *sql_load_host_labels(uuid_t *host_id); + +// TODO: move to metadata +int update_node_id(uuid_t *host_id, uuid_t *node_id); + +void invalidate_node_instances(uuid_t *host_id, uuid_t *claim_id); + +// Provide statistics +int sql_metadata_cache_stats(int op); + +void sql_drop_table(const char *table); +void sqlite_now_usec(sqlite3_context *context, int argc, sqlite3_value **argv); +#endif //NETDATA_SQLITE_FUNCTIONS_H |