开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 行业应用 > 金融证券系统 > 查看源码
TWSocket.cpp
资源名称：NetTS.rar　[点击查看]
上传用户：nameszq
上传日期：2014-08-12
资源大小：336k
文件大小：14k
源码类别：

金融证券系统

开发平台：

Visual C++

TWSocket.cpp：源码内容
  1. // TWSocket.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "TWSocket.h"
  6. #include "TSCache.h"
  7. #include "StockDrv.h"
  8. #include "SpTime.h"
  10. #ifdef _DEBUG
  11. #define new DEBUG_NEW
  12. #undef THIS_FILE
  13. static char THIS_FILE[] = __FILE__;
  14. #endif
  16. /////////////////////////////////////////////////////////////////////////////
  17. // TWAutoReportThreadMain
  19. // 自动Report
  20. UINT TWAutoReportThreadMain( LPVOID pParam )
  21. {
  22. while( TRUE )
  23. {
  24. if( CTWSocket::GetInstance().GetAutoReport() && CSPTime::InTradeTime(CSPTime::GetCurrentTime().GetTime(),900) )
  25. {
  26. int nSize = CTSCache::GetInstance().GetTotalNumber();
  27. if( nSize <= 0 )
  28. continue;
  30. TW_STOCK stocks[160];
  31. memset( stocks, 0, sizeof(stocks) );
  32. int put = 0;
  34. RCV_REPORT_STRUCTEx report;
  35. for( int i=0; i<nSize; i++ )
  36. {
  37. if( CTSCache::GetInstance().GetStockByNoEx( i, &report ) )
  38. {
  39. strncpy( stocks[put].m_code, report.m_szLabel, sizeof(stocks[put].m_code) );
  40. stocks[put].m_type = typeshA;
  41. if( SH_MARKET_EX == report.m_wMarket )
  42. stocks[put].m_type = typeshA;
  43. else if( SZ_MARKET_EX == report.m_wMarket )
  44. stocks[put].m_type = typeszA;
  45. put ++;
  46. if( put >= 160 )
  47. {
  48. put = 0;
  49. CTWSocket::GetInstance().RequestReport( stocks, put );
  50. Sleep( 10000 );
  51. }
  52. }
  53. }
  55. if( put > 0 )
  56. CTWSocket::GetInstance().RequestReport( stocks, put );
  57. }
  59. if( !CTWSocket::GetInstance().IsReceiving() )
  60. {
  61. CTWCommand cmd;
  62. if( CTWSocket::GetInstance().PopCommand( cmd ) )
  63. {
  64. CTWSocket::GetInstance().RequestStockData( cmd.m_nDataType, 
  65. cmd.m_stocks.GetData(), cmd.m_stocks.GetSize(),
  66. cmd.m_nKType, cmd.m_nDataCount );
  67. }
  68. }
  70. Sleep( 30000 );
  71. }
  72. AfxEndThread( 0, TRUE );
  73. }
  75. /////////////////////////////////////////////////////////////////////////////
  76. // CTWSocket
  78. CTWSocket & CTWSocket::GetInstance()
  79. {
  80. static CTWSocket s_twsocket;
  81. return s_twsocket;
  82. }
  84. void CTWSocket::Init( )
  85. {
  86. AfxSocketInit( );
  87. }
  89. void CTWSocket::Release( )
  90. {
  91. if( GetInstance().IsWorking() )
  92. GetInstance().EndWorking();
  93. AfxSocketTerm( );
  94. }
  96. CTWSocket::CTWSocket()
  97. {
  98. m_mapCommands.InitHashTable( 2000, FALSE );
  99. m_bIsReceiving = FALSE;
  101. SetAutoReport( );
  102. AfxBeginThread( TWAutoReportThreadMain, NULL, THREAD_PRIORITY_NORMAL );
  103. }
  105. CTWSocket::~CTWSocket()
  106. {
  107. POSITION pos = m_mapCommands.GetStartPosition();
  108. while( NULL != pos )
  109. {
  110. CString rKey;
  111. void * rValue;
  112. m_mapCommands.GetNextAssoc( pos, rKey, rValue );
  113. if( NULL != rValue )
  114. {
  115. CSimpTWCommandArray * ptr = (CSimpTWCommandArray *) rValue;
  116. delete ptr;
  117. }
  118. }
  119. m_mapCommands.RemoveAll();
  120. }
  122. BOOL CTWSocket::IsWorking( )
  123. {
  124. CSingleLock lock( &m_mutex, TRUE );
  126. if( INVALID_SOCKET == m_hSocket )
  127. return FALSE;
  128. CString rPeerAddress;
  129. UINT rPeerPort;
  130. if( GetPeerName( rPeerAddress, rPeerPort)
  131. && !rPeerAddress.IsEmpty() )
  132. return TRUE;
  133. return FALSE;
  134. }
  136. BOOL CTWSocket::BeginWorking(LPCTSTR lpszHostAddress, UINT nHostPort, LPCTSTR lpszUser, LPCTSTR lpszPasswd )
  137. {
  138. CSingleLock lock( &m_mutex, TRUE );
  139. m_strHostAddress = lpszHostAddress;
  140. m_nHostPort = nHostPort;
  141. m_strUser = lpszUser;
  142. m_strPasswd = lpszPasswd;
  144. if( m_strHostAddress.IsEmpty() )
  145. return FALSE;
  147. if( !Create( ) )
  148. return FALSE;
  150. int nProxyType;
  151. CString strProxyAddress;
  152. UINT nProxyPort;
  153. AfxGetQSProfile().GetCompoundConfig( nProxyType, strProxyAddress, nProxyPort );
  154. SetProxyInfo( nProxyType, strProxyAddress, nProxyPort,
  155. AfxGetQSProfile().GetProxyUser(), AfxGetQSProfile().GetProxyPasswd() );
  157. if( !ConnectThroughProxy( m_strHostAddress, m_nHostPort ) )
  158. {
  159. m_strLastError = AfxModuleLoadString( IDS_TWSOCKET_ERRLOGIN );
  161. // 没有连接成功，以后调用无参数的BeginWorking()不再连接
  162. m_strHostAddress.Empty();
  163. m_nHostPort = 0;
  164. m_strUser.Empty();
  165. m_strPasswd.Empty();
  166. Close();
  167. return FALSE;
  168. }
  170. if( !Login(lpszUser,lpszPasswd) )
  171. {
  172. m_strLastError = AfxModuleLoadString( IDS_TWSOCKET_ERRLOGIN );
  174. // 没有连接成功，以后调用无参数的BeginWorking()不再连接
  175. m_strHostAddress.Empty();
  176. m_nHostPort = 0;
  177. m_strUser.Empty();
  178. m_strPasswd.Empty();
  179. Close();
  180. return FALSE;
  181. }
  183. return TRUE;
  184. }
  186. BOOL CTWSocket::BeginWorking( )
  187. {
  188. CSingleLock lock( &m_mutex, TRUE );
  190. if( m_strHostAddress.IsEmpty() )
  191. return FALSE;
  193. if( !Create( ) )
  194. return FALSE;
  196. int nProxyType;
  197. CString strProxyAddress;
  198. UINT nProxyPort;
  199. AfxGetQSProfile().GetCompoundConfig( nProxyType, strProxyAddress, nProxyPort );
  200. SetProxyInfo( nProxyType, strProxyAddress, nProxyPort,
  201. AfxGetQSProfile().GetProxyUser(), AfxGetQSProfile().GetProxyPasswd() );
  203. if( !ConnectThroughProxy( m_strHostAddress, m_nHostPort ) )
  204. {
  205. Close();
  206. return FALSE;
  207. }
  209. if( !Login(m_strUser,m_strPasswd) )
  210. {
  211. m_strLastError = AfxModuleLoadString( IDS_TWSOCKET_ERRLOGIN );
  212. Close();
  213. return FALSE;
  214. }
  216. return TRUE;
  217. }
  219. void CTWSocket::EndWorking( )
  220. {
  221. CSingleLock lock( &m_mutex, TRUE );
  223. ShutDown();
  224. Close();
  225. }
  227. BOOL CTWSocket::Login( LPCTSTR lpszUser, LPCTSTR lpszPasswd )
  228. {
  229. BYTE sbuffer[512];
  230. int lenSend = 0;
  231. if( lenSend = ConstructLoginBuffer(sbuffer,sizeof(sbuffer),lpszUser,lpszPasswd) )
  232. {
  233. if( Send( sbuffer, lenSend ) == lenSend )
  234. {
  235. int lenRcv = Receive( m_rbuffer, sizeof(m_rbuffer) );
  236. if( IsLoginOK( m_rbuffer, lenRcv ) )
  237. {
  238. CTSCache::GetInstance().LoadReports();
  239. RequestInit();
  240. return TRUE;
  241. }
  242. return FALSE;
  243. }
  244. }
  245. return FALSE;
  246. }
  248. int CTWSocket::SetAutoReport( int bAutoReport )
  249. {
  250. int old = m_bAutoReport;
  251. m_bAutoReport = bAutoReport;
  252. return old;
  253. }
  255. int CTWSocket::GetAutoReport( )
  256. {
  257. return m_bAutoReport;
  258. }
  260. BOOL CTWSocket::OnNewCommand( int nDataType, TW_STOCK * pStock, int nMinTimeGap )
  261. {
  262. if( nMinTimeGap <= 0 )
  263. return TRUE;
  265. CSingleLock lock( &m_mutex, TRUE );
  267. int stocktype = 0x00;
  268. if( pStock )
  269. stocktype = pStock->m_type;
  270. char stockcode[sizeof(TW_STOCK)+1];
  271. memset( stockcode, 0, sizeof(stockcode) );
  272. if( pStock )
  273. strncpy( stockcode, pStock->m_code, min(sizeof(stockcode),sizeof(pStock->m_code)) );
  274. if( strlen(stockcode) <= 0 )
  275. strncpy( stockcode, "NOCODE", sizeof(stockcode)-1 );
  277. time_t now = time(NULL);
  278. CSimpTWCommand cmd( nDataType, stocktype, stockcode, now );
  280. void * rValue = NULL;
  281. CSimpTWCommandArray * ptr = NULL;
  282. if( m_mapCommands.Lookup( stockcode, rValue ) && rValue )
  283. {
  284. ptr = (CSimpTWCommandArray *)rValue;
  285. for( int i=0; i<ptr->GetSize(); i++ )
  286. {
  287. if( cmd.m_nType == ptr->ElementAt(i).m_nType
  288. && cmd.m_nDataType == ptr->ElementAt(i).m_nDataType )
  289. {
  290. if( (cmd.m_time - ptr->ElementAt(i).m_time) < nMinTimeGap  )
  291. return FALSE;
  292. else
  293. {
  294. ptr->ElementAt(i).m_time = now;
  295. return TRUE;
  296. }
  297. }
  298. }
  299. ptr->Add( cmd );
  300. return TRUE;
  301. }
  303. ptr = new CSimpTWCommandArray();
  304. if( ptr )
  305. {
  306. ptr->SetSize( 0, TWCMD_DATATYPE_COUNT );
  307. ptr->Add( cmd );
  308. m_mapCommands.SetAt( stockcode, ptr );
  309. return TRUE;
  310. }
  312. return TRUE;
  313. }
  315. void CTWSocket::PushCommand( CTWCommand & cmd )
  316. {
  317. CSingleLock lock( &m_mutex, TRUE );
  318. m_queueCommands.Add( cmd );
  319. }
  321. BOOL CTWSocket::PopCommand( CTWCommand & cmd )
  322. {
  323. CSingleLock lock( &m_mutex, TRUE );
  325. if( m_queueCommands.GetSize() > 0 )
  326. {
  327. cmd = m_queueCommands.ElementAt(0);
  328. m_queueCommands.RemoveAt(0);
  329. return TRUE;
  330. }
  331. return FALSE;
  332. }
  334. int CTWSocket::RequestStockData( int nDataType, TW_STOCK * pStocks, int nSize, int nKType, int nDataCount )
  335. {
  336. if( m_bIsReceiving && time(NULL) - m_timeReceiveLast > 60 )
  337. m_bIsReceiving = FALSE;
  339. if( m_bIsReceiving )
  340. {
  341. CTWCommand cmd( nDataType, pStocks, nSize, nKType, nDataCount );
  342. PushCommand( cmd );
  343. return nSize;
  344. }
  346. int nRet = 0;
  347. switch( nDataType )
  348. {
  349. case RCV_REPORT:
  350. if( OnNewCommand( nDataType, pStocks, 0 ) )
  351. nRet = RequestReport( pStocks, nSize );
  352. break;
  353. case FILE_HISTORY_EX:
  354. if( OnNewCommand( nDataType, pStocks, 300 ) )
  355. nRet = RequestHistory( pStocks, nSize, nKType, nDataCount );
  356. break;
  357. case FILE_MINUTE_EX:
  358. if( OnNewCommand( nDataType, pStocks, 60 ) )
  359. nRet = RequestMinute( pStocks, nSize );
  360. break;
  361. case FILE_POWER_EX:
  362. if( OnNewCommand( nDataType, pStocks, 300 ) )
  363. nRet = RequestPower( pStocks, nSize );
  364. break;
  365. case FILE_MULTISORT_EX:
  366. if( OnNewCommand( nDataType, pStocks, 0 ) )
  367. nRet = RequestMultisort( pStocks, nSize );
  368. break;
  369. case FILE_DETAIL_EX:
  370. if( OnNewCommand( nDataType, pStocks, 60 ) )
  371. nRet = RequestDetail( pStocks, nSize );
  372. break;
  373. case FILE_BASE_EX:
  374. if( OnNewCommand( nDataType, pStocks, 300 ) )
  375. nRet = RequestBase( pStocks, nSize );
  376. break;
  377. case FILE_NEWS_EX:
  378. if( OnNewCommand( nDataType, pStocks, 60 ) )
  379. nRet = RequestNews( pStocks, nSize );
  380. break;
  381. case FILE_HTML_EX:
  382. if( OnNewCommand( nDataType, pStocks, 60 ) )
  383. nRet = RequestHtml( pStocks, nSize );
  384. break;
  385. case FILE_SOFTWARE_EX:
  386. if( OnNewCommand( nDataType, pStocks, 60 ) )
  387. nRet = RequestSoftware( pStocks, nSize );
  388. break;
  389. case FILE_SHAZQDATA_EX:
  390. if( OnNewCommand( nDataType, pStocks, 60 ) )
  391. nRet = RequestShazqdata( pStocks, nSize );
  392. break;
  393. default:;
  394. }
  395. return nRet;
  396. }
  398. int CTWSocket::RequestInit( )
  399. {
  400. TW_ASK ask;
  401. int lenSend = ConstructAskInitBuffer(ask);
  402. if( lenSend > 0 )
  403. {
  404. Send(&ask,lenSend);
  405. m_bIsReceiving = TRUE;
  406. m_timeReceiveLast = time(NULL);
  407. return 1;
  408. }
  409. return 0;
  410. }
  412. int CTWSocket::RequestReport( TW_STOCK * pStock, int nSize )
  413. {
  414. TW_STOCK stocks[32];
  415. memset( stocks, 0, sizeof(stocks) );
  416. int nMaxCount = sizeof(stocks)/sizeof(TW_STOCK);
  418. int putsize = 0;
  419. for( int i=0; i<nSize; i++ )
  420. {
  421. memcpy( &(stocks[putsize]), &(pStock[i]), sizeof(TW_STOCK) );
  422. putsize ++;
  423. if( putsize >= nMaxCount || i==nSize-1 )
  424. {
  425. TW_ASK ask;
  426. int lenSend = ConstructAskReportBuffer(ask,stocks,putsize);
  427. if( lenSend > 0 )
  428. Send(&ask,lenSend);
  429. putsize = 0;
  430. }
  431. }
  432. return nSize;
  433. }
  435. int CTWSocket::RequestHistory( TW_STOCK * pStock, int nSize, int nKType, int nDataCount )
  436. {
  437. for( int i=0; i<nSize; i++ )
  438. {
  439. BOOL bChangeStock = TRUE;
  440. BOOL bFirstRequest = TRUE;
  442. while( nDataCount > 0 )
  443. {
  444. int nCountNow = (nDataCount > 0x100 ? 0x100 : nDataCount);
  445. if( bChangeStock || bFirstRequest ) nCountNow = 1;
  446. TW_ASK ask;
  447. int lenSend = ConstructAskHistoryBuffer(ask,&(pStock[i]),nKType,nCountNow,bChangeStock,bFirstRequest);
  448. if( lenSend > 0 )
  449. Send(&ask,lenSend);
  450. nDataCount -= nCountNow;
  452. bChangeStock = FALSE;
  453. bFirstRequest = FALSE;
  454. }
  455. }
  456. return nSize;
  457. }
  459. int CTWSocket::RequestMinute( TW_STOCK * pStock, int nSize )
  460. {
  461. for( int i=0; i<nSize; i++ )
  462. {
  463. TW_ASK ask;
  464. int lenSend = ConstructAskMinuteBuffer(ask,&(pStock[i]));
  465. if( lenSend > 0 )
  466. Send(&ask,lenSend);
  467. }
  468. return nSize;
  469. }
  471. int CTWSocket::RequestPower( TW_STOCK * pStock, int nSize )
  472. {
  473. //NOTSUPPORT
  474. return 0;
  475. }
  477. int CTWSocket::RequestMultisort( TW_STOCK * pStock, int nSize )
  478. {
  479. for( int i=0; i<nSize; i++ )
  480. {
  481. TW_ASK ask;
  482. int lenSend = ConstructAskMultisortBuffer(ask, &(pStock[i]));
  483. if( lenSend > 0 )
  484. Send(&ask,lenSend);
  485. }
  486. return nSize;
  487. }
  489. int CTWSocket::RequestDetail( TW_STOCK * pStock, int nSize )
  490. {
  491. for( int i=0; i<nSize; i++ )
  492. {
  493. TW_ASK ask;
  494. int lenSend = ConstructAskDetailBuffer(ask,&(pStock[i]));
  495. if( lenSend > 0 )
  496. Send(&ask,lenSend);
  497. }
  498. return nSize;
  499. }
  501. int CTWSocket::RequestBase( TW_STOCK * pStock, int nSize )
  502. {
  503. for( int i=0; i<nSize; i++ )
  504. {
  505. TW_ASK ask;
  506. int lenSend = ConstructAskBaseBuffer(ask,&(pStock[i]));
  507. if( lenSend > 0 )
  508. {
  509. Send(&ask,lenSend);
  510. // m_bIsReceiving = TRUE;
  511. // m_timeReceiveLast = time(NULL);
  512. }
  513. }
  514. return nSize;
  515. }
  517. int CTWSocket::RequestNews( TW_STOCK * pStock, int nSize )
  518. {
  519. //NOTSUPPORT
  520. return 0;
  521. }
  523. int CTWSocket::RequestHtml( TW_STOCK * pStock, int nSize )
  524. {
  525. //NOTSUPPORT
  526. return 0;
  527. }
  529. int CTWSocket::RequestSoftware( TW_STOCK * pStock, int nSize )
  530. {
  531. //NOTSUPPORT
  532. return 0;
  533. }
  535. int CTWSocket::RequestShazqdata( TW_STOCK * pStock, int nSize )
  536. {
  537. //NOTSUPPORT
  538. return 0;
  539. }
  541. // Do not edit the following lines, which are needed by ClassWizard.
  542. #if 0
  543. BEGIN_MESSAGE_MAP(CTWSocket, CSocket)
  544. //{{AFX_MSG_MAP(CTWSocket)
  545. //}}AFX_MSG_MAP
  546. END_MESSAGE_MAP()
  547. #endif // 0
  549. /////////////////////////////////////////////////////////////////////////////
  550. // CTWSocket member functions
  552. void CTWSocket::OnClose(int nErrorCode) 
  553. {
  554. CSocket::Close( );
  556. CSocket::OnClose(nErrorCode);
  558. if( AfxGetQSProfile().GetCycleConnect() )
  559. BeginWorking();
  560. }
  562. void CTWSocket::OnConnect(int nErrorCode) 
  563. {
  564. // TODO: Add your specialized code here and/or call the base class
  566. CSocket::OnConnect(nErrorCode);
  567. }
  569. void CTWSocket::OnReceive(int nErrorCode) 
  570. {
  571. m_timeReceiveLast = time(NULL);
  573. int nReceive = Receive( m_rbuffer, sizeof(m_rbuffer) );
  574. if( nReceive > 0 )
  575. {
  576. CTSCache::GetInstance().OnReceive( m_rbuffer, nReceive );
  578. if( nReceive < 256 && TryGetLength(m_rbuffer,nReceive) < 256 ) // 收到小块包,说明大包接收完毕
  579. m_bIsReceiving = FALSE;
  580. if( nReceive == sizeof(m_rbuffer) )
  581. OnReceive(nErrorCode);
  582. }
  584. CSocket::OnReceive(nErrorCode);
  585. }
  587. void CTWSocket::OnSend(int nErrorCode) 
  588. {
  589. // TODO: Add your specialized code here and/or call the base class
  591. CSocket::OnSend(nErrorCode);
  592. }
  594. int CTWSocket::Receive(void* lpBuf, int nBufLen, int nFlags) 
  595. {
  596. // return CSocket::Receive(lpBuf, nBufLen, nFlags);
  598. if (m_pbBlocking != NULL)
  599. {
  600. WSASetLastError(WSAEINPROGRESS);
  601. return  FALSE;
  602. }
  603. int nResult;
  604. while ((nResult = CAsyncSocket::Receive(lpBuf, nBufLen, nFlags)) == SOCKET_ERROR)
  605. {
  606. if (GetLastError() == WSAEWOULDBLOCK)
  607. {
  608. if (!PumpMessages(FD_READ))
  609. return SOCKET_ERROR;
  610. }
  611. else
  612. return SOCKET_ERROR;
  613. }
  614. return nResult;
  615. }
  617. int CTWSocket::Send(const void* lpBuf, int nBufLen, int nFlags) 
  618. {
  619. // TODO: Add your specialized code here and/or call the base class
  621. return CSocket::Send(lpBuf, nBufLen, nFlags);
  622. }