classset.cpp

来自「深入浅出Visual C++入门进阶与应用实例 随书光盘 作者 何志丹」· C++ 代码 · 共 158 行

// ClassSet.cpp : implementation file
//

#include "stdafx.h"
#include "ClassSet.h"

#include "db.h"

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

/////////////////////////////////////////////////////////////////////////////
// CClassSet

IMPLEMENT_DYNAMIC(CClassSet, CRecordset)

CClassSet::CClassSet(CDatabase* pdb)
: CRecordset(/*pdb*/NULL == pdb ? CDB::GetDefaultDB() : pdb)
{//pdb为空时,取默认数据库
	//{{AFX_FIELD_INIT(CClassSet)
	m_nID = 0;
	m_strText = _T("");
	m_nParentID = 0;
	m_strDir = _T("");
	m_nFields = 4;
	//}}AFX_FIELD_INIT
	m_nDefaultType = snapshot;
}


CString CClassSet::GetDefaultConnect()
{
	ASSERT(false);//不会被调用,因为重载了构造函数	
	//return _T("ODBC;DSN=community");
	return _T("ODBC;DSN=" + CDB::m_strDsn);
}

CString CClassSet::GetDefaultSQL()
{
	return _T("[ClassTree]");
}

void CClassSet::DoFieldExchange(CFieldExchange* pFX)
{
	//{{AFX_FIELD_MAP(CClassSet)
	pFX->SetFieldType(CFieldExchange::outputColumn);
	RFX_Long(pFX, _T("[ID]"), m_nID);
	RFX_Text(pFX, _T("[sText]"), m_strText);
	RFX_Long(pFX, _T("[ParentID]"), m_nParentID);
	RFX_Text(pFX, _T("[对应文件夹名称]"), m_strDir);
	//}}AFX_FIELD_MAP
}

/////////////////////////////////////////////////////////////////////////////
// CClassSet diagnostics

#ifdef _DEBUG
void CClassSet::AssertValid() const
{
	CRecordset::AssertValid();
}

void CClassSet::Dump(CDumpContext& dc) const
{
	CRecordset::Dump(dc);
}
#endif //_DEBUG

//==============================================================================//
//                        静态成员函数
//==============================================================================//

//增加一个分类树节点,由于sql语句无法返回新生成的ID,故用此类的一般成员函数
bool CClassSet::AddTreeNode(const CString strText, int nParentID,CString strDir,int& nID)
{
	CClassSet rs(CDB::GetDefaultDB());
	try
	{
		rs.Open(CRecordset::dynaset);
		rs.AddNew();
		rs.m_strText	= strText ;
		rs.m_nParentID  = nParentID ;
		rs.m_strDir     = strDir ;
		rs.Update();
		rs.MoveLast();//用update没有刷新,无法得到id的值
		nID = rs.m_nID ;
		rs.Close();
	}
	catch(CDBException * e)
	{
		ASSERT(false);
		e->m_strError;
		return false;
	}
	
	return true;
}

//修改分类树节点
bool CClassSet::EditTreeNode(int nID,const CString strText, int nParentID,CString strDir)
{
	//取得默认库
	CDatabase * pDatabase = CDB::GetDefaultDB() ;
	if(NULL == pDatabase || !pDatabase->IsOpen())
	{
		ASSERT(false);
		return false;
	}
	
	//执行sql语句
	CString strSQL;
	strSQL.Format("update ClassTree set sText=\'%s\',ParentID=%d,对应文件夹名称=\'%s\' WHERE ID= %d",strText,nParentID,strDir,nID);
	try
	{
		pDatabase->ExecuteSQL(strSQL);
	}
	catch(CDBException* e)
	{
		ASSERT(false);
		e->m_strError;
		return false;
	}
	
	return true;
}

bool CClassSet::DeleteTreeNode(int nID)
{
	//取得默认库
	CDatabase * pDatabase = CDB::GetDefaultDB() ;
	if(NULL == pDatabase || !pDatabase->IsOpen())
	{
		ASSERT(false);
		return false;
	}
	
	//执行sql语句
	CString strSQL;
	strSQL.Format("delete * from  ClassTree where ID = %d",nID);
	try
	{
		pDatabase->ExecuteSQL(strSQL);
	}
	catch(CDBException* e)
	{
		ASSERT(false);
		e->m_strError;
		return false;
	}
	
	return true;
}