flightplandlg.cpp

使用Visual C++ .NET实现简单了一个票务管理系统

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

#include "stdafx.h"
#include "FlightPlan.h"
#include "FlightPlanDlg.h"
#include ".\flightplandlg.h"
#include "DBSetDlg.h"
#include "Tools.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// 对话框数据
	enum { IDD = IDD_ABOUTBOX };

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()


// CFlightPlanDlg 对话框



CFlightPlanDlg::CFlightPlanDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CFlightPlanDlg::IDD, pParent)
	, m_Fi(_T(""))
	, m_An(_T(""))
	, m_Length(0)
	, m_StartCity(_T(""))
	, m_StartAir(_T(""))
	, m_EndCity(_T(""))
	, m_EndAir(_T(""))
	, m_MaxPassenger(0)
	, m_Passanger(0)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
	char c[256];
	CString temp;
	::GetCurrentDirectory( 256, c );
	temp.Format("%s\\", c );
	m_IniFile = temp + "FlightPlane.ini";
	m_Index = NULL;
	m_CurSlt = -1;
	m_Sql = "";
}
CFlightPlanDlg::~CFlightPlanDlg()
{
	SAFE_DELARRAY(m_Index);
}
void CFlightPlanDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_FIINFOLIST, m_FiInfo);
	DDX_Control(pDX, IDC_DATETIMEPICKERSTR, m_SltTime);
	DDX_Control(pDX, IDC_DATETIMEPICKER1, m_FlightDate);
	DDX_Text(pDX, IDC_EDIT1, m_Fi);
	DDX_Text(pDX, IDC_EDIT2, m_An);
	DDX_Text(pDX, IDC_EDIT3, m_Length);
	DDX_Text(pDX, IDC_EDIT4, m_StartCity);
	DDX_Text(pDX, IDC_EDIT5, m_StartAir);
	DDX_Text(pDX, IDC_EDIT6, m_EndCity);
	DDX_Text(pDX, IDC_EDIT7, m_EndAir);
	DDX_Control(pDX, IDC_DATETIMEPICKER2, m_StartTime);
	DDX_Control(pDX, IDC_DATETIMEPICKER3, m_EndTime);
	DDX_Text(pDX, IDC_EDIT14, m_MaxPassenger);
	DDX_Text(pDX, IDC_EDIT15, m_Passanger);
}

BEGIN_MESSAGE_MAP(CFlightPlanDlg, CDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(IDC_BTDBSET, OnBnClickedBtdbset)
	ON_BN_CLICKED(IDC_BTSLT, OnBnClickedBtslt)
	ON_BN_CLICKED(IDOK, OnBnClickedOk)
	ON_BN_CLICKED(IDC_BTADDFI, OnBnClickedBtaddfi)
	ON_NOTIFY(LVN_ITEMCHANGED, IDC_FIINFOLIST, OnLvnItemchangedFiinfolist)
	ON_BN_CLICKED(IDC_BTADDFI2, OnBnClickedBtaddfi2)
	ON_BN_CLICKED(IDC_BTADDFI3, OnBnClickedBtaddfi3)
END_MESSAGE_MAP()


// CFlightPlanDlg 消息处理程序

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

	// 读去配置参数
	LoadDBSet();

	// IDM_ABOUTBOX 必须在系统命令范围内。
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

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

	// TODO: 在此添加额外的初始化代码
	Init();
	
	return TRUE;  // 除非设置了控件的焦点，否则返回 TRUE
}

void CFlightPlanDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

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

void CFlightPlanDlg::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 CFlightPlanDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

void CFlightPlanDlg::OnBnClickedBtdbset()
{
	CDBSetDlg dlg;
	dlg.SetData(m_DBData.type,m_DBData.Source,m_DBData.UserName,m_DBData.PassWord,m_IniFile );
	if( dlg.DoModal() == IDOK )
	{
		LoadDBSet();
	}
}
// 读去数据库配置参数
void CFlightPlanDlg::LoadDBSet()
{
	char tmp[256];
	GetPrivateProfileString("DBSET","DBTYPE","0",tmp,256,m_IniFile);
	m_DBData.type = atol(tmp);
	GetPrivateProfileString("DBSET","DBSOURCE","",tmp,256,m_IniFile);
	m_DBData.Source = tmp;
	m_DBData.Source.Trim();
	GetPrivateProfileString("DBSET","USERNAME","",tmp,256,m_IniFile);
	m_DBData.UserName = tmp;
	m_DBData.UserName.Trim();
	CTools * ptool = NULL;
	GetPrivateProfileString("DBSET","PASSWORD","",tmp,256,m_IniFile);
	m_DBData.PassWord = ptool->GetPriString( tmp, false );

	if( m_DBData.Source != "" )
	{
		CString jet = "";
		switch( m_DBData.type )
		{
		case 0: // Access
			jet = "Microsoft.Jet.OLEDB.4.0";
			break;
		case 1: // Oracle
			jet = "MSDAORA";
			break;
		default:
			jet = "Microsoft.Jet.OLEDB.4.0";
			break;
		}
		m_DBData.Linkbuf.Format( "Provider=%s;Data Source=%s",jet,m_DBData.Source);
	}
}
// 初始化配置
void CFlightPlanDlg::Init()
{
	m_FiInfo.SetExtendedStyle( LVS_EX_GRIDLINES | LVS_EX_FULLROWSELECT | LVS_EX_HEADERDRAGDROP );
	m_FiInfo.InsertColumn(0,"航班号",LVCFMT_LEFT,50);
	m_FiInfo.InsertColumn(1,"日期",LVCFMT_CENTER,80);
	m_FiInfo.InsertColumn(2,"始发城市",LVCFMT_LEFT,61);
	m_FiInfo.InsertColumn(3,"目的城市",LVCFMT_LEFT,61);
	m_FiInfo.InsertColumn(4,"起飞机场",LVCFMT_LEFT,61);
	m_FiInfo.InsertColumn(5,"降落机场",LVCFMT_LEFT,61);
	m_FiInfo.InsertColumn(6,"起飞时间",LVCFMT_LEFT,61);
	m_FiInfo.InsertColumn(7,"降落时间",LVCFMT_LEFT,61);
	m_FiInfo.InsertColumn(8,"机尾号",LVCFMT_LEFT,61);
	m_FiInfo.InsertColumn(9,"最大人数",LVCFMT_LEFT,61);
	m_FiInfo.InsertColumn(10,"实际人数",LVCFMT_LEFT,61);
	m_FiInfo.InsertColumn(11,"距离",LVCFMT_LEFT,61);

	m_StartTime.SetFormat("HH:mm");
	m_EndTime.SetFormat("HH:mm");
	m_FlightDate.SetFormat("yyyy-hh-dd");

	LoadPlan();
}

void CFlightPlanDlg::OnBnClickedOk()
{
	// TODO: 在此添加控件通知处理程序代码
	OnOK();
}
// 添加航班
void CFlightPlanDlg::OnBnClickedBtaddfi()
{
	if( UpdateData() == false )
	{
		return ;
	}
	if( m_Fi == "" )
	{
		MessageBox("必须填写航班号");
		return;
	}
	if( AddNote() == false)
	{
		MessageBox("记录添加失败");
	}
	else
	{
		MessageBox("记录添加成功");
		LoadPlan();
	}
}
// 添加记录
bool CFlightPlanDlg::AddNote()
{
	//指向Connection对象的指针，用于连接到数据库
	_ConnectionPtr pConnection;
	//Recordset 指针， 记录集指针
	_RecordsetPtr pRecordset;

	try
	{
		HRESULT hr;

		// 创建Connection对象
		hr = pConnection.CreateInstance("ADODB.Connection");
		if( SUCCEEDED( hr ) )
		{
			// 连接数据库
			hr = pConnection->Open(m_DBData.Linkbuf.GetBuffer(),
				m_DBData.UserName.GetBuffer(),m_DBData.PassWord.GetBuffer(),
				adModeUnknown);
		}
		else
		{
			MessageBox("创建数据库连接失败","失败",MB_OK);
			return false;
		}
	}
	catch( ... )
	{
		MessageBox( "连接数据库失败","失败",MB_OK);
		return false;
	}

	try
	{
		// 生成Recordset实例
		pRecordset.CreateInstance("ADODB.Recordset");

		// 察看全部
		pRecordset->Open("SELECT * FROM FlightIDInfo",   // 查询所有语句
			_variant_t((IDispatch*)pConnection,true),adOpenStatic,adLockOptimistic,adCmdText);
	}
	catch( _com_error * e )
	{
		pConnection->Close();
		MessageBox( e->ErrorMessage(),"异常",MB_OK );
		return false;
	}

	try
	{
		COleDateTime t,t1;
		CString temp;

		pRecordset->AddNew();
		m_FlightDate.GetTime( t1 );
		t.SetDate(t1.GetYear(),t1.GetMonth(),t1.GetDay());
		pRecordset->PutCollect( "Date", _variant_t(t) );
		pRecordset->PutCollect( "FlightID", _variant_t(m_Fi) );
		// 起飞时间
		m_StartTime.GetTime(t1);
		t.SetTime(t1.GetHour(),t1.GetMinute(),0);
		pRecordset->PutCollect( "TakeoffTime", _variant_t(t) );
		// 降落时间
		m_EndTime.GetTime(t);
		t.SetTime(t1.GetHour(),t1.GetMinute(),0);
		pRecordset->PutCollect( "LandingTime", _variant_t(t) );
		// 出发城市
		pRecordset->PutCollect( "TakeoffPlace", _variant_t(m_StartCity) );
		// 降落城市
		pRecordset->PutCollect( "LandingPlace", _variant_t(m_EndCity) );
		// 起飞机场
		pRecordset->PutCollect( "TakeoffAir", _variant_t(m_StartAir) );
		// 降落机场
		pRecordset->PutCollect( "LandingAir", _variant_t(m_EndAir) );

		pRecordset->PutCollect( "Plane", _variant_t(m_An) );
		pRecordset->PutCollect( "MaxPassengerNum", _variant_t(m_MaxPassenger) );
		pRecordset->PutCollect( "PassengerNum", _variant_t(m_Passanger) );
		pRecordset->PutCollect( "Distance", _variant_t(m_Length) );
		pRecordset->Update();
	}
	catch(...)
	{
		MessageBox("添加记录异常","异常",MB_OK );
		pRecordset->Close();
		pConnection->Close();
		return false;
	}

	try
	{
		pRecordset->Close();
		pConnection->Close();
	}
	catch( _com_error * e )
	{
		MessageBox(e->ErrorMessage(),"失败",MB_OK );
	}
	return true;
}

// 读入航班计划,写入List
bool CFlightPlanDlg::LoadPlan()
{
	if( m_Sql == "" )
	{
		// 得到当天的航班计划
		CString temp;
		SYSTEMTIME t;
		GetLocalTime(&t);
		temp.Format("DateSerial(%d,%d,%d)",t.wYear,t.wMonth,t.wDay );

		m_Sql.Format("SELECT * FROM FlightIDInfo WHERE (Date = %s)",temp);
	}
	_ConnectionPtr pConnection;
	_RecordsetPtr pRecordset;

	try
	{
		HRESULT hr;

		// 创建Connection对象
		hr = pConnection.CreateInstance("ADODB.Connection");
		if( SUCCEEDED( hr ) )
		{
			// 连接数据库
			hr = pConnection->Open(m_DBData.Linkbuf.GetBuffer(),
				m_DBData.UserName.GetBuffer(),m_DBData.PassWord.GetBuffer(),
				adModeUnknown);
		}
		else
		{
			MessageBox("创建数据库连接失败","失败",MB_OK);
			return false;
		}
	}
	catch( ... )
	{
		MessageBox( "连接数据库失败","失败",MB_OK);
		return false;
	}

	try
	{
		// 生成Recordset实例
		pRecordset.CreateInstance("ADODB.Recordset");

		pRecordset->Open( m_Sql.GetBuffer(),
			_variant_t((IDispatch*)pConnection,true),adOpenStatic,adLockOptimistic,adCmdText);
	}
	catch( _com_error * e )
	{
		pConnection->Close();
		MessageBox( e->ErrorMessage(),"异常",MB_OK );
		return false;
	}

	// 统计记录数
	m_CurNum = 0;
	try
	{
		if( !pRecordset->BOF )
			pRecordset->MoveFirst();

		while( !pRecordset->adoEOF )
		{
			m_CurNum++;
			pRecordset->MoveNext();
		}
	}
	catch(...)
	{
		MessageBox("读记录计数异常","异常",MB_OK );
		return false;
	}
	m_FiInfo.DeleteAllItems();

	if( m_CurNum > 0 )
	{
		_variant_t value;
		COleDateTime t;
		CString temp;
		try
		{
			SAFE_DELARRAY(m_Index);
			m_Index = new long[m_CurNum];

			if( !pRecordset->BOF )
				pRecordset->MoveFirst();
			for( int i = 0; i < m_CurNum; i++ )
			{