后台进程(守护进程)自动备份PostgreSQL数据库

从当前目录中读取ini配置文件的登录数据库必要的参数,登录数据库后获取两次备份的间隔天数,然后启动一个线程隔1分钟检查一下是否需要备份。

之前查资料查了好久,才找到

"pg_dump \"host=%s port=%d user=%s password=%s dbname=%s\" > db_bak\\%s.bak"

这种可以避免输入密码,实现自动备份(不用配置环境变量或改动数据库的登录权限)

#include <windows.h>
#include <stdio.h>
#include <stdlib.h>
#include <process.h>
#include "inc/libpq-fe.h"
#include <time.h>
 
#pragma comment(linker, "/subsystem:windows /entry:mainCRTStartup")
#pragma comment(lib, "lib/libpq.lib")
 
#define DIRECTORY "db_bak"
#define INIFILE "sysinfo.ini"
#define CONNINFO "host=localhost port=5432 user=postgres password=123456 dbname=DB_SPS"
#define QUERYSQL "select sys_val from tb_sysinfo where sys_key=\'backup_interval_time\'"
#define SPACEUNIT (60*60*24)

// BEGTIME:00 ~ ENDTIME:59
#define BEGTIME 2
#define ENDTIME 3

typedef struct tagArgs
{
	int t_space;
	char db_ip[16];
	char db_name[10];
	char db_user[10];
	char db_passwd[10];
	unsigned short db_port;
}Args;

unsigned int __stdcall ThreadCheckAndSave(void* arglist)
{
	struct tm tlasttime = {0}, tcurtime = {0};
	time_t lasttime = 0, curtime = 0;
	Args args = {0};
	char cmd[256] = {0}, strtime[256] = {0};

	if(arglist == NULL)
	{
		printf("error for begin and space args !\n");
		_endthreadex(1);
	}
	memcpy(&args,arglist,sizeof(args));

	lasttime = time(NULL);
	do{
		curtime = time(NULL);
		if((difftime(curtime,lasttime)/SPACEUNIT) > args.t_space)
		{
			localtime_s(&tcurtime,&curtime);
			if(tcurtime.tm_hour >= BEGTIME && tcurtime.tm_hour <= ENDTIME)
			{
				//备份
				memset(cmd,0,sizeof(cmd));
				memset(strtime,0,sizeof(strtime));
				localtime_s(&tcurtime, &curtime);
				strftime(strtime,sizeof(strtime),"%Y-%m-%d(%H_%M_%S)",&tcurtime);
				CreateDirectoryA(DIRECTORY,NULL);
				sprintf(cmd,"pg_dump \"host=%s port=%d user=%s password=%s dbname=%s\" > db_bak\\%s.bak",
										args.db_ip,
										args.db_port,
										args.db_user,
										args.db_passwd,
										args.db_name,
										strtime);
				system(cmd);
				lasttime = curtime;
			}
			Sleep(60000);
		}
	}while(true);

	_endthreadex(0);
	return 0;
}

int main(int argc, const char* argv[])
{
	PGconn* conn = NULL;
	PGresult* result = NULL;
	char* getstr = NULL;
	
	Args args = {0};
	HANDLE hthread = NULL;
	char inipath[MAX_PATH] = {0};
	char strsql[MAX_PATH] = {0};

	GetCurrentDirectoryA(sizeof(inipath),inipath);
	strcat(inipath,"\\");
	strcat(inipath,INIFILE);
	GetPrivateProfileStringA("db","#ip","127.0.0.1",args.db_ip,sizeof(args.db_ip),inipath);
	GetPrivateProfileStringA("db","#name","",args.db_name,sizeof(args.db_name),inipath);
	GetPrivateProfileStringA("db","#user","",args.db_user,sizeof(args.db_user),inipath);
	GetPrivateProfileStringA("db","#passwd","",args.db_passwd,sizeof(args.db_passwd),inipath);
	args.db_port = GetPrivateProfileIntA("db","#port",0,inipath);
	sprintf(strsql,"host=%s port=%d user=%s password=%s dbname=%s",args.db_ip,args.db_port,args.db_user,args.db_passwd,args.db_name);

	conn = PQconnectdb(strsql);
	if(PQstatus(conn) != CONNECTION_OK)
		goto END;
	result = PQexec(conn,QUERYSQL);
	if(PQresultStatus(result) != PGRES_TUPLES_OK)
		goto END;
	//获取配置参数
	getstr = PQgetvalue(result,0,0);
	if(getstr == NULL)
		goto END;
	args.t_space = atoi(getstr);

	//循环检查备份数据库
	hthread = (HANDLE)_beginthreadex(NULL,0,ThreadCheckAndSave,&args,CREATE_SUSPENDED,NULL);
	SetPriorityClass(GetCurrentProcess(),BELOW_NORMAL_PRIORITY_CLASS);
	//SetThreadPriority(hthread,THREAD_PRIORITY_IDLE);
	ResumeThread(hthread);

END:
	if(result != NULL)
	{
		printf("PQexec : %s\n", PQerrorMessage(conn));
		PQclear(result);
		result = NULL;
	}
	if(conn != NULL)
	{
		printf("PQconn : %s\n", PQerrorMessage(conn));
		PQfinish(conn);
		conn = NULL;
	}
	if(hthread != NULL)
		WaitForSingleObject(hthread,INFINITE);
	else
		return -1;

	return 0;
}

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏FreeBuf

新手科普 | MySQL手工注入之基本注入流程

MySQL手工注入的基本步骤以及一些技巧的记录,当出现学习手工注入的时候,网上的文章参差不齐,导致很长一段时间对手工注入的理解一直处于一知半解的状态,特此记录本...

1967
来自专栏乐沙弥的世界

Oracle AWR 阙值影响历史执行计划

      最近有网友提到为什么在dba_hist_sql_plan中无法查看到sql语句的历史执行计划,对于这个问题是由于缺省情况下,Oracle 设定的阙值...

662
来自专栏码云1024

sql 复习练习

4006
来自专栏晨星先生的自留地

mysql注入高级篇1--内置系统表注入

1853
来自专栏谢庆玲的专栏

MySQL 查询分析

本文主要由一个案例引发对 MySQL 性能问题的思考,主要讲述 MySQL 慢查询和 explain 工具这两个定位 MySQL 性能瓶颈的方法。

1.1K1
来自专栏机器学习算法与Python学习

SQL Server常用命令(平时不用别忘了)

SQL Server 2008 在Microsoft的数据平台上发布,可以组织管理任何数据。可以将结构化、半结构化和非结构化文档的数据直接存储到数据库中。可以对...

2907
来自专栏杨建荣的学习笔记

手工创建/删除数据库的步骤

今天和大家分享下数据库的创建和删除的步骤,里面有很多细节需要大家考虑。创建数据库不只是一个create database语句。删除数据库 drop databa...

3136
来自专栏熊二哥

那些年我们写过的T-SQL(下篇)

下篇的内容很多都会在工作中用到,尤其是可编程对象,那些年我们写过的存储过程,有木有?到目前为止很多大型传统企业仍然很依赖存储过程。这部分主要难理解的部分是事务和...

2105
来自专栏MYSQL轻松学

MYSQL RR隔离级别下MVCC及锁解读

MVCC(Multi-Version Concurrent Control):多版本并发控制,只作用于RC和RR隔离级别,主要是为了避免脏读、非重复读,而非幻读...

4658
来自专栏zcqshine's blog

Mysql Packet for query is too large解决方法

3827

扫码关注云+社区