open-dsi
/
postgres
mirror of https://github.com/postgres/postgres
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add Win32 service capability to pg_autovacuum.

Browse Source 
Dave Page
REL8_0_STABLE
Bruce Momjian 22 years ago
parent 15d3f9f6b7
commit 3afd7eaffd
2 changed files with 548 additions and 78 deletions
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Show Stats Download Patch File Download Diff File
  1. 581
      contrib/pg_autovacuum/pg_autovacuum.c
  2. 45
      contrib/pg_autovacuum/pg_autovacuum.h

581
contrib/pg_autovacuum/pg_autovacuum.c
Unescape View File

@ -2,24 +2,123 @@
* All the code for the pg_autovacuum program
* (c) 2003 Matthew T. O'Connor
* Revisions by Christopher B. Browne, Liberty RMS
* Win32 Service code added by Dave Page
*/
#include "pg_autovacuum.h"
#ifdef WIN32
#include <windows.h>
unsigned int sleep();
SERVICE_STATUS ServiceStatus;
SERVICE_STATUS_HANDLE hStatus;
int appMode = 0;
#endif
FILE *LOGOUTPUT;
char logbuffer[4096];
static void
log_entry(const char *logentry)
log_entry(const char *logentry, int level)
{
/*
* Note: Under Windows we dump the log entries to the normal
* stderr/logfile
*/
/*
* as well, otherwise it can be a pain to debug service install
* failures etc.
*/
time_t curtime;
struct tm *loctime;
char timebuffer[128];
char timebuffer[128],
slevel[10];
#ifdef WIN32
static HANDLE evtHandle = INVALID_HANDLE_VALUE;
static int last_level;
WORD elevel;
#endif
switch (level)
{
case LVL_DEBUG:
sprintf(slevel, "DEBUG: ");
break;
case LVL_INFO:
sprintf(slevel, "INFO: ");
break;
case LVL_WARNING:
sprintf(slevel, "WARNING: ");
break;
case LVL_ERROR:
sprintf(slevel, "ERROR: ");
break;
case LVL_EXTRA:
sprintf(slevel, " ");
break;
default:
sprintf(slevel, " ");
break;
}
curtime = time(NULL);
loctime = localtime(&curtime);
strftime(timebuffer, sizeof(timebuffer), "%Y-%m-%d %H:%M:%S %Z", loctime);
fprintf(LOGOUTPUT, "[%s] %s\n", timebuffer, logentry);
fprintf(LOGOUTPUT, "[%s] %s%s\n", timebuffer, slevel, logentry);
#ifdef WIN32
/* Restore the previous level if this is extra info */
if (level == LVL_EXTRA)
level = last_level;
last_level = level;
switch (level)
{
case LVL_DEBUG:
elevel = EVENTLOG_INFORMATION_TYPE;
break;
case LVL_INFO:
elevel = EVENTLOG_SUCCESS;
break;
case LVL_WARNING:
elevel = EVENTLOG_WARNING_TYPE;
break;
case LVL_ERROR:
elevel = EVENTLOG_ERROR_TYPE;
break;
default:
elevel = EVENTLOG_SUCCESS;
break;
}
if (evtHandle == INVALID_HANDLE_VALUE)
{
evtHandle = RegisterEventSource(NULL, "PostgreSQL Auto Vacuum");
if (evtHandle == NULL)
{
evtHandle = INVALID_HANDLE_VALUE;
return;
}
}
ReportEvent(evtHandle, elevel, 0, 1, NULL, 1, 0, &logentry, NULL);
#endif
}
/*
@ -29,6 +128,7 @@ log_entry(const char *logentry)
* This code is mostly ripped directly from pm_dameonize in postmaster.c with
* unneeded code removed.
*/
#ifndef WIN32
static void
daemonize()
{
@ -37,7 +137,7 @@ daemonize()
pid = fork();
if (pid == (pid_t) -1)
{
log_entry("Error: cannot disassociate from controlling TTY");
log_entry("cannot disassociate from controlling TTY", LVL_ERROR);
fflush(LOGOUTPUT);
_exit(1);
}
@ -52,13 +152,14 @@ daemonize()
#ifdef HAVE_SETSID
if (setsid() < 0)
{
log_entry("Error: cannot disassociate from controlling TTY");
log_entry("cannot disassociate from controlling TTY", LVL_ERROR);
fflush(LOGOUTPUT);
_exit(1);
}
#endif
}
#endif /* WIN32 */
/* Create and return tbl_info struct with initialized to values from row or res */
static tbl_info *
@ -68,7 +169,7 @@ init_table_info(PGresult *res, int row, db_info * dbi)
if (!new_tbl)
{
log_entry("init_table_info: Cannot get memory");
log_entry("init_table_info: Cannot get memory", LVL_ERROR);
fflush(LOGOUTPUT);
return NULL;
}
@ -82,7 +183,7 @@ init_table_info(PGresult *res, int row, db_info * dbi)
malloc(strlen(PQgetvalue(res, row, PQfnumber(res, "schemaname"))) + 1);
if (!new_tbl->schema_name)
{
log_entry("init_table_info: malloc failed on new_tbl->schema_name");
log_entry("init_table_info: malloc failed on new_tbl->schema_name", LVL_ERROR);
fflush(LOGOUTPUT);
return NULL;
}
@ -94,7 +195,7 @@ init_table_info(PGresult *res, int row, db_info * dbi)
strlen(new_tbl->schema_name) + 6);
if (!new_tbl->table_name)
{
log_entry("init_table_info: malloc failed on new_tbl->table_name");
log_entry("init_table_info: malloc failed on new_tbl->table_name", LVL_ERROR);
fflush(LOGOUTPUT);
return NULL;
}
@ -287,7 +388,7 @@ update_table_list(db_info * dbi)
{
sprintf(logbuffer, "added table: %s.%s", dbi->dbname,
((tbl_info *) DLE_VAL(DLGetTail(dbi->table_list)))->table_name);
log_entry(logbuffer);
log_entry(logbuffer, LVL_DEBUG);
}
}
} /* end of for loop that adds tables */
@ -311,7 +412,7 @@ remove_table_from_list(Dlelem *tbl_to_remove)
if (args->debug >= 1)
{
sprintf(logbuffer, "Removing table: %s from list.", tbl->table_name);
log_entry(logbuffer);
log_entry(logbuffer, LVL_DEBUG);
fflush(LOGOUTPUT);
}
DLRemove(tbl_to_remove);
@ -366,20 +467,20 @@ static void
print_table_info(tbl_info * tbl)
{
sprintf(logbuffer, " table name: %s.%s", tbl->dbi->dbname, tbl->table_name);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " relid: %u; relisshared: %i", tbl->relid, tbl->relisshared);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " reltuples: %f; relpages: %u", tbl->reltuples, tbl->relpages);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " curr_analyze_count: %li; curr_vacuum_count: %li",
tbl->curr_analyze_count, tbl->curr_vacuum_count);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " last_analyze_count: %li; last_vacuum_count: %li",
tbl->CountAtLastAnalyze, tbl->CountAtLastVacuum);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " analyze_threshold: %li; vacuum_threshold: %li",
tbl->analyze_threshold, tbl->vacuum_threshold);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
fflush(LOGOUTPUT);
}
@ -398,7 +499,7 @@ init_db_list()
DLAddHead(db_list, DLNewElem(init_dbinfo((char *) "template1", 0, 0)));
if (DLGetHead(db_list) == NULL)
{ /* Make sure init_dbinfo was successful */
log_entry("init_db_list(): Error creating db_list for db: template1.");
log_entry("init_db_list(): Error creating db_list for db: template1.", LVL_ERROR);
fflush(LOGOUTPUT);
return NULL;
}
@ -478,7 +579,7 @@ update_db_list(Dllist *db_list)
if (args->debug >= 2)
{
log_entry("updating the database list");
log_entry("updating the database list", LVL_DEBUG);
fflush(LOGOUTPUT);
}
@ -566,7 +667,7 @@ update_db_list(Dllist *db_list)
if (args->debug >= 1)
{
sprintf(logbuffer, "added database: %s", ((db_info *) DLE_VAL(DLGetTail(db_list)))->dbname);
log_entry(logbuffer);
log_entry(logbuffer, LVL_DEBUG);
}
}
} /* end of for loop that adds tables */
@ -626,7 +727,7 @@ remove_db_from_list(Dlelem *db_to_remove)
if (args->debug >= 1)
{
sprintf(logbuffer, "Removing db: %s from list.", dbi->dbname);
log_entry(logbuffer);
log_entry(logbuffer, LVL_DEBUG);
fflush(LOGOUTPUT);
}
DLRemove(db_to_remove);
@ -694,27 +795,27 @@ static void
print_db_info(db_info * dbi, int print_tbl_list)
{
sprintf(logbuffer, "dbname: %s", (dbi->dbname) ? dbi->dbname : "(null)");
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " oid: %u", dbi->oid);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " username: %s", (dbi->username) ? dbi->username : "(null)");
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " password: %s", (dbi->password) ? dbi->password : "(null)");
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
if (dbi->conn != NULL)
log_entry(" conn is valid, (connected)");
log_entry(" conn is valid, (connected)", LVL_INFO);
else
log_entry(" conn is null, (not connected)");
log_entry(" conn is null, (not connected)", LVL_INFO);
sprintf(logbuffer, " default_analyze_threshold: %li", dbi->analyze_threshold);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " default_vacuum_threshold: %li", dbi->vacuum_threshold);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
fflush(LOGOUTPUT);
if (print_tbl_list > 0)
@ -737,7 +838,7 @@ db_connect(db_info * dbi)
{
sprintf(logbuffer, "Failed connection to database %s with error: %s.",
dbi->dbname, PQerrorMessage(db_conn));
log_entry(logbuffer);
log_entry(logbuffer, LVL_ERROR);
fflush(LOGOUTPUT);
PQfinish(db_conn);
db_conn = NULL;
@ -779,7 +880,7 @@ send_query(const char *query, db_info * dbi)
return NULL;
if (args->debug >= 4)
log_entry(query);
log_entry(query, LVL_DEBUG);
res = PQexec(dbi->conn, query);
@ -788,9 +889,9 @@ send_query(const char *query, db_info * dbi)
sprintf(logbuffer,
"Fatal error occured while sending query (%s) to database %s",
query, dbi->dbname);
log_entry(logbuffer);
log_entry(logbuffer, LVL_ERROR);
sprintf(logbuffer, "The error is [%s]", PQresultErrorMessage(res));
log_entry(logbuffer);
log_entry(logbuffer, LVL_EXTRA);
fflush(LOGOUTPUT);
return NULL;
}
@ -800,9 +901,9 @@ send_query(const char *query, db_info * dbi)
sprintf(logbuffer,
"Can not refresh statistics information from the database %s.",
dbi->dbname);
log_entry(logbuffer);
log_entry(logbuffer, LVL_ERROR);
sprintf(logbuffer, "The error is [%s]", PQresultErrorMessage(res));
log_entry(logbuffer);
log_entry(logbuffer, LVL_EXTRA);
fflush(LOGOUTPUT);
PQclear(res);
return NULL;
@ -837,7 +938,14 @@ get_cmd_args(int argc, char *argv[])
args->analyze_base_threshold = -1;
args->analyze_scaling_factor = -1;
args->debug = AUTOVACUUM_DEBUG;
#ifndef WIN32
args->daemonize = 0;
#else
args->install_as_service = 0;
args->remove_as_service = 0;
args->service_user = 0;
args->service_password = 0;
#endif
args->user = 0;
args->password = 0;
args->host = 0;
@ -848,7 +956,11 @@ get_cmd_args(int argc, char *argv[])
* Fixme: Should add some sanity checking such as positive integer
* values etc
*/
#ifndef WIN32
while ((c = getopt(argc, argv, "s:S:v:V:a:A:d:U:P:H:L:p:hD")) != -1)
#else
while ((c = getopt(argc, argv, "s:S:v:V:a:A:d:U:P:H:L:p:hIRN:W:")) != -1)
#endif
{
switch (c)
{
@ -870,9 +982,11 @@ get_cmd_args(int argc, char *argv[])
case 'A':
args->analyze_scaling_factor = atof(optarg);
break;
#ifndef WIN32
case 'D':
args->daemonize++;
break;
#endif
case 'd':
args->debug = atoi(optarg);
break;
@ -894,6 +1008,20 @@ get_cmd_args(int argc, char *argv[])
case 'h':
usage();
exit(0);
#ifdef WIN32
case 'I':
args->install_as_service++;
break;
case 'R':
args->remove_as_service++;
break;
case 'N':
args->service_user = optarg;
break;
case 'W':
args->service_password = optarg;
break;
#endif
default:
/*
@ -925,7 +1053,14 @@ usage()
float f = 0;
fprintf(stderr, "usage: pg_autovacuum \n");
#ifndef WIN32
fprintf(stderr, " [-D] Daemonize (Detach from tty and run in the background)\n");
#else
fprintf(stderr, " [-I] Install as a Windows service\n");
fprintf(stderr, " [-R] Remove as a Windows service (all other options will be ignored)\n");
fprintf(stderr, " [-N] Username to run service as (only useful when installing as a Windows service)\n");
fprintf(stderr, " [-W] Password to run service with (only useful when installing as a Windows service)\n");
#endif
i = AUTOVACUUM_DEBUG;
fprintf(stderr, " [-d] debug (debug level=0,1,2,3; default=%i)\n", i);
@ -957,41 +1092,226 @@ static void
print_cmd_args()
{
sprintf(logbuffer, "Printing command_args");
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " args->host=%s", (args->host) ? args->host : "(null)");
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " args->port=%s", (args->port) ? args->port : "(null)");
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " args->username=%s", (args->user) ? args->user : "(null)");
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " args->password=%s", (args->password) ? args->password : "(null)");
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " args->logfile=%s", (args->logfile) ? args->logfile : "(null)");
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
#ifndef WIN32
sprintf(logbuffer, " args->daemonize=%i", args->daemonize);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
#else
sprintf(logbuffer, " args->install_as_service=%i", args->install_as_service);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " args->remove_as_service=%i", args->remove_as_service);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " args->service_user=%s", (args->service_user) ? args->service_user : "(null)");
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " args->service_password=%s", (args->service_password) ? args->service_password : "(null)");
log_entry(logbuffer, LVL_INFO);
#endif
sprintf(logbuffer, " args->sleep_base_value=%i", args->sleep_base_value);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " args->sleep_scaling_factor=%f", args->sleep_scaling_factor);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " args->vacuum_base_threshold=%i", args->vacuum_base_threshold);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " args->vacuum_scaling_factor=%f", args->vacuum_scaling_factor);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " args->analyze_base_threshold=%i", args->analyze_base_threshold);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " args->analyze_scaling_factor=%f", args->analyze_scaling_factor);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
sprintf(logbuffer, " args->debug=%i", args->debug);
log_entry(logbuffer);
log_entry(logbuffer, LVL_INFO);
fflush(LOGOUTPUT);
}
/* Beginning of AutoVacuum Main Program */
#ifdef WIN32
/* Handle control requests from the Service Control Manager */
static void
ControlHandler(DWORD request)
{
switch (request)
{
case SERVICE_CONTROL_STOP:
case SERVICE_CONTROL_SHUTDOWN:
log_entry("pg_autovacuum service stopping...", LVL_INFO);
fflush(LOGOUTPUT);
ServiceStatus.dwWin32ExitCode = 0;
ServiceStatus.dwCurrentState = SERVICE_STOPPED;
SetServiceStatus(hStatus, &ServiceStatus);
return;
default:
break;
}
/* Report current status */
SetServiceStatus(hStatus, &ServiceStatus);
return;
}
/* Register with the Service Control Manager */
static int
InstallService()
{
SC_HANDLE schService = NULL;
SC_HANDLE schSCManager = NULL;
char szFilename[MAX_PATH],
szKey[MAX_PATH],
szCommand[MAX_PATH + 1024],
szMsgDLL[MAX_PATH];
HKEY hk = NULL;
DWORD dwData = 0;
/*
* Register the service with the SCM
*/
GetModuleFileName(NULL, szFilename, MAX_PATH);
/* Open the Service Control Manager on the local computer. */
schSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
if (!schSCManager)
return -1;
schService = CreateService(
schSCManager, /* SCManager database */
TEXT("pg_autovacuum"), /* Name of service */
TEXT("PostgreSQL Auto Vacuum"), /* Name to display */
SERVICE_ALL_ACCESS, /* Desired access */
SERVICE_WIN32_OWN_PROCESS, /* Service type */
SERVICE_AUTO_START, /* Start type */
SERVICE_ERROR_NORMAL, /* Error control type */
szFilename, /* Service binary */
NULL, /* No load ordering group */
NULL, /* No tag identifier */
NULL, /* Dependencies */
args->service_user, /* Service account */
args->service_password); /* Account password */
if (!schService)
return -2;
/*
* Rewrite the command line for the service
*/
sprintf(szKey, "SYSTEM\\CurrentControlSet\\Services\\pg_autovacuum");
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, szKey, 0, KEY_ALL_ACCESS, &hk))
return -3;
/* Build the command line */
sprintf(szCommand, "\"%s\"", szFilename);
if (args->host)
sprintf(szCommand, "%s -H %s", szCommand, args->host);
if (args->port)
sprintf(szCommand, "%s -p %s", szCommand, args->port);
if (args->user)
sprintf(szCommand, "%s -U %s", szCommand, args->user);
if (args->password)
sprintf(szCommand, "%s -p %s", szCommand, args->password);
if (args->logfile)
sprintf(szCommand, "%s -L %s", szCommand, args->logfile);
if (args->sleep_base_value != (int) SLEEPBASEVALUE)
sprintf(szCommand, "%s -s %i", szCommand, args->sleep_base_value);
if (args->sleep_scaling_factor != (float) SLEEPSCALINGFACTOR)
sprintf(szCommand, "%s -S %f", szCommand, args->sleep_scaling_factor);
if (args->vacuum_base_threshold != (int) VACBASETHRESHOLD)
sprintf(szCommand, "%s -v %i", szCommand, args->vacuum_base_threshold);
if (args->vacuum_scaling_factor != (float) VACSCALINGFACTOR)
sprintf(szCommand, "%s -V %f", szCommand, args->vacuum_scaling_factor);
if (args->analyze_base_threshold != (int) (VACBASETHRESHOLD / 2))
sprintf(szCommand, "%s -a %i", szCommand, args->analyze_base_threshold);
if (args->analyze_scaling_factor != (float) (VACSCALINGFACTOR / 2))
sprintf(szCommand, "%s -A %f", szCommand, args->analyze_scaling_factor);
if (args->debug != (int) AUTOVACUUM_DEBUG)
sprintf(szCommand, "%s -d %i", szCommand, args->debug);
/* And write the new value */
if (RegSetValueEx(hk, "ImagePath", 0, REG_EXPAND_SZ, (LPBYTE) szCommand, (DWORD) strlen(szCommand) + 1))
return -4;
RegCloseKey(hk);
/*
* Set the Event source for the application log
*/
sprintf(szKey, "SYSTEM\\CurrentControlSet\\Services\\EventLog\\Application\\PostgreSQL Auto Vacuum");
if (RegCreateKeyEx(HKEY_LOCAL_MACHINE, szKey, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hk, NULL))
return -5;
/* TODO Create an actual message file! */
/* Message count */
sprintf(szMsgDLL, "pgmessages.dll");
if (RegSetValueEx(hk, "EventMessageFile", 0, REG_EXPAND_SZ, (LPBYTE) szMsgDLL, (DWORD) strlen(szMsgDLL) + 1))
return -6;
/* Category message file */
if (RegSetValueEx(hk, "CategoryMessageFile", 0, REG_EXPAND_SZ, (LPBYTE) szMsgDLL, (DWORD) strlen(szMsgDLL) + 1))
return -7;
/* Category message count */
dwData = 0;
if (RegSetValueEx(hk, "CategoryCount", 0, REG_DWORD, (LPBYTE) & dwData, sizeof(DWORD)))
return -8;
/* Set the event types supported */
dwData = EVENTLOG_ERROR_TYPE | EVENTLOG_WARNING_TYPE | EVENTLOG_INFORMATION_TYPE | EVENTLOG_SUCCESS;
if (RegSetValueEx(hk, "TypesSupported", 0, REG_DWORD, (LPBYTE) & dwData, sizeof(DWORD)))
return -9;
RegCloseKey(hk);
return 0;
}
/* Unregister from the Service Control Manager */
static int
RemoveService()
{
SC_HANDLE schService = NULL;
SC_HANDLE schSCManager = NULL;
char szKey[MAX_PATH];
HKEY hk = NULL;
/* Open the SCM */
schSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
if (!schSCManager)
return -1;
/* Open the service */
schService = OpenService(schSCManager, TEXT("pg_autovacuum"), SC_MANAGER_ALL_ACCESS);
if (!schService)
return -2;
/* Now delete the service */
if (!DeleteService(schService))
return -3;
/*
* Remove the Event source from the application log
*/
sprintf(szKey, "SYSTEM\\CurrentControlSet\\Services\\EventLog\\Application");
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, szKey, 0, KEY_ALL_ACCESS, &hk))
return -4;
if (RegDeleteKey(hk, "PostgreSQL Auto Vacuum"))
return -5;
return 0;
}
#endif /* WIN32 */
static
int
main(int argc, char *argv[])
VacuumLoop(int argc, char **argv)
{
char buf[256];
int j = 0,
@ -1006,29 +1326,32 @@ main(int argc, char *argv[])
tbl_info *tbl;
PGresult *res = NULL;
double diff;
struct timeval now,
then;
args = get_cmd_args(argc, argv); /* Get Command Line Args and put
* them in the args struct */
#ifdef WIN32
/* Dameonize if requested */
if (args->daemonize == 1)
daemonize();
if (appMode)
log_entry("pg_autovacuum starting in Windows Application mode", LVL_INFO);
else
log_entry("pg_autovacuum starting in Windows Service mode", LVL_INFO);
if (args->logfile)
ServiceStatus.dwServiceType = SERVICE_WIN32;
ServiceStatus.dwCurrentState = SERVICE_START_PENDING;
ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
ServiceStatus.dwWin32ExitCode = 0;
ServiceStatus.dwServiceSpecificExitCode = 0;
ServiceStatus.dwCheckPoint = 0;
ServiceStatus.dwWaitHint = 0;
if (!appMode)
{
LOGOUTPUT = fopen(args->logfile, "a");
if (!LOGOUTPUT)
{
fprintf(stderr, "Could not open log file - [%s]\n", args->logfile);
exit(-1);
}
hStatus = RegisterServiceCtrlHandler("pg_autovacuum", (LPHANDLER_FUNCTION) ControlHandler);
if (hStatus == (SERVICE_STATUS_HANDLE) 0)
return -1;
}
else
LOGOUTPUT = stderr;
if (args->debug >= 2)
print_cmd_args();
#endif /* WIN32 */
/* Init the db list with template1 */
db_list = init_db_list();
@ -1037,8 +1360,8 @@ main(int argc, char *argv[])
if (check_stats_enabled(((db_info *) DLE_VAL(DLGetHead(db_list)))) != 0)
{
log_entry("Error: GUC variable stats_row_level must be enabled.");
log_entry(" Please fix the problems and try again.");
log_entry("GUC variable stats_row_level must be enabled.", LVL_ERROR);
log_entry(" Please fix the problems and try again.", LVL_EXTRA);
fflush(LOGOUTPUT);
exit(1);
@ -1046,8 +1369,19 @@ main(int argc, char *argv[])
gettimeofday(&then, 0); /* for use later to caluculate sleep time */
#ifndef WIN32
while (1)
{ /* Main Loop */
#else
/* We can now report the running status to SCM. */
ServiceStatus.dwCurrentState = SERVICE_RUNNING;
if (!appMode)
SetServiceStatus(hStatus, &ServiceStatus);
while (ServiceStatus.dwCurrentState == SERVICE_RUNNING)
#endif
{
/* Main Loop */
db_elem = DLGetHead(db_list); /* Reset cur_db_node to the
* beginning of the db_list */
@ -1059,9 +1393,16 @@ main(int argc, char *argv[])
if (dbs->conn == NULL)
{ /* Serious problem: We can't connect to
* template1 */
log_entry("Error: Cannot connect to template1, exiting.");
log_entry("Cannot connect to template1, exiting.", LVL_ERROR);
fflush(LOGOUTPUT);
fclose(LOGOUTPUT);
#ifdef WIN32
ServiceStatus.dwCurrentState = SERVICE_STOPPED;
ServiceStatus.dwWin32ExitCode = ERROR_SERVICE_SPECIFIC_ERROR;
ServiceStatus.dwServiceSpecificExitCode = -1;
if (!appMode)
SetServiceStatus(hStatus, &ServiceStatus);
#endif
exit(1);
}
}
@ -1135,7 +1476,7 @@ main(int argc, char *argv[])
if (args->debug >= 1)
{
sprintf(logbuffer, "Performing: %s", buf);
log_entry(logbuffer);
log_entry(logbuffer, LVL_DEBUG);
fflush(LOGOUTPUT);
}
send_query(buf, dbs);
@ -1149,7 +1490,7 @@ main(int argc, char *argv[])
if (args->debug >= 1)
{
sprintf(logbuffer, "Performing: %s", buf);
log_entry(logbuffer);
log_entry(logbuffer, LVL_DEBUG);
fflush(LOGOUTPUT);
}
send_query(buf, dbs);
@ -1193,7 +1534,7 @@ main(int argc, char *argv[])
sprintf(logbuffer,
"%i All DBs checked in: %.0f usec, will sleep for %i secs.",
loops, diff, sleep_secs);
log_entry(logbuffer);
log_entry(logbuffer, LVL_DEBUG);
fflush(LOGOUTPUT);
}
@ -1209,5 +1550,99 @@ main(int argc, char *argv[])
*/
free_db_list(db_list);
free_cmd_args();
return 0;
}
/* Beginning of AutoVacuum Main Program */
int
main(int argc, char *argv[])
{
#ifdef WIN32
LPVOID lpMsgBuf;
SERVICE_TABLE_ENTRY ServiceTable[2];
#endif
args = get_cmd_args(argc, argv); /* Get Command Line Args and put
* them in the args struct */
#ifndef WIN32
/* Dameonize if requested */
if (args->daemonize == 1)
daemonize();
#endif
if (args->logfile)
{
LOGOUTPUT = fopen(args->logfile, "a");
if (!LOGOUTPUT)
{
fprintf(stderr, "Could not open log file - [%s]\n", args->logfile);
exit(-1);
}
}
else
LOGOUTPUT = stderr;
if (args->debug >= 2)
print_cmd_args();
#ifdef WIN32
/* Install as a Windows service if required */
if (args->install_as_service)
{
if (InstallService() != 0)
{
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, NULL, GetLastError(), MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR) & lpMsgBuf, 0, NULL);
sprintf(logbuffer, "%s", (char *) lpMsgBuf);
log_entry(logbuffer, LVL_ERROR);
fflush(LOGOUTPUT);
exit(-1);
}
else
{
log_entry("Successfully installed Windows service", LVL_INFO);
fflush(LOGOUTPUT);
exit(0);
}
}
/* Remove as a Windows service if required */
if (args->remove_as_service)
{
if (RemoveService() != 0)
{
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, NULL, GetLastError(), MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR) & lpMsgBuf, 0, NULL);
sprintf(logbuffer, "%s", (char *) lpMsgBuf);
log_entry(logbuffer, LVL_ERROR);
fflush(LOGOUTPUT);
exit(-1);
}
else
{
log_entry("Successfully removed Windows service", LVL_INFO);
fflush(LOGOUTPUT);
exit(0);
}
}
/* Normal service startup */
ServiceTable[0].lpServiceName = "pg_autovacuum";
ServiceTable[0].lpServiceProc = (LPSERVICE_MAIN_FUNCTION) VacuumLoop;
ServiceTable[1].lpServiceName = NULL;
ServiceTable[1].lpServiceProc = NULL;
/* Start the control dispatcher thread for our service */
if (!StartServiceCtrlDispatcher(ServiceTable))
{
appMode = 1;
VacuumLoop(0, NULL);
}
#else /* Unix */
/* Call the main program loop. */
VacuumLoop(0, NULL);
#endif /* WIN32 */
return EXIT_SUCCESS;
}

45
contrib/pg_autovacuum/pg_autovacuum.h
Unescape View File

@ -1,8 +1,10 @@
/* pg_autovacuum.h
/* pg_autovacuum.hszCmdline
* Header file for pg_autovacuum.c
* (c) 2003 Matthew T. O'Connor
*/
#define FRONTEND
#include "postgres_fe.h"
#include <unistd.h>
@ -23,7 +25,7 @@
#include "/usr/include/pgsql/server/lib/dllist.h"
*/
#define AUTOVACUUM_DEBUG 1
#define AUTOVACUUM_DEBUG 0
#define VACBASETHRESHOLD 1000
#define VACSCALINGFACTOR 2
#define SLEEPBASEVALUE 300
@ -34,9 +36,9 @@
#define VACUUM_ANALYZE 0
#define ANALYZE_ONLY 1
#define TABLE_STATS_QUERY "select a.oid,a.relname,a.relnamespace,a.relpages,a.relisshared,a.reltuples,b.schemaname,b.n_tup_ins,b.n_tup_upd,b.n_tup_del from pg_class a, pg_stat_all_tables b where a.oid=b.relid and a.relkind = 'r' and schemaname not like 'pg_temp_%'"
#define FRONTEND
#define TABLE_STATS_QUERY "select a.oid,a.relname,a.relnamespace,a.relpages,a.relisshared,a.reltuples,b.schemaname,b.n_tup_ins,b.n_tup_upd,b.n_tup_del from pg_class a, pg_stat_all_tables b where a.oid=b.relid and a.relkind = 'r'"
#define PAGES_QUERY "select oid,reltuples,relpages from pg_class where oid=%u"
#define FROZENOID_QUERY "select oid,age(datfrozenxid) from pg_database where datname = 'template1'"
#define FROZENOID_QUERY2 "select oid,datname,age(datfrozenxid) from pg_database where datname!='template0'"
@ -44,6 +46,16 @@
/* define atooid */
#define atooid(x) ((Oid) strtoul((x), NULL, 10))
/* Log levels */
enum
{
LVL_DEBUG = 1,
LVL_INFO,
LVL_WARNING,
LVL_ERROR,
LVL_EXTRA
};
/* define cmd_args stucture */
struct cmdargs
{
@ -51,18 +63,28 @@ struct cmdargs
analyze_base_threshold,
sleep_base_value,
debug,
#ifndef WIN32
daemonize;
#else
install_as_service,
remove_as_service;
#endif
float vacuum_scaling_factor,
analyze_scaling_factor,
sleep_scaling_factor;
char *user,
*password,
#ifdef WIN32
*service_user,
*service_password,
#endif
*host,
*logfile,
*port;
};
typedef struct cmdargs cmd_args;
/* define cmd_args as global so we can get to them everywhere */
cmd_args *args;
@ -106,6 +128,9 @@ struct tableinfo
};
typedef struct tableinfo tbl_info;
/* The main program loop function */
static int VacuumLoop(int argc, char **argv);
/* Functions for dealing with command line arguements */
static cmd_args *get_cmd_args(int argc, char *argv[]);
static void print_cmd_args(void);
@ -138,5 +163,15 @@ static void db_disconnect(db_info * dbi);
static PGresult *send_query(const char *query, db_info * dbi);
/* Other Generally needed Functions */
#ifndef WIN32
static void daemonize(void);
static void log_entry(const char *logentry);
#endif
static void log_entry(const char *logentry, int level);
#ifdef WIN32
/* Windows Service related functions */
static void ControlHandler(DWORD request);
static int InstallService();
static int RemoveService();
#endif

Write Preview
Loading…
Cancel
Save