NoteBook开发手记（二）

     NoteBook的重点也是难点的地方在数据库的使用。当我需要把编辑好的内容存储到数据库，当打开软件，读出数据的时候也需要访问数据库。所以和数据库的交互是本程序的精髓，然而在设计数据库的时候却遇到了麻烦。

 

    当需要浏览或者创建新的记录的时候就需要通过树型控件来进行选择，该树可以这样生成

 |--根类别

 |       |

 |       |----子类别                                                 

 |       |           |

 |       |           |----子记录

 |       |                 

 |       |----子记录

 |

 |--根记录

 

也就是说我们的数据库存储要根据树型的构造来存储数据，想了很久也得不到答案。后来，今天问了一个搞了10多年数据库的同事，终于得到了一种比较满意的答案。建立一张表，然后建一个字段“父ID”，每个子类别或者子记录对应一个ID号（这个ID号是自动的），根类别或者根记录的“父ID”就为0，也即空。这样就可以把整棵数的内容存储到数据库了，当然，还需要其他的一些字段如：标题、内容、日期，等等和记录相关的字段。

 

    好了，有了数据库，就可以封装一个类来进行数据库的添加、删除、修改操作了。利用ADO的连接方式，下面给出了这个类的具体代码：

1. #pragma once
2. #include <vector>
3. #include "StdAfx.h"
4. using std::vector;
8. // 数据成员
9. class Item
10. {
11. public:
12.     int     m_nID;              // ID
13.     CString m_strTitle;         // 标题   
14.     CString m_strTime;          // 时间
15.     CString m_strKey;           // 关键字
16.     CString m_strContext;       // 内容
17.     CString m_strMark;          // 备注
18.     int     m_nParentID;        // 父亲ID
19. };
22. class CDataBase
23. {
24. public:
25.     CDataBase(CString strName);
26.     virtual ~CDataBase(void);
29. public:
30.     void    Init();             // 初始化数据库
31.     void    Destroy();          // 销毁数据库连接
33. private:
34.     _ConnectionPtr  m_pConnection;  // 连接数据库指针
35.     _CommandPtr     m_pCommand;     // 命令集指针
36.     _RecordsetPtr   m_pRecordset;   // 记录集指针
38.     CString         m_strDatabaseName;// 数据库名称
40. 保存所有的数据
41.     BOOL    Load();             // 加载所
42. public:
43.     
44.     BOOL    Add(Item &item);    // 添加一条记录
45.     BOOL    Delete(int nID);    // 删除一条记录
46.     //BOOL  Modify()
47. };
48. vector<Item>    m_vecItem;  // 数据容器
50.     // 接口
51. public:
52.     BOOL    Save();             // 有的数据
54. 1. #include "StdAfx.h"
    2. #include "DataBase.h"
    4. CDataBase::CDataBase(CString strName)
    5.     : m_strDatabaseName(strName)
    6.     , m_pConnection(NULL)
    7.     , m_pCommand(NULL)
    8.     , m_pRecordset(NULL)
    9. {
    10.     Init();
    11. }
    13. CDataBase::~CDataBase(void)
    14. {
    15.     Destroy();
    16. }
    18. // 初始化数据库
    19. void CDataBase::Init()
    20. {
    21.     m_pConnection.CreateInstance(__uuidof(Connection));
    23.     try                 
    24.     {   
    25.         // 打开本地Access库NoteBook.mdb
    26.         CString strSQL=_T("Provider=Microsoft.Jet.OLEDB.4.0;Data Source=")+m_strDatabaseName+_T(".mdb");
    27.         m_pConnection->Open(_bstr_t(strSQL),_T(""),_T(""),adModeUnknown);
    29.         m_pCommand.CreateInstance(__uuidof(Recordset));
    30.         m_pRecordset.CreateInstance(__uuidof(Recordset));
    32.         try
    33.         {
    34.             // 获取所有字段数据
    35.             m_pRecordset->Open(_T("SELECT * FROM NoteBook"),m_pConnection.GetInterfacePtr(),
    36.                 adOpenDynamic,adLockOptimistic,adCmdText);
    37.         }
    38.         catch(_com_error &e)
    39.         {
    40.             AfxMessageBox(e.ErrorMessage());
    42.             return;
    43.         }
    44.     }
    45.     catch(_com_error &e)
    46.     {
    47.         
    48.         AfxMessageBox(e.ErrorMessage());
    49.         return;
    50.     }      
    52.     
    53. }
    55. // 销毁释放资源
    56. void CDataBase::Destroy()
    57. {
    58.     if( m_pRecordset!=NULL )
    59.     {
    60.         m_pRecordset->Close();
    61.         m_pRecordset.Release();
    62.         m_pRecordset=NULL;
    63.     }
    65.     if( m_pCommand!=NULL )
    66.     {
    67.         m_pCommand.Release();
    68.         m_pCommand=NULL;
    69.     }
    71.     if( m_pConnection->State )
    72.     {
    73.         m_pConnection->Close();
    74.         m_pConnection.Release();
    75.         m_pConnection= NULL;  
    76.     }
    78.     
    79. }
    81. // 加载数据库数据
    82. BOOL CDataBase::Load()
    83. {
    84.     if( m_pRecordset==NULL )
    85.         return FALSE;
    87.     _variant_t varID,varTitle,varTime,varKey,varContext,varMark,varParentID;
    88.     Item item;
    90.     try
    91.     {
    92.         if( !m_pRecordset->BOF )
    93.             m_pRecordset->MoveFirst();
    94.         else
    95.         {
    96.             AfxMessageBox(_T("表内数据为空"));
    97.             return FALSE;
    98.         }
    100.         // 读出库中各字段数据
    101.         while( !m_pRecordset->adoEOF )
    102.         {
    103.             varID = m_pRecordset->GetCollect(_T("ID"));
    104.             if(varID.vt != VT_NULL)
    105.                 item.m_nID = (long)(varID);
    107.             varTitle = m_pRecordset->GetCollect(_T("标题"));
    108.             if(varTitle.vt != VT_NULL)
    109.                 item.m_strTitle = (LPCSTR)_bstr_t(varTitle);
    111.             varTime = m_pRecordset->GetCollect(_T("日期"));
    112.             if(varTime.vt != VT_NULL)
    113.                 item.m_strTime = (LPCSTR)_bstr_t(varTime);
    115.             varKey = m_pRecordset->GetCollect(_T("关键字"));
    116.             if(varKey.vt != VT_NULL)
    117.                 item.m_strKey = (LPCSTR)_bstr_t(varKey);
    119.             varContext = m_pRecordset->GetCollect(_T("内容"));
    120.             if(varContext.vt != VT_NULL)
    121.                 item.m_strKey = (LPCSTR)_bstr_t(varContext);
    123.             varMark = m_pRecordset->GetCollect(_T("备注"));
    124.             if(varMark.vt != VT_NULL)
    125.                 item.m_strMark = (LPCSTR)_bstr_t(varMark);
    127.             varParentID = m_pRecordset->GetCollect(_T("父亲"));
    128.             if(varParentID.vt != VT_NULL)
    129.                 item.m_nParentID = (long)(varParentID);
    131.             // 送到容器
    132.             m_vecItem.push_back(item);
    134.             // 移动到下一条记录
    135.             m_pRecordset->MoveNext();
    136.         }
    137.     }
    138.     catch(_com_error &e)
    139.     {
    140.         AfxMessageBox(e.ErrorMessage());
    141.         return FALSE;
    142.     }
    144.     return TRUE;
    145. }
    148. // 存储到数据库
    149. BOOL CDataBase::Save()
    150. {
    151.     if( m_pRecordset==NULL )
    152.         return FALSE;
    154.     try
    155.     {
    156.         m_pCommand->ActiveConnection = m_pConnection;
    157.         m_pCommand->CommandText=_T("TRUNCATE TABLE NoteBook");
    158.         m_pCommand->Execute(NULL,NULL,adCmdText);       // 执行SQL语句
    161.         // 写入各字段值
    162.         m_pRecordset->AddNew();
    163.         for(vector<Item>::iterator iter=m_vecItem.begin(); iter!=m_vecItem.end(); ++iter)
    164.         {
    165.             m_pRecordset->PutCollect(_T("标题"), _variant_t(iter->m_strTitle));
    166.             m_pRecordset->PutCollect(_T("日期"), _variant_t(iter->m_strTime));
    167.             m_pRecordset->PutCollect(_T("关键字"), _variant_t(iter->m_strKey));
    168.             m_pRecordset->PutCollect(_T("内容"), _variant_t(iter->m_strContext));
    169.             m_pRecordset->PutCollect(_T("备注"), _variant_t(iter->m_strMark));
    170.             m_pRecordset->PutCollect(_T("父亲"), _variant_t(iter->m_nParentID));
    171.         }
    172.         m_pRecordset->Update();
    173.     }
    174.     catch(_com_error *e)
    175.     {
    176.         AfxMessageBox(e->ErrorMessage());
    177.         return FALSE;
    178.     }    
    180.     return TRUE;
    181. }
    184. // 增加记录
    185. BOOL CDataBase::Add(Item &item)
    186. {
    187.     if( m_pRecordset==NULL )
    188.         return FALSE;
    190.     // 向容器添加记录
    191.     m_vecItem.push_back(item);
    193.     try
    194.     {
    195.         // 向数据库增加记录
    196.         m_pRecordset->AddNew();
    197.         
    198.         m_pRecordset->PutCollect(_T("标题"), _variant_t(item.m_strTitle));
    199.         m_pRecordset->PutCollect(_T("日期"), _variant_t(item.m_strTime));
    200.         m_pRecordset->PutCollect(_T("关键字"), _variant_t(item.m_strKey));
    201.         m_pRecordset->PutCollect(_T("内容"), _variant_t(item.m_strContext));
    202.         m_pRecordset->PutCollect(_T("备注"), _variant_t(item.m_strMark));
    203.         m_pRecordset->PutCollect(_T("父亲"), _variant_t(item.m_nParentID));
    204.         
    205.         m_pRecordset->Update();
    206.     }
    207.     catch(_com_error *e)
    208.     {
    209.         AfxMessageBox(e->ErrorMessage());
    210.         return FALSE;
    211.     }    
    213.     return TRUE;
    214. }
    216. // 删除指定记录
    217. BOOL CDataBase::Delete(int nID)
    218. {
    219.     try
    220.     {
    221.         CString strSQL;
    222.         strSQL.Format(_T("DELETE FROM NoteBook WHERE ID=%d"),nID);
    224.         m_pCommand->ActiveConnection = m_pConnection;
    225.         m_pCommand->CommandText=(_bstr_t)strSQL;
    226.         m_pCommand->Execute(NULL,NULL,adCmdText);       // 执行SQL语句
    227.     }
    228.     catch(_com_error *e)
    229.     {
    230.         AfxMessageBox(e->ErrorMessage());
    231.         return FALSE;
    232.     }  
    234.     return TRUE;
    235. }

    这样，我们的操作数据库就简单了。就可以通过树型控件来进行数据库的访问操作了！

 

    后记：在测试这个类的时候，发现退出程序会出现recordset异常，察看了调用堆栈，居然说recordset的Close方法错误？

My GOD！ 怎么可能呢？recordset智能指针时需要先Close再Release的嘛！，后来检查Destroy函数，才发现原来是connection指针已经释放了，导致recordset指针出错！把释放顺序改成先recordset再connection就没问题了~具体什么原因，也不想追踪了，但是一定记得--释放recordset指针的时候一定要赶在connection指针释放前！！