mirror of
https://github.com/Cacti/spine.git
synced 2026-11-09 14:05:14 +09:00
Larry Adams
ec5f681f65
git-svn-id: svn+ssh://repo.cacti.net/var/data/svnroot/cacti/branches/BRANCH_0_8_6/cactid_unix@2747 860744bd-22fc-0310-8c96-e9fe5004b5ca
113 lines
3.7 KiB
C
113 lines
3.7 KiB
C
/*
|
|
+-------------------------------------------------------------------------+
|
|
| Copyright (C) 2004 Ian Berry |
|
|
| |
|
|
| This program is free software; you can redistribute it and/or |
|
|
| modify it under the terms of the GNU General Public License |
|
|
| as published by the Free Software Foundation; either version 2 |
|
|
| of the License, or (at your option) any later version. |
|
|
| |
|
|
| 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. See the |
|
|
| GNU General Public License for more details. |
|
|
+-------------------------------------------------------------------------+
|
|
| cactid: a backend data gatherer for cacti |
|
|
+-------------------------------------------------------------------------+
|
|
| This poller would not have been possible without: |
|
|
| - Rivo Nurges (rrd support, mysql poller cache, misc functions) |
|
|
| - RTG (core poller code, pthreads, snmp, autoconf examples) |
|
|
| - Brady Alleman/Doug Warner (threading ideas, implimentation details) |
|
|
+-------------------------------------------------------------------------+
|
|
| - raXnet - http://www.raxnet.net/ |
|
|
+-------------------------------------------------------------------------+
|
|
*/
|
|
|
|
#include "common.h"
|
|
#include "cactid.h"
|
|
#include "locks.h"
|
|
#include "util.h"
|
|
#include "sql.h"
|
|
|
|
int db_insert(MYSQL *mysql, char *query) {
|
|
char logmessage[LOGSIZE];
|
|
|
|
if (set.verbose == POLLER_VERBOSITY_DEBUG) {
|
|
snprintf(logmessage, LOGSIZE, "DEBUG: SQLCMD: %s\n", query);
|
|
cacti_log(logmessage);
|
|
}
|
|
|
|
if (mysql_query(mysql, query)) {
|
|
snprintf(logmessage, LOGSIZE, "ERROR: Problem with MySQL: %s\n", mysql_error(mysql));
|
|
cacti_log(logmessage);
|
|
return (FALSE);
|
|
}else{
|
|
return (TRUE);
|
|
}
|
|
}
|
|
|
|
MYSQL_RES *db_query(MYSQL *mysql, char *query) {
|
|
MYSQL_RES *mysql_res;
|
|
int return_code;
|
|
|
|
return_code = mysql_query(mysql, query);
|
|
if (return_code) {
|
|
cacti_log("MYSQL: ERROR encountered while attempting to retrieve records from query\n");
|
|
exit -1;
|
|
}else{
|
|
mysql_res = mysql_store_result(mysql);
|
|
}
|
|
|
|
return mysql_res;
|
|
}
|
|
|
|
int db_connect(char *database, MYSQL *mysql) {
|
|
char logmessage[LOGSIZE];
|
|
int tries;
|
|
int result;
|
|
|
|
/* initialalize my variables */
|
|
tries = 10;
|
|
result = 0;
|
|
|
|
if (set.verbose == POLLER_VERBOSITY_DEBUG) {
|
|
snprintf(logmessage, LOGSIZE, "MYSQL: Connecting to MySQL database '%s' on '%s'...\n", database, set.dbhost);
|
|
cacti_log(logmessage);
|
|
}
|
|
|
|
mysql_init(mysql);
|
|
|
|
while (tries > 0){
|
|
tries--;
|
|
if (!mysql_real_connect(mysql, set.dbhost, set.dbuser, set.dbpass, database, set.dbport, NULL, 0)) {
|
|
if (set.verbose == POLLER_VERBOSITY_DEBUG) {
|
|
snprintf(logmessage, LOGSIZE, "MYSQL: Connection Failed: %s\n", mysql_error(mysql));
|
|
cacti_log(logmessage);
|
|
}
|
|
result = 1;
|
|
}else{
|
|
tries = 0;
|
|
result = 0;
|
|
if (set.verbose == POLLER_VERBOSITY_DEBUG) {
|
|
snprintf(logmessage, LOGSIZE, "MYSQL: Connected to MySQL database '%s' on '%s'...\n", database, set.dbhost);
|
|
cacti_log(logmessage);
|
|
}
|
|
}
|
|
}
|
|
if (result == 1){
|
|
snprintf(logmessage, LOGSIZE, "MYSQL: Connection Failed: %s\n", mysql_error(mysql));
|
|
cacti_log(logmessage);
|
|
thread_mutex_unlock(LOCK_MYSQL);
|
|
exit(0);
|
|
}else{
|
|
thread_mutex_unlock(LOCK_MYSQL);
|
|
return (0);
|
|
}
|
|
}
|
|
|
|
void db_disconnect(MYSQL *mysql) {
|
|
mysql_close(mysql);
|
|
}
|
|
|
|
|