// MsgAttachment.cpp : implementation file
|
//
|
|
#include "stdafx.h"
|
#include "mbamdb.h"
|
#include "MsgAttachment.h"
|
#include "MBAMDBFun.h"
|
#include "Msg.h"
|
// CMsgAttachment
|
CMsgAttachmentItem::CMsgAttachmentItem()
|
{
|
m_nRowType = ROWTYPE_FOLDERMGR;
|
Clear();
|
}
|
|
CMsgAttachmentItem::~CMsgAttachmentItem()
|
{
|
}
|
void CMsgAttachmentItem::operator =( const CMsgAttachmentItem& item )
|
{
|
m_strFileName = item.m_strFileName;
|
m_strFilePath = item.m_strFilePath;
|
m_nFileSize = item.m_nFileSize;
|
m_strMsgID = item.m_strMsgID;
|
m_strFileID = item.m_strFileID;
|
m_strFileSFlag = item.m_strFileSFlag;
|
m_strFileMD5 = item.m_strFileMD5;
|
m_nFileEnType = item.m_nFileEnType;
|
m_tmSendDate = item.m_tmSendDate;
|
}
|
void CMsgAttachmentItem::Clear()
|
{
|
m_strFileName = _T("");
|
m_strFilePath = _T("");
|
m_nFileSize = 0;
|
m_strFileID = _T("");
|
m_strFileSFlag = _T("");
|
m_nFileEnType = 0;
|
m_strFileMD5 = _T("");
|
m_tmSendDate = COleDateTime::GetCurrentTime( );
|
}
|
|
BOOL CMsgAttachmentItem::GetItemValue( CString strName, COleVariant &varData )
|
{
|
varData.Clear();
|
if ( strName.IsEmpty() )
|
return false;
|
if ( strName.CompareNoCase( _T( "MsgID" ) ) == 0 )
|
varData = m_strMsgID;
|
else if ( strName.CompareNoCase( _T( "FileName" ) ) == 0 )
|
varData = m_strFileName;
|
else if ( strName.CompareNoCase( _T( "FileSize" ) ) == 0 )
|
varData = (_int64)m_nFileSize;
|
else if ( strName.CompareNoCase( _T( "FilePath" ) ) == 0 )
|
varData = m_strFilePath;
|
else if ( strName.CompareNoCase( _T( "FileID" ) ) == 0 )
|
varData = m_strFileID;
|
else if ( strName.CompareNoCase( _T( "FileSFlag" ) ) == 0 )
|
varData = m_strFileSFlag;
|
else if ( strName.CompareNoCase( _T( "FileEnType" ) ) == 0 )
|
varData = (long)m_nFileEnType;
|
else if ( strName.CompareNoCase( _T( "FileMD5" ) ) == 0 )
|
varData = m_strFileMD5;
|
|
else
|
return false;
|
|
return true;
|
}
|
BOOL CMsgAttachmentItem::InsertRecord( CDBConnect *pAConnect)
|
{
|
CString strSQL;
|
BOOL bReturn = true;
|
CString strFileName;
|
|
strFileName = CMBAMDBFun::GetDBReplaceStr( m_strFileName );
|
if( strFileName.Right(1) == _T("\\") )
|
{
|
strFileName.TrimRight(_T("\\"));
|
if(strFileName.IsEmpty())
|
strFileName = _T("AMTempFolder\\");
|
else
|
strFileName += _T("\\");
|
}
|
else
|
{
|
if(strFileName.IsEmpty())
|
strFileName = _T("AMTempFile");
|
}
|
strSQL.Format( _T("INSERT INTO TN_STK_ATTACHMENT(G_C1, S_C2, N_C3, S_C4, G_C5,S_C6, N_C7,S_C8) VALUES ( \
|
'%s','%s', %I64d, '%s', '%s', '%s', %d, '%s')"),m_strMsgID, strFileName, m_nFileSize,
|
m_strFileMD5, m_strFileID, m_strFileSFlag, m_nFileEnType, CMBAMDBFun::GetDBReplaceStr( m_strFilePath ));
|
try
|
{
|
bReturn = CDBSQLiteCommFun::RunSQL(strSQL, pAConnect);
|
}
|
catch ( CDBSQliteException ex )
|
{
|
CString strError = ex.GetErrInfo();
|
CString strErrInfo;
|
|
strErrInfo = _T("InsertRecord - MsgAttachment- ") + strError;
|
CMBAMDBFun::WriteDBErrToFile( strErrInfo );
|
return false;
|
}
|
return true;
|
}
|
BOOL CMsgAttachmentItem::UpdateRecord( CDBConnect *pAConnect )
|
{
|
CString strSQL;
|
BOOL bReturn = true;
|
CString strFileName;
|
|
strFileName = CMBAMDBFun::GetDBReplaceStr( m_strFileName );
|
if( strFileName.Right(1) == _T("\\") )
|
{
|
strFileName.TrimRight(_T("\\"));
|
if(strFileName.IsEmpty())
|
strFileName = _T("AMTempFolder\\");
|
else
|
strFileName += _T("\\");
|
}
|
else
|
{
|
if(strFileName.IsEmpty())
|
strFileName = _T("AMTempFile");
|
}
|
strSQL.Format( _T("UPDATE TN_STK_ATTACHMENT SET N_C3 = %I64d WHERE S_C2 = '%s' AND G_C1 = '%s' AND G_C5 = '%s' AND S_C6 = '%s'"),
|
m_nFileSize, strFileName,
|
m_strMsgID, m_strFileID, m_strFileSFlag );
|
|
try
|
{
|
bReturn = CDBSQLiteCommFun::RunSQL(strSQL, pAConnect);
|
}
|
catch ( CDBSQliteException ex )
|
{
|
CString strError = ex.GetErrInfo();
|
CString strErrInfo;
|
|
strErrInfo.Format( _T("UpdateRecord(%s) - MsgAttachment - %s "),strSQL, strError);
|
CMBAMDBFun::WriteDBErrToFile( strErrInfo );
|
return false;
|
}
|
return true;
|
}
|
CMsgAttachments::CMsgAttachments(CMBAMDBConnect *pAConnect)
|
{
|
m_pMsgs = NULL;
|
m_pAConnect = pAConnect;
|
m_bHaveDetail = false;
|
}
|
|
CMsgAttachments::~CMsgAttachments()
|
{
|
}
|
|
// CMsgAttachment member functions
|
void CMsgAttachments::operator =( const CMsgAttachments& item )
|
{
|
m_pMsgs = item.m_pMsgs;
|
|
// m_ARecordSet = item.m_ARecordSet;
|
m_pAConnect = item.m_pAConnect;
|
m_strSQL = item.m_strSQL;
|
m_nRowType = item.m_nRowType;
|
|
m_strFileName = item.m_strFileName;
|
m_strFilePath = item.m_strFilePath;
|
m_nFileSize = item.m_nFileSize;
|
m_strMsgID = item.m_strMsgID;
|
m_strFileSFlag = item.m_strFileSFlag;
|
m_strFileID = item.m_strFileID;
|
m_nFileEnType = item.m_nFileEnType;
|
m_bHaveDetail = item.m_bHaveDetail;
|
m_tmSendDate = item.m_tmSendDate;
|
m_strFileMD5 = item.m_strFileMD5;
|
}
|
|
bool CMsgAttachments::GetData( )
|
{
|
CString strTempValue;
|
try
|
{
|
strTempValue = _T("");
|
m_ARecordSet.GetValue( 1, strTempValue, true );
|
m_strMsgID = strTempValue;
|
|
strTempValue = _T("");
|
m_ARecordSet.GetValue( 2, strTempValue, true );
|
m_strFileID = strTempValue;
|
if ( m_bHaveDetail )
|
{
|
strTempValue = _T("");
|
m_ARecordSet.GetValue( 3, strTempValue, true );
|
m_strFileName = strTempValue;
|
|
m_nFileSize = 0;
|
m_ARecordSet.GetValue( 4, m_nFileSize );
|
|
strTempValue = _T("");
|
m_ARecordSet.GetValue( 5, strTempValue, true );
|
m_strFileMD5 = strTempValue;
|
|
strTempValue = _T("");
|
m_ARecordSet.GetValue( 6, strTempValue, true );
|
m_strFileSFlag = strTempValue;
|
|
m_ARecordSet.GetValue( 7, strTempValue );
|
m_nFileEnType = _tstoi( strTempValue );
|
|
strTempValue = _T("");
|
m_ARecordSet.GetValue( 8, strTempValue, true );
|
m_strFilePath = strTempValue;
|
strTempValue = _T("");
|
|
m_ARecordSet.GetValue( 9, m_tmSendDate);
|
/*m_ARecordSet.GetValue( 9, strTempValue, true );
|
m_strSendDate = strTempValue;*/
|
}
|
}
|
catch ( CDBSQliteException ex )
|
{
|
CString strError;
|
|
strError = (LPCTSTR)ex.GetErrInfo( );
|
return false;
|
}
|
|
return true;
|
}
|
|
|
int CMsgAttachments::ListSpMsgID(LPCTSTR lpszMsgID, LPCTSTR lpsFileID)
|
{
|
if ( lpszMsgID == NULL )
|
return 0;
|
|
CString strSQL;
|
int nCount;
|
strSQL.Format( _T("SELECT a.G_C1, b.G_C5,b.S_C2, b.N_C3,b.S_C4, b.S_C6, b.N_C7, b.S_C8, a.T_C6 ")
|
_T("FROM TN_STK_MSG a INNER JOIN TN_STK_ATTACHMENT b ")
|
_T("ON a.G_C1 = b.G_C1 ")
|
_T("WHERE b.G_C1 = '%s' AND b.G_C5 = '%s' "), lpszMsgID, lpsFileID);
|
m_strSQL = strSQL;
|
if( !m_pAConnect || !m_pAConnect->IsOpenDB())
|
return 0;
|
|
try
|
{
|
m_ARecordSet.Close();
|
m_ARecordSet.Open( m_pAConnect->GetDBConnect(), strSQL );
|
m_ARecordSet.MoveFirst();
|
nCount = m_ARecordSet.GetRecordCount( );
|
}
|
catch( CDBSQliteException ex )
|
{
|
CString strError;
|
|
strError = (LPCTSTR)ex.GetErrInfo( );
|
|
return 0;
|
}
|
m_bHaveDetail = TRUE;
|
return nCount;
|
}
|
void CMsgAttachments::LoadData( CMsgAttachmentItem msgAttachment )
|
{
|
m_strFileName = msgAttachment.m_strFileName;
|
m_strFilePath = msgAttachment.m_strFilePath;
|
m_strFileID = msgAttachment.m_strFileID;
|
m_strFileSFlag = msgAttachment.m_strFileSFlag;
|
m_nFileSize = msgAttachment.m_nFileSize;
|
m_strMsgID = msgAttachment.m_strMsgID;
|
m_strFileMD5 = msgAttachment.m_strFileMD5;
|
m_nFileEnType = msgAttachment.m_nFileEnType;
|
}
|
|
bool CMsgAttachments::GetMsg( CMsgs &msg )
|
{
|
if ( m_pMsgs == NULL )
|
return false;
|
msg = (*m_pMsgs);
|
return true;
|
}
|
long CMsgAttachments::ListRecordCount( )
|
{
|
CString strTempSQL;
|
if(m_strSQL.IsEmpty())
|
return m_ARecordSet.GetRecordCount( );
|
else
|
{
|
strTempSQL.Format(_T(" SELECT COUNT(*) FROM ( %s)T"), m_strSQL);
|
long nTempValue = 0;
|
if( !m_pAConnect || !m_pAConnect->IsOpenDB())
|
return nTempValue;
|
CDBRecord ARecordset;
|
try
|
{
|
ARecordset.Open( m_pAConnect->GetDBConnect(), strTempSQL );
|
while ( S_OK == ARecordset.MoveNext() )
|
{
|
ARecordset.GetValue( 1, nTempValue );
|
break;
|
}
|
}
|
catch ( CDBSQliteException ex )
|
{
|
CString strError;
|
|
strError = (LPCTSTR)ex.GetErrInfo( );
|
return nTempValue;
|
}
|
return nTempValue;
|
}
|
}
|
|
int CMsgAttachments::ListAttachments( CString strUser,CString strUserName, CString strKey )
|
{
|
CString strSQL;
|
CString strStartDate;
|
CString strEndDate;
|
CString strSQLWhere= _T("");
|
int nCount;
|
CString strSQLTemp = _T("");
|
CDBRecord ARecordSet;
|
|
strUser.MakeLower();
|
if( !strKey.IsEmpty())
|
{
|
strSQLTemp.Format( _T("AND b.S_C2 LIKE '%%%s%%' "), strKey );
|
strSQLWhere += strSQLTemp;
|
}
|
if ( m_tmStartDate.GetYear() > 1970 )
|
{
|
strStartDate = CMBAMDBFun::FormatDBTIMEToStr( m_tmStartDate );
|
if ( m_tmEndDate > m_tmStartDate)
|
{
|
strEndDate = CMBAMDBFun::FormatDBTIMEToStr( m_tmEndDate );
|
strSQLTemp.Format( _T(" AND a.T_C6 >= '%s' AND a.T_C6 <= '%s'"), strStartDate, strEndDate );
|
}
|
else
|
{
|
strSQLTemp.Format( _T(" AND a.T_C6 >= '%s'"), strStartDate );
|
}
|
strSQLWhere += strSQLTemp;
|
}
|
else
|
{
|
if ( m_tmEndDate.GetYear() > 1970 )
|
{
|
strEndDate = CMBAMDBFun::FormatDBTIMEToStr( m_tmEndDate );
|
strSQLTemp.Format( _T(" AND a.T_C6 <= '%s'"), strEndDate );
|
strSQLWhere += strSQLTemp;
|
}
|
}
|
|
if( !strUser.IsEmpty())
|
{
|
if(strUserName.IsEmpty() )
|
{
|
strSQLTemp.Format( _T("AND (a.S_C4 LIKE '%s%%' OR a.S_C5 LIKE '%s%%') "), strUser,strUser );
|
}
|
else
|
strSQLTemp.Format( _T("AND a.S_C4='%s' "), strUser );
|
strSQLWhere += strSQLTemp;
|
}
|
//strSQL.Format( _T("SELECT a.G_C1, a.S_C4, a.S_C5, b.S_C2, b.N_C3, b.S_C4, b.G_C5, b.S_C6, b.N_C7, b.S_C8, a.T_C6, a.C_C9, a.N_C11 ")
|
// _T("FROM TN_STK_MSG a INNER JOIN TN_STK_ATTACHMENT b ")
|
// _T("ON a.G_C1 = b.G_C1 ")
|
// _T("WHERE a.N_C12 > 0 %s ORDER BY a.T_C6 DESC, a.G_C1, b.S_C2"),strSQLWhere);
|
strSQL.Format( _T("SELECT a.G_C1, b.G_C5 ")
|
_T("FROM TN_STK_MSG a INNER JOIN TN_STK_ATTACHMENT b ")
|
_T("ON a.G_C1 = b.G_C1 ")
|
_T("WHERE a.N_C12 > 0 %s ORDER BY a.T_C6 DESC, a.G_C1, b.S_C2"),strSQLWhere);
|
m_strSQL = strSQL;
|
if( !m_pAConnect || !m_pAConnect->IsOpenDB())
|
return 0;
|
try
|
{
|
nCount = ListRecordCount();
|
m_ARecordSet.Close( );
|
m_ARecordSet.Open( m_pAConnect->GetDBConnect(), strSQL );
|
}
|
catch( CDBSQliteException ex )
|
{
|
CString strError;
|
|
strError = (LPCTSTR)ex.GetErrInfo( );
|
|
return 0;
|
}
|
return nCount;
|
}
|
