Windows编程

开发平台：
Visual C++

DBDOC.CPP：源码内容
							// DBDoc.cpp : implementation of the CDBViewDoc class
//
// This is a part of the Microsoft Foundation Classes and
// Templates (MFC&T).
// Copyright (C) 1998 Microsoft Corporation
// All rights reserved.
//
// This source code is only intended as a supplement to the
// MFC&T Reference and related electronic documentation provided
// with the library.  See these sources for detailed information
// regarding the MFC&T product.
//
#include "stdafx.h"
#include "DBViewer.h"
#include "DBDoc.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
/////////////////////////////////////////////////////////////////////////////
// CDBViewDoc
IMPLEMENT_DYNCREATE(CDBViewDoc, CDocument)
BEGIN_MESSAGE_MAP(CDBViewDoc, CDocument)
	//{{AFX_MSG_MAP(CDBViewDoc)
	ON_COMMAND(ID_FILE_OPEN, OnFileOpen)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CDBViewDoc construction/destruction
CDBViewDoc::CDBViewDoc()
{
	m_pListView = NULL;
	m_pTreeView = NULL;
}
CDBViewDoc::~CDBViewDoc()
{
	m_pListView = NULL;
	m_pTreeView = NULL;
}
BOOL CDBViewDoc::OnNewDocument()
{
	if (!CDocument::OnNewDocument())
		return FALSE;
	// TODO: add reinitialization code here
	// (SDI documents will reuse this document)
	return TRUE;
}
/////////////////////////////////////////////////////////////////////////////
// CDBViewDoc serialization
void CDBViewDoc::Serialize(CArchive& ar)
{
	if (ar.IsStoring())
	{
		// TODO: add storing code here
	}
	else
	{
		// TODO: add loading code here
	}
}
/////////////////////////////////////////////////////////////////////////////
// CDBViewDoc diagnostics
#ifdef _DEBUG
void CDBViewDoc::AssertValid() const
{
	CDocument::AssertValid();
}
void CDBViewDoc::Dump(CDumpContext& dc) const
{
	CDocument::Dump(dc);
}
#endif //_DEBUG
/////////////////////////////////////////////////////////////////////////////
// CDBViewDoc commands
void CDBViewDoc::RefreshViews()
{
	if (m_pListView)
		m_pListView->EraseList();
	if (m_pTreeView)
		m_pTreeView->PopulateTree();
}
BOOL CDBViewDoc::OnOpenDocument(LPCTSTR lpszPathName)
{
	if (m_Session.m_spOpenRowset != NULL)
		m_Session.m_spOpenRowset.Release();
	m_strConnect = _T("");
	if (m_Connect.Open(AfxGetMainWnd()->GetSafeHwnd()) != S_OK)
	{
		AfxMessageBox(_T("Unable to connect to data source"));
		return FALSE;
	}
	else
	{
		USES_CONVERSION;
		if (m_Session.Open(m_Connect) != S_OK)
		{
			AfxMessageBox(_T("Couldn't create session on data source"));
			return FALSE;
		}
		CComVariant var;
		m_Connect.GetProperty(DBPROPSET_DATASOURCEINFO, DBPROP_DATASOURCENAME, &var);
		m_strConnect = OLE2T(var.bstrVal);
		m_pTreeView->m_pSession = &m_Session;
		m_pListView->m_pSession = &m_Session;
		RefreshViews();
		return TRUE;
	}
	return FALSE;
}
CString CDBViewDoc::GetDSN()
{
	// pull DSN from database connect string
	CString string = m_strConnect;
//  string = string.Right(string.GetLength() - (string.Find(_T("DSN=")) + 4));
//  string = string.Left(string.Find(_T(";")));
	return string;
}
CString CDBViewDoc::GetConnect()
{
	return m_strConnect;
}
void CDBViewDoc::OnFileOpen()
{
	OnOpenDocument(NULL);
}
CDataSource* CDBViewDoc::GetDataSource()
{
	return &m_Connect;
}

						