arxdatabasedlg.cpp

如何用利OLEDB连接数据库

// ArxDatabaseDlg.cpp : implementation file
//

#include "stdafx.h"
#include "ArxDatabase.h"
#include "ArxDatabaseDlg.h"

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

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About
extern CArxDatabaseApp theApp;

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

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

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

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

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CArxDatabaseDlg dialog

CArxDatabaseDlg::CArxDatabaseDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CArxDatabaseDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CArxDatabaseDlg)
	m_CompKind = _T("");
	m_CompName = _T("");
	m_CompStyle = _T("");
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CArxDatabaseDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CArxDatabaseDlg)
	DDX_Control(pDX, IDC_LIST, m_List);
	DDX_Text(pDX, IDC_COMPKIND, m_CompKind);
	DDX_Text(pDX, IDC_COMPNAME, m_CompName);
	DDX_Text(pDX, IDC_COMPSTYLE, m_CompStyle);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CArxDatabaseDlg, CDialog)
	//{{AFX_MSG_MAP(CArxDatabaseDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_READCOMP, OnReadcomp)
	ON_BN_CLICKED(IDC_WRITECOMP, OnWritecomp)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CArxDatabaseDlg message handlers

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

	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	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);
		}
	}

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon
	
	// TODO: Add extra initialization here
	    m_pRecordset.CreateInstance(__uuidof(Recordset));

	// 在ADO操作中建议语句中要常用try...catch()来捕获错误信息，
	// 因为它有时会经常出现一些想不到的错误。
	try
	{
		m_pRecordset->Open("SELECT * FROM Component",                // 查询DemoTable表中所有字段
							theApp.m_pConnection.GetInterfacePtr(),	 // 获取库接库的IDispatch指针
							adOpenDynamic,
							adLockOptimistic,
							adCmdText);
	}
	catch(_com_error *e)
	{
		AfxMessageBox(e->ErrorMessage());
	}
	
	return TRUE;  // return TRUE  unless you set the focus to a control
}

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

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CArxDatabaseDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle
		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;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CArxDatabaseDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}


void CArxDatabaseDlg::OnReadcomp() 
{
	// TODO: Add your control notification handler code here
	_variant_t var;
	CString strCompID,strCompName,strCompKind,strCompStyle,strCompPic,strCompDwg;

	// 清空列表框
	m_List.ResetContent();
	strCompID=strCompName=strCompKind=strCompStyle=strCompPic=strCompDwg="";

	try
	{
		if(!m_pRecordset->BOF)
			m_pRecordset->MoveFirst();
		else
		{
			AfxMessageBox("表内数据为空");
			return;
		}

		// 读入库中各字段并加入列表框中
		while(!m_pRecordset->adoEOF)
		{
			var = m_pRecordset->GetCollect("CompName");
			if(var.vt != VT_NULL)
				strCompName = (LPCSTR)_bstr_t(var);
			var = m_pRecordset->GetCollect("CompKind");
			if(var.vt != VT_NULL)
				strCompKind = (LPCSTR)_bstr_t(var);
			var=m_pRecordset->GetCollect("CompStyle");
			if(var.vt!=VT_NULL)
				strCompKind=(LPCSTR)_bstr_t(var);
			var=m_pRecordset->GetCollect("CompID");
			if(var.vt!=VT_NULL)
				strCompID=(LPCSTR)_bstr_t(var);
			
			//读取dwg文件
			try
			{
				CString strFileNameDwg;
                strFileNameDwg.Format("%s.dwg",strCompName);
				char tmpPath1[_MAX_PATH+1];
				GetCurrentDirectory(_MAX_PATH,tmpPath1);
				strFileNameDwg.Insert(0,tmpPath1);
				CFile outFile1(strFileNameDwg,CFile::modeCreate|CFile::modeWrite);
				long nSize1 = m_pRecordset->GetFields()->GetItem("DWG")->ActualSize;
				if(nSize1 > 0)
				{
					_variant_t	varBLOB1;
					varBLOB1 = m_pRecordset->GetFields()->GetItem("DWG")->GetChunk(nSize1);
					if(varBLOB1.vt == (VT_ARRAY | VT_UI1))
					{
						if(BYTE *pBuffer1 = new BYTE [nSize1+1])		///重新申请必要的存储空间
						{	
							char *pBuf1 = NULL;
							SafeArrayAccessData(varBLOB1.parray,(void **)&pBuf1);
							memcpy(pBuffer1,pBuf1,nSize1);				
							
							///复制数据到缓冲区m_pBMPBuffer
							SafeArrayUnaccessData (varBLOB1.parray);
							//	int nSize = lDataSize;
							//(m_Pic.LoadPictureData(pBuffer, nSize));
							outFile1.WriteHuge(pBuffer1,nSize1);
							
							delete [] pBuffer1;
							pBuf1=0;
							
						}
					}
				}
			}
			catch (_com_error e) {
			}
			
			//读取jpg文件
			try
			{
				CString strFileNameJpg;
                strFileNameJpg.Format("%s.jpg",strCompName);
				char tmpPath[_MAX_PATH+1];
				GetCurrentDirectory(_MAX_PATH,tmpPath);
				strFileNameJpg.Insert(0,tmpPath);
				CFile outFile(strFileNameJpg,CFile::modeCreate|CFile::modeWrite);
				long nSize = m_pRecordset->GetFields()->GetItem("Picture")->ActualSize;
				if(nSize > 0)
				{
					_variant_t	varBLOB;
					varBLOB = m_pRecordset->GetFields()->GetItem("Picture")->GetChunk(nSize);
					if(varBLOB.vt == (VT_ARRAY | VT_UI1))
					{
						if(BYTE *pBuffer = new BYTE [nSize+1])		///重新申请必要的存储空间
						{	
							char *pBuf = NULL;
							SafeArrayAccessData(varBLOB.parray,(void **)&pBuf);
							memcpy(pBuffer,pBuf,nSize);				
							
							///复制数据到缓冲区m_pBMPBuffer
							SafeArrayUnaccessData (varBLOB.parray);
							//	int nSize = lDataSize;
							//(m_Pic.LoadPictureData(pBuffer, nSize));
							outFile.WriteHuge(pBuffer,nSize);
							
							delete [] pBuffer;
							pBuf=0;
							
						}
					}
				}
			}
			catch (_com_error e) {
			}



			m_List.AddString(strCompID+"->"+strCompName+"->"+strCompKind+"->"+strCompStyle);

			m_pRecordset->MoveNext();
		}

		// 默认列表指向第一项，同时移动记录指针并显示
		m_List.SetCurSel(0);
	}
	catch(_com_error *e)
	{
		AfxMessageBox(e->ErrorMessage());
	}
}
	


void CArxDatabaseDlg::OnWritecomp() 
{
	// TODO: Add your control notification handler code here
	UpdateData();
	if(m_CompName=="")
	{
		AfxMessageBox("部件名称不能为空！");
		return;
	}
	try
	{
		//写入字段值
		m_pRecordset->AddNew();
		m_pRecordset->PutCollect("CompName",_variant_t(m_CompName));
		m_pRecordset->PutCollect("CompKind",_variant_t(m_CompKind));
		m_pRecordset->PutCollect("CompStyle",_variant_t(m_CompStyle));
		
		//在数据库中加入dwg文件
        CFile f1;
		CString  FilePathName1;
		CFileException e1;
		
		CFileDialog dlg1(TRUE,NULL,NULL,0,"dwg Files (*.dwg)|*.dwg||",this);///TRUE为OPEN对话框，FALSE为SAVE AS对话框
		if(dlg1.DoModal()==IDOK)
		{FilePathName1=dlg1.GetPathName();
		
		if(m_Pic1.m_IPicture != NULL) m_Pic1.FreePictureData(); // Important - Avoid Leaks...
		
		if(f1.Open(FilePathName1, CFile::modeRead | CFile::typeBinary, &e1)) //打开了一个dwg文件
		{	
			
			int nSize1 = f1.GetLength();          //先得到dwg文件长度
			BYTE * pBuffer1 = new BYTE [nSize1];  //按文件的大小在堆上申请一块内存
			
			if (f1.Read(pBuffer1, nSize1) > 0 )    //把dwg文件读到pBuffer(堆上申请一块内存)
			{   // +----------------------------------------------
				BYTE *pBuf1 = pBuffer1;     ///下面这一大段是把pBuffer里的DWG数据放到库中
				VARIANT			varBLOB1;
				SAFEARRAY		*psa1;
				SAFEARRAYBOUND	rgsabound1[1];
				
				if(pBuf1)
				{    
					rgsabound1[0].lLbound = 0;
					rgsabound1[0].cElements = nSize1;
					psa1 = SafeArrayCreate(VT_UI1, 1, rgsabound1);
					for (long i = 0; i < (long)nSize1; i++)
						SafeArrayPutElement (psa1, &i, pBuf1++);
					varBLOB1.vt = VT_ARRAY | VT_UI1;
					varBLOB1.parray = psa1;
					m_pRecordset->GetFields()->GetItem("DWG")->AppendChunk(varBLOB1);
				}				
				
				(m_Pic1.LoadPictureData(pBuffer1, nSize1));//接作调用函数读pBuffer的jpg数据准备显示
				delete [] pBuffer1;     //删掉堆上申请的那一块内存
				pBuf1=0;                //以防二次乱用
			}
			f1.Close();	
		}
		
		}
		
		
		//在数据库中加入jpg文件
		CFile f;
		CString  FilePathName;
		CFileException e;
		
		CFileDialog dlg(TRUE,NULL,NULL,0,"jpg Files (*.jpg)|*.jpg||",this);///TRUE为OPEN对话框，FALSE为SAVE AS对话框
		if(dlg.DoModal()==IDOK)
		{FilePathName=dlg.GetPathName();
		
		if(m_Pic.m_IPicture != NULL) m_Pic.FreePictureData(); // Important - Avoid Leaks...
		
		if(f.Open(FilePathName, CFile::modeRead | CFile::typeBinary, &e)) //打开了一个jpg文件
		{	
			
			int nSize = f.GetLength();          //先得到jpg文件长度
			BYTE * pBuffer = new BYTE [nSize];  //按文件的大小在堆上申请一块内存
			
			if (f.Read(pBuffer, nSize) > 0 )    //把jpg文件读到pBuffer(堆上申请一块内存)
			{   // +----------------------------------------------
				BYTE *pBuf = pBuffer;     ///下面这一大段是把pBuffer里的jpg数据放到库中
				VARIANT			varBLOB;
				SAFEARRAY		*psa;
				SAFEARRAYBOUND	rgsabound[1];
				
				if(pBuf)
				{    
					rgsabound[0].lLbound = 0;
					rgsabound[0].cElements = nSize;
					psa = SafeArrayCreate(VT_UI1, 1, rgsabound);
					for (long i = 0; i < (long)nSize; i++)
						SafeArrayPutElement (psa, &i, pBuf++);
					varBLOB.vt = VT_ARRAY | VT_UI1;
					varBLOB.parray = psa;
					m_pRecordset->GetFields()->GetItem("Picture")->AppendChunk(varBLOB);
				}				
				
				(m_Pic.LoadPictureData(pBuffer, nSize));//接作调用函数读pBuffer的jpg数据准备显示
				delete [] pBuffer;     //删掉堆上申请的那一块内存
				pBuf=0;                //以防二次乱用
			}
			f.Close();	
		}
		
		}
		
		
		
		m_pRecordset->Update();
		
		AfxMessageBox("插入成功！");
		//更新显示其库内容
		int nCurSel=m_List.GetCurSel();
		OnReadcomp();
		m_List.SetCurSel(nCurSel);

		
	}
	catch(_com_error *e)
	{
		AfxMessageBox(e->ErrorMessage());
	}
	
}
 
	

BOOL CArxDatabaseDlg::DestroyWindow() 
{
	// 关闭记录集
	m_pRecordset->Close();
	m_pRecordset = NULL;
	
	return CDialog::DestroyWindow();
}