cgetsqlfromdbdlg.cpp

自定义格式读取mysql数据库记录。用于批量处理mysql数据库。

// CGetSqlFromDbDlg.cpp : 实现文件
//

#include "stdafx.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// CCGetSqlFromDbDlg 对话框



CCGetSqlFromDbDlg::CCGetSqlFromDbDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CCGetSqlFromDbDlg::IDD, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
	mysql_init(&m_mysql);
}

CCGetSqlFromDbDlg::~CCGetSqlFromDbDlg()
{
	mysql_close(&m_mysql);
	if (m_pDlg != NULL) {
		delete m_pDlg;
		m_pDlg = NULL;
	}
}

void CCGetSqlFromDbDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_EDIT_FORMAT, m_edtFormat);
	DDX_Control(pDX, IDC_EDIT_FROM, m_edtFrom);
	DDX_Control(pDX, IDC_EDIT_KEY, m_edtKey);
	DDX_Control(pDX, IDC_EDIT_PER, m_edtPer);
	DDX_Control(pDX, IDC_EDIT_PRO, m_edtPro);
	DDX_Control(pDX, IDC_EDIT_TABLE, m_edtTable);
	DDX_Control(pDX, IDC_EDIT_TO, m_edtTo);
	DDX_Control(pDX, IDC_RADIO_ALL, m_rdoAll);
	DDX_Control(pDX, IDC_RADIO_KEY, m_rdoKey);
}

BEGIN_MESSAGE_MAP(CCGetSqlFromDbDlg, CDialog)
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(IDOK, OnBnClickedOk)
	ON_BN_CLICKED(IDC_RADIO_ALL, OnBnClickedRadioAll)
	ON_BN_CLICKED(IDC_RADIO_KEY, OnBnClickedRadioAll)
END_MESSAGE_MAP()


// CCGetSqlFromDbDlg 消息处理程序

BOOL CCGetSqlFromDbDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	// TODO: 在此添加额外的初始化代码

	CSettingDlg dlg;
DB:	if (dlg.DoModal() == IDOK) {
		if (!mysql_real_connect(&m_mysql, dlg.m_sAddr, dlg.m_sUser, dlg.m_sPswd, dlg.m_sName, atoi(dlg.m_sPort), NULL, 0)) {
			AfxMessageBox("数据库连接出错！");
			goto DB;
		}
	} else {
		PostQuitMessage(0);
		return FALSE;
	}
	mysql_query(&m_mysql, "SET NAMES 'utf8'");

	m_rdoAll.SetCheck(1);
	m_edtPer.SetWindowText("10000");
	SwitchInput();

	m_pDlg = new CProcessDlg;
	m_pDlg->Create(IDD_DIALOG_PROCESS, NULL);
	m_pDlg->CenterWindow();
	
	return TRUE;  // 除非设置了控件的焦点，否则返回 TRUE
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CCGetSqlFromDbDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于绘制的设备上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使图标在工作矩形中居中
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// 绘制图标
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标显示。
HCURSOR CCGetSqlFromDbDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

void CCGetSqlFromDbDlg::SwitchInput(void)
{
	BOOL bEnable = TRUE;
	if (m_rdoAll.GetCheck() != 0) {
		bEnable = FALSE;
	}
	m_edtPro.EnableWindow(bEnable);
	m_edtFrom.EnableWindow(bEnable);
	m_edtTo.EnableWindow(bEnable);
}

void CCGetSqlFromDbDlg::OnBnClickedOk()
{
	// TODO: 在此添加控件通知处理程序代码
	GetValueFromControl();
	if (m_sTable.IsEmpty() || m_sKey.IsEmpty()) {
		AfxMessageBox("填写不正确！");
		return;
	}
	stringstream sql;
	sql << "SELECT " << m_sKey << " FROM " << m_sTable;
	if (m_rdoAll.GetCheck() == 0)
		sql << " WHERE " << m_sPro << ">=" << m_sFrom << " and " << m_sPro << "<=" << m_sTo;
	mysql_query(&m_mysql, sql.str().c_str());
	m_pDlg->ShowWindow(SW_SHOW);
	m_pDlg->SetRange(0, 100);
	AfxBeginThread(GetSqlTxt, (void *)this);
	//OnOK();
}

void CCGetSqlFromDbDlg::OnBnClickedRadioAll()
{
	// TODO: 在此添加控件通知处理程序代码
	SwitchInput();
}

void CCGetSqlFromDbDlg::GetValueFromControl(void)
{
	m_edtFormat.GetWindowText(m_sFormat);
	m_edtKey.GetWindowText(m_sKey);
	m_edtPer.GetWindowText(m_sPer);
	m_edtTable.GetWindowText(m_sTable);
	if (m_rdoAll.GetCheck() == 0) {
		m_edtFrom.GetWindowText(m_sFrom);
		m_edtPro.GetWindowText(m_sPro);
		m_edtTo.GetWindowText(m_sTo);
	}
	m_nCols = 0;
	for (int i = 0; i < m_sKey.GetLength(); i ++) {
		if (m_sKey.GetAt(i) == ',') m_nCols ++;
	}
	m_nCols ++;
}

UINT CCGetSqlFromDbDlg::GetSqlTxt(LPVOID lparam)
{
	if (lparam == NULL) return 1;
	CCGetSqlFromDbDlg *getSql = (CCGetSqlFromDbDlg *)lparam;
	getSql->m_pQuery = mysql_store_result(&getSql->m_mysql);
	int nRows = (int)mysql_num_rows(getSql->m_pQuery);
	int ntxt = 1;
	int k = 0;
	FILE *fp = NULL;
	for (int i = 0; i < nRows; i++) {
		if (fp == NULL) {
			char file[100];
			sprintf(file, "%s%d.sql", getSql->m_sTable, ntxt);
			ntxt ++;
			getSql->m_pDlg->SetInfo(file);
			fp = fopen(file, "w");
		}
		getSql->m_pDlg->SetPosition(k * 100 / 1000);
		getSql->m_row = mysql_fetch_row(getSql->m_pQuery);
		CString sWrite = getSql->m_sFormat;
		for (int j = getSql->m_nCols - 1; j >= 0; j --) {
			char sOld[10];
			sprintf(sOld, "$%d", j+1);
			CString sTemp;
			sTemp.Format(getSql->m_row[j]);
			sTemp.Replace("\"", "\\\"");
			sWrite.Replace(sOld, sTemp);
		}
		sWrite.Replace("$tb", getSql->m_sTable);
		sWrite += "\n";
		if (fp != NULL)
			fwrite(sWrite, sizeof(char), sWrite.GetLength(), fp);
		k ++;
		if (k >= atoi(getSql->m_sPer)) {
			k = 0;
			if (fp != NULL)
				fclose(fp);
			fp = NULL;
		}
	}
	mysql_free_result(getSql->m_pQuery);
	if (fp != NULL)
		fclose(fp);
	fp = NULL;
	getSql->m_pDlg->ShowWindow(SW_HIDE);
	return 0;
}