前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >后台进程(守护进程)自动备份PostgreSQL数据库

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

作者头像
_gongluck
发布2018-03-08 15:22:58
1.3K0
发布2018-03-08 15:22:58
举报

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

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

代码语言:javascript
复制
"pg_dump \"host=%s port=%d user=%s password=%s dbname=%s\" > db_bak\\%s.bak"

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

代码语言:javascript
复制
#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;
}
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2017年10月27日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档