agtpay.ec

简单的银行代收系统

/************************************************************
 *	File: agtpay.ec
 *	Function: 批量代发工资
 *
 *  Author: bc    Date: 2008.03.06
 ***********************************************************/


#include <stdio.h>
#include <math.h>
#include <time.h>
#include <varargs.h>

EXEC SQL include "dbbuf.h";
EXEC SQL include sqlca;

#define RETNAME "agt_ret.txt"
#define ERRNAME "errlog"
#define ZEROAMT 0.0001 

void batlog();
char *rtrim();
char *strchr();
double atof();
char *get_value();
int read_tot_line();
int read_body();
int insert_payret();
int gen_retfile();

int
main(int argc, char * argv[])
{
	EXEC SQL BEGIN DECLARE SECTION;
		struct s_acc_bal acc_bal; /* 帐户余额表结构 */
		struct s_agt_payret agt_payret; /* 发放结果表结构 */
		struct s_data_head datahead; /* 第一行汇总信息 */
		struct s_data_body databody; /* 明细信息 */
		char database[20+1]; /* 数据库名 */
		char msg[60+1];
		char flag[1+1];
		double bal = 0.00;
	EXEC SQL END DECLARE SECTION;
	char tmpstr[200+1];
	long reflag = 0; /* 补发标志 */
	long succ = 0; /* 成功失败标志 */
	long tot_num = 0; /* 总记录条数 */
	long suc_num = 0; /* 成功记录数 */
	long bad_num = 0; /* 失败记录数 */
	double suc_amt = 0.00; /* 成功金额 */
	double bad_amt = 0.00; /* 失败金额 */
	long i = 0;
	long ret = 0;


	FILE * fp = NULL; /* 文件指针 */

	memset(&acc_bal, 0, sizeof(struct s_acc_bal));
	memset(&agt_payret, 0, sizeof(struct s_agt_payret));
	memset(&datahead, 0, sizeof(struct s_data_head));
	memset(&databody, 0, sizeof(struct s_data_body));
	memset(tmpstr, 0, sizeof(tmpstr));
	memset(database, 0, sizeof(database));
	memset(msg, 0, sizeof(msg));
	memset(flag, 0, sizeof(flag));


	/* 判断参数是否合法 */
    if (argc != 3 )
	{
		printf("Usge: agtpay database_name agt_pay.txt\n");	
		return (0);
	}

	fp = fopen(argv[2], "r");
	if (fp == NULL) 
	{
		batlog("打开文件失败!file_name[%s]  [%s][%d]", argv[2], __FILE__, __LINE__);
		return (-1);
	}

	/* 打开数据库 */
	strcpy(database, argv[1]);
	rtrim(database);
	EXEC SQL DATABASE :database;
	if (SQLCODE)
	{
		batlog("打开数据库失败!database_name[%s][%d]  [%s][%d]", database, SQLCODE, __FILE__, __LINE__);
		return (-1);
	}

	batlog("批量代发工资开始!  [%s][%d]", __FILE__, __LINE__);

	/* 读取文件第一行汇总信息 */	
	if (fgets(tmpstr, sizeof(tmpstr), fp) == NULL)
	{
		batlog("文件可能为空!file_name[%s]  [%s][%d]", argv[2], __FILE__, __LINE__);
		fclose(fp);
		return (-1);
	}
	ret = read_tot_line(tmpstr, &datahead);
	if (ret < 0)
	{
		batlog("读取文件汇总信息失败!file_name[%s]  [%s][%d]", argv[2], __FILE__, __LINE__);
		fclose(fp);
		return (-1);
	}

	/* 开始事务 */
	EXEC SQL BEGIN WORK;
	if (SQLCODE)
	{
		batlog("启动事务失败!sqlcode[%d]  [%s][%d]", SQLCODE, __FILE__, __LINE__);
		fclose(fp);
		return (-1);
	}

	EXEC SQL SET LOCK MODE TO WAIT 5;
	if (SQLCODE)
	{
		batlog("设置锁等待失败!sqlcode[%d]  [%s][%d]", SQLCODE, __FILE__, __LINE__);
		fclose(fp);
		EXEC SQL ROLLBACK WORK;
		return (-1);
	}

	/* 锁内部帐号 */
	EXEC SQL UPDATE t_acc_bal SET flag = flag 
		WHERE acc = :datahead.inn_acc;
	if (SQLCODE)
	{
		batlog("锁内部帐户失败!sqlcode[%d]acc[%s]  [%s][%d]", SQLCODE, datahead.inn_acc,__FILE__, __LINE__);
		fclose(fp);
		EXEC SQL ROLLBACK WORK;
		return (-1);
	}

	/* 查询内部帐余额,和状态 */
	EXEC SQL SELECT bal, flag INTO :bal, :flag
		FROM t_acc_bal
		WHERE acc = :datahead.inn_acc;
	if (SQLCODE)
	{
		batlog("查询内部帐户余额失败!sqlcode[%d]acc[%s]  [%s][%d]", SQLCODE, datahead.inn_acc,__FILE__, __LINE__);
		fclose(fp);
		EXEC SQL ROLLBACK WORK;
		return (-1);
	}

	/* 内部帐户状态检查 */
	if(flag[0] != '0')
	{
		batlog("内部帐户状态不正常!sqlcode[%d]acc[%s]  [%s][%d]", SQLCODE, datahead.inn_acc,__FILE__, __LINE__);
		fclose(fp);
		EXEC SQL ROLLBACK WORK;
		return (-1);
	}

	reflag = 0;
	/* 逐条处理汇总信息 */
    while(fgets(tmpstr, sizeof(tmpstr), fp) != NULL)
    {
		tot_num++;
		succ = 0;
		memset(&databody, 0, sizeof(databody));
		memset(msg, 0, sizeof(msg));
		/* 取明细信息 */
		ret = read_body(tmpstr, &databody);
		if (ret < 0)
		{
			batlog("取得数据不合法!acc[%s][%d]  [%s][%d]", databody.acc, i, __FILE__, __LINE__);
			if( strlen(databody.acc) != 0 )
			{
				strcpy(msg,"取得的数据不合法!");
				insert_payret(&datahead, &databody, 1, msg);
			}
			continue;
		}

		/* 判断该条是否已处理 */
		EXEC SQL SELECT * INTO :agt_payret
			FROM t_agt_payret
			WHERE acc = :databody.acc
			AND	corp_name = :datahead.corp_name
			AND tran_date = :datahead.tran_date;
		if (SQLCODE == 0 && agt_payret.flag[0] == '0')
		{ /* 已经成功处理过 */
			reflag = 1;	
			continue;
		}
		else if (SQLCODE == 0 && agt_payret.flag[0] != '0')
		{ /* 已经处理过,但上次处理失败 */
			reflag = 1;	

			/* 判断余额是否足够 */
			if (bal < databody.amt)
			{
				succ = 1;
				strcpy(msg, "内部帐户余额不足!");
				batlog("%sacc[%s][%d]  [%s][%d]", msg,databody.acc, i, __FILE__, __LINE__);
			}
			
			/* 查询帐户状态 */
			if (succ == 0)	
			{
				EXEC SQL SELECT * INTO :acc_bal
					FROM t_acc_bal
					WHERE acc = :databody.acc;
				if (SQLCODE == 100)
				{
					succ = 1;
					strcpy(msg, "帐户不存在!");
					batlog("%sacc[%s][%d]  [%s][%d]", msg,databody.acc, i, __FILE__, __LINE__);
				}
				else if (SQLCODE < 0)
				{
					succ = 1;
					sprintf(msg, "数据库操作错![%d]", SQLCODE);
					batlog("%sacc[%s][%d]  [%s][%d]", msg,databody.acc, i, __FILE__, __LINE__);
				}
			}

			/* 检查户名 */
			if (succ == 0)
			{
				rtrim(acc_bal.acc_name);
				rtrim(databody.acc_name);
				if (strcmp(acc_bal.acc_name, databody.acc_name) != 0)
				{
					succ = 1;
					sprintf(msg, "帐户户名不符!");
					batlog("%sacc[%s][%d]  [%s][%d]", msg,databody.acc, i, __FILE__, __LINE__);
				}
			}

			/* 检查帐户状态 */
			if (succ == 0)
			{
				if(acc_bal.flag[0] != '0')
				{
					succ = 1;
					sprintf(msg, "帐户状态不正常!");
					batlog("%sacc[%s][%d]  [%s][%d]", msg,databody.acc, i, __FILE__, __LINE__);
				}
			}

			if ( succ == 0 )
			{ /* 检查成功 */
				suc_num++;
				suc_amt += databody.amt;
				bal -= databody.amt;
				/* 扣减内部帐 */
				EXEC SQL UPDATE t_acc_bal SET bal = bal - :databody.amt
					WHERE acc = :datahead.inn_acc;
				if (SQLCODE || sqlca.sqlerrd[2] != 1)
				{
					batlog("更新帐户余额表失败acc[%s]sqlcode[%d][%d]  [%s][%d]", datahead.inn_acc, SQLCODE, i, __FILE__, __LINE__);
					EXEC SQL ROLLBACK WORK;
					fclose(fp);
					return (-1);
				}
				/* 增加客户帐 */
				EXEC SQL UPDATE t_acc_bal SET bal = bal + :databody.amt
					WHERE acc = :databody.acc;
				if (SQLCODE || sqlca.sqlerrd[2] != 1)
				{
					batlog("更新帐户余额表失败acc[%s]sqlcode[%d][%d]  [%s][%d]", databody.acc, SQLCODE, i, __FILE__, __LINE__);
					EXEC SQL ROLLBACK WORK;
					fclose(fp);
					return (-1);
				}
				/* 更新标志 */
				EXEC SQL UPDATE t_agt_payret SET remark = :msg, flag = '0'
            		WHERE acc = :databody.acc
            		AND corp_name = :datahead.corp_name
					AND tran_date = :datahead.tran_date;
				if (SQLCODE || sqlca.sqlerrd[2] != 1)
				{
					batlog("更新结果表失败acc[%s]sqlcode[%d][%d]  [%s][%d]", databody.acc, SQLCODE, i, __FILE__, __LINE__);
					EXEC SQL ROLLBACK WORK;
					fclose(fp);
					return (-1);
				}

			}
			else
			{ /* 检查失败 */
				bad_num++;
				bad_amt += databody.amt;
				EXEC SQL UPDATE t_agt_payret SET remark = :msg
            		WHERE acc = :databody.acc
            		AND corp_name = :datahead.corp_name
					AND tran_date = :datahead.tran_date;
				if (SQLCODE || sqlca.sqlerrd[2] != 1)
				{
					batlog("更新结果表失败acc[%s]sqlcode[%d][%d]  [%s][%d]", databody.acc, SQLCODE, i, __FILE__, __LINE__);
					EXEC SQL ROLLBACK WORK;
					fclose(fp);
					return (-1);
				}
			}
		}
		else if (SQLCODE == 100)
		{ /* 未处理过 */

			/* 判断余额是否足够 */
			if (bal < databody.amt)
			{
				succ = 1;
				strcpy(msg, "内部帐户余额不足!");
				batlog("%sacc[%s][%d]  [%s][%d]", msg,databody.acc, i, __FILE__, __LINE__);
			}
			
			/* 查询帐户状态 */
			if (succ == 0)	
			{
				EXEC SQL SELECT * INTO :acc_bal
					FROM t_acc_bal
					WHERE acc = :databody.acc;
				if (SQLCODE == 100)
				{
					succ = 1;
					strcpy(msg, "帐户不存在!");
					batlog("%sacc[%s][%d]  [%s][%d]", msg,databody.acc, i, __FILE__, __LINE__);
				}
				else if (SQLCODE < 0)
				{
					succ = 1;
					sprintf(msg, "数据库操作错![%d]", SQLCODE);
					batlog("%sacc[%s][%d]  [%s][%d]", msg,databody.acc, i, __FILE__, __LINE__);
				}
			}

			/* 检查户名 */
			if (succ == 0)
			{
				rtrim(acc_bal.acc_name);
				rtrim(databody.acc_name);
				if (strcmp(acc_bal.acc_name, databody.acc_name) != 0)
				{
					succ = 1;
					sprintf(msg, "帐户户名不符!");
					batlog("%sacc[%s][%d]  [%s][%d]", msg,databody.acc, i, __FILE__, __LINE__);
				}
			}

			/* 检查帐户状态 */
			if (succ == 0)
			{
				if(acc_bal.flag[0] != '0')
				{
					succ = 1;
					sprintf(msg, "帐户状态不正常!");
					batlog("%sacc[%s][%d]  [%s][%d]", msg,databody.acc, i, __FILE__, __LINE__);
				}
			}

			if ( succ == 0 )
			{ /* 检查成功 */
				suc_num++;
				suc_amt += databody.amt;
				bal -= databody.amt;
				/* 扣减内部帐 */
				EXEC SQL UPDATE t_acc_bal SET bal = bal - :databody.amt
					WHERE acc = :datahead.inn_acc;
				if (SQLCODE || sqlca.sqlerrd[2] != 1)
				{
					batlog("更新帐户余额表失败acc[%s]sqlcode[%d][%d]  [%s][%d]", datahead.inn_acc, SQLCODE, i, __FILE__, __LINE__);
					EXEC SQL ROLLBACK WORK;
					fclose(fp);
					return (-1);
				}
				/* 增加客户帐 */
				EXEC SQL UPDATE t_acc_bal SET bal = bal + :databody.amt
					WHERE acc = :databody.acc;
				if (SQLCODE || sqlca.sqlerrd[2] != 1)
				{
					batlog("更新帐户余额表失败acc[%s]sqlcode[%d][%d]  [%s][%d]", databody.acc, SQLCODE, i, __FILE__, __LINE__);
					EXEC SQL ROLLBACK WORK;
					fclose(fp);
					return (-1);
				}
				ret = insert_payret(&datahead, &databody, 0, msg);
				if ( ret != 0 )
				{
					batlog("插入结果表失败acc[%s]sqlcode[%d][%d]  [%s][%d]", databody.acc, SQLCODE, i, __FILE__, __LINE__);
					EXEC SQL ROLLBACK WORK;
					fclose(fp);
					return (-1);
				}
			}
			else
			{ /* 检查失败 */
				bad_num++;
				bad_amt += databody.amt;
				ret = insert_payret(&datahead, &databody, 1, msg);
				if ( ret != 0 )
				{
					batlog("插入结果表失败acc[%s]sqlcode[%d][%d]  [%s][%d]", databody.acc, SQLCODE, i, __FILE__, __LINE__);
					EXEC SQL ROLLBACK WORK;
					fclose(fp);
					return (-1);
				}
			}
		}
		else
		{ /* 查询失败 */
			batlog("查询结果表失败acc[%s]sqlcode[%d][%d]  [%s][%d]", databody.acc, SQLCODE, i, __FILE__, __LINE__);
			EXEC SQL ROLLBACK WORK;
			fclose(fp);
			return (-1);
		}
	}

	fclose(fp);

	EXEC SQL COMMIT WORK;
	if (SQLCODE)
	{
		batlog("提交事务失败!sqlcode[%d]  [%s][%d]", SQLCODE, __FILE__, __LINE__);