insertcustomer.cpp

这是一个简单的酒店管理系统

// INSERTCUSTOMER.cpp : implementation file
//

#include "stdafx.h"
#include "shujuku.h"
#include "INSERTCUSTOMER.h"
#include "CusSet.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CINSERTCUSTOMER dialog


CINSERTCUSTOMER::CINSERTCUSTOMER(CWnd* pParent /*=NULL*/)
	: CDialog(CINSERTCUSTOMER::IDD, pParent)
{
	//{{AFX_DATA_INIT(CINSERTCUSTOMER)
	m_CusName = _T("");
	m_CusID = _T("");
	//}}AFX_DATA_INIT
}


void CINSERTCUSTOMER::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CINSERTCUSTOMER)
	DDX_Text(pDX, IDC_CusName, m_CusName);
	DDX_Text(pDX, IDC_CusID, m_CusID);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CINSERTCUSTOMER, CDialog)
	//{{AFX_MSG_MAP(CINSERTCUSTOMER)
	ON_BN_CLICKED(IDC_Insert, OnInsert)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CINSERTCUSTOMER message handlers

void CINSERTCUSTOMER::OnInsert() 
{
	// TODO: Add your control notification handler code here
	CString id;
	CRecordset reAgent(&dbAgent);

	this->UpdateData(TRUE);
	if(!IsLeagle())
	{
		this->MessageBox("Error!");
		return ;
	}
	CCusSet set;
	if(!set.IsOpen())
		set.Open();
	set.AddNew();
	set.m_ID = m_CusID ;
	set.m_Name = m_CusName ;

	try
	{
		reAgent.Open(CRecordset::forwardOnly,"SELECT ID FROM TCustomer WHERE ID = "+ m_CusID,CRecordset::readOnly);	
	}
	catch(CException e)
	{
		e.ReportError();
	}
	
	while(!reAgent.IsEOF())
	{
		reAgent.GetFieldValue("ID",id);
		reAgent.MoveNext();
	}
	reAgent.Close();

	if(id == m_CusID)
	{
		MessageBox("ID已存在!");
	}
	else
	{
		set.Update();
		this->MessageBox("Add Success!");
	}

	set.Close();
}

BOOL CINSERTCUSTOMER::IsLeagle()
{
	if(m_CusID.GetLength()!=11)
		return 0;
	if(m_CusName.IsEmpty())
		return 0;
	return 1;
}