net_func

#include <winsock2.h>  
#include <stdio.h>
 
#pragma once
 
struct HQServerInfo
{
	char   szServer[32];
	short iPort;
 
	int iCmdId;
};
 
BOOL WinSockInit();
 
unsigned int  __stdcall GetHQAvgCost( void * pInfo  );
 
unsigned int  __stdcall TestFunc( void * pInfo  );
 
unsigned int  __stdcall HQRefreshCheck( void *pPara );
 
 
 
 

#include "stdafx.h"
 
#include <stdio.h>    
#include <string.h>    
#include <iostream>    
 
#include "StockIo.h"
#include "request.h"
#include "Channel.h"
#include "net_func.h"
#include "util.hpp"
#include "NBSocket.h"
 
#pragma comment(lib,"ws2_32.lib")    
 
using namespace std;
 
 
 
BOOL WinSockInit()  
{  
	WSADATA data = {0};  
	if ( WSAStartup( MAKEWORD( 2, 2 ), &data ) ) 
	{
		return FALSE; 
	}
 
	if ( LOBYTE(data.wVersion) !=2 || HIBYTE(data.wVersion) != 2 )
	{  
		WSACleanup();  
		return FALSE;  
	}  
	return TRUE;  
}
 
 
 
 
 
unsigned int  __stdcall GetHQAvgCost( void *pPara )  
{
	CmdQueue_t g_Queue;
	NBSocket  * pSocket = new NBSocket();
 
	Channel * pChannel = new Channel();
 
	pSocket->SetChannel( pChannel );
 
	HQServerInfo * pInfo = (HQServerInfo *)pPara;
 
	//cout << "begin to connect to " << pInfo->szServer << ":" << pInfo->iPort << endl;
	struct timeval tv;
	tv.tv_sec = 3;
	tv.tv_usec = 0;
 
	int iRet = pSocket->Connect( pInfo->szServer, pInfo->iPort, &tv );
 
	if ( iRet == 0 )
	{
		//cout << "connect succ " << endl;
	}
	else
	{
		cout << "connect failed " << endl;
		return -1;
	}
 
	pChannel->SetServerInfo( pInfo );
 
 
	// just to calculate the total stocks in each market
	//AddStockListRequest( false );
 
	int iCount = 0;
 
	while ( true )
	{
		CmdQueue_t &q = g_Queue;
 
		if ( q.empty() &&  iCount < 100   )
		{
			//CmdData_t r( new GetStockCountReq( (uchar)0x00 ) );
			vector<QuotePara> vec;
			QuotePara stPara;
			stPara.cMarket = 0x00;
			memcpy( stPara.sStockCode, "000001", sizeof( stPara.sStockCode ) );
			vec.push_back( stPara );
		
			CmdData_t r( new GetStockQuotesReq(  vec ) );
			q.push_back( r );
			
			//cout << "add one GetStockCountReq cmd" << endl;
			//cout << "add one GetStockQuotesReq cmd" << endl;
			++iCount;
		}
 
		while ( !q.empty() )
		{
			CmdData_t  d = q.front();			// retreive a command
			d->SetSeqId( pChannel->GetSeqId() );
			d->send( pSocket );
 
			ReqTimeInfo stInfo;
			d->SetRequestInfo( stInfo );
 
			pChannel->StartRequest( stInfo );
 
			q.pop_front();
		}
 
		iRet = pSocket->Recv();
		if ( iRet != 0 )
		{
			break;
		}
	 
	}
 
	delete pInfo;
	pInfo = NULL;
 
	delete pChannel;
	pChannel = NULL;
 
	delete pSocket;
	pSocket = NULL;
 
	return -2;
}
 
 
void addRequest( int iCmdId, CmdQueue_t & q)
{
	if ( iCmdId == CMD_GET_STOCK_QUOTES )
	{	
		vector<QuotePara> vec;
		QuotePara stPara;
		stPara.cMarket = 0x00;
		memcpy( stPara.sStockCode, "000001", sizeof( stPara.sStockCode ) );
		vec.push_back( stPara );
		CmdData_t r( new GetStockQuotesReq( vec ) );
		q.push_back( r );
	}
	else if ( iCmdId == CMD_GET_MINUTE_DATA )
	{
		CmdData_t r( new GetMinuteTimeDataReq(  1, "600300") );
		q.push_back( r );
	}
	else if ( iCmdId == CMD_GET_HIS_MINUTE_DATA )
	{
		CmdData_t r( new GetHisMinuteTimeDataReq(  20171026, 1, "600300") );
		q.push_back( r );
	}
	else if ( iCmdId == CMD_GET_TRANSACTION_DATA )
	{
		CmdData_t r( new GetTransactionDataReq(    0, "000001", 0, 30 ) );
		q.push_back( r );
	}
	else if ( iCmdId == CMD_GET_HIS_TRANSACTION_DATA )
	{
		CmdData_t r( new GetHisTransactionDataReq(   20171026,  0, "000001", 0, 30 ) );
		q.push_back( r );
	}
	else if ( iCmdId == CMD_GET_INDEX_BARS )
	{
		CmdData_t r( new GetIndexBarsReq(   9, 1, "000001", 1, 2 ) );
		q.push_back( r );
	}
}
 
 
unsigned int  __stdcall TestFunc( void *pPara )  
{
	CmdQueue_t g_Queue;
	NBSocket  * pSocket = new NBSocket();
 
	Channel * pChannel = new Channel();
 
	pSocket->SetChannel( pChannel );
 
	HQServerInfo * pInfo = (HQServerInfo *)pPara;
 
	//cout << "begin to connect to " << pInfo->szServer << ":" << pInfo->iPort << endl;
	struct timeval tv;
	tv.tv_sec = 3;
	tv.tv_usec = 0;
 
	//114.67.61.70
	//int iRet = pSocket->Connect( pInfo->szServer, pInfo->iPort, &tv );
	int iRet = pSocket->Connect(  "114.67.61.70",  7709, &tv );
	if ( iRet == 0 )
	{
		//cout << "connect succ " << endl;
	}
	else
	{
		cout << "connect failed " << endl;
		return -1;
	}
 
	pChannel->SetServerInfo( pInfo );
 
 
	// just to calculate the total stocks in each market
	//AddStockListRequest( false );
 
	int iCount = 0;
 
	while ( true )
	{
		CmdQueue_t &q = g_Queue;
 
		if ( q.empty() &&  iCount < 1   )
		{
		
			addRequest( pInfo->iCmdId,  g_Queue );
			
			++iCount;
		}
 
		while ( !q.empty() )
		{
			CmdData_t  d = q.front();			// retreive a command
			d->SetSeqId( pChannel->GetSeqId() );
			d->send( pSocket );
 
			ReqTimeInfo stInfo;
			d->SetRequestInfo( stInfo );
			pChannel->StartRequest( stInfo );
 
			q.pop_front();
		}
 
		iRet = pSocket->Recv();
		if ( iRet != 0 )
		{
			break;
		}
	 
	}
 
	delete pInfo;
	pInfo = NULL;
 
	delete pChannel;
	pChannel = NULL;
 
	delete pSocket;
	pSocket = NULL;
 
	return -2;
}
 
 
 
unsigned int  __stdcall HQRefreshCheck( void *pPara )  
{
	CmdQueue_t g_Queue;
	NBSocket  * pSocket = new NBSocket();
 
	Channel * pChannel = new Channel();
 
	pSocket->SetChannel( pChannel );
 
	HQServerInfo * pInfo = (HQServerInfo *)pPara;
 
	//cout << "begin to connect to " << pInfo->szServer << ":" << pInfo->iPort << endl;
	struct timeval tv;
	tv.tv_sec = 3;
	tv.tv_usec = 0;
 
	//114.67.61.70
	//int iRet = pSocket->Connect( pInfo->szServer, pInfo->iPort, &tv );
	int iRet = pSocket->Connect(  "114.67.61.70",  7709, &tv );
	if ( iRet == 0 )
	{
		//cout << "connect succ " << endl;
	}
	else
	{
		cout << "connect failed " << endl;
		return -1;
	}
 
	pChannel->SetServerInfo( pInfo );
 
 
	// just to calculate the total stocks in each market
	//AddStockListRequest( false );
 
	int iCount = 0;
 
	while ( true )
	{
		CmdQueue_t &q = g_Queue;
 
		if ( q.empty()     )
		{
			addRequest( pInfo->iCmdId,  g_Queue );
		}
 
		while ( !q.empty() )
		{
			CmdData_t  d = q.front();			// retreive a command
			d->SetSeqId( pChannel->GetSeqId() );
			d->send( pSocket );
 
			ReqTimeInfo stInfo;
			d->SetRequestInfo( stInfo );
			pChannel->StartRequest( stInfo );
 
			q.pop_front();
		}
 
		iRet = pSocket->Recv();
		if ( iRet != 0 )
		{
			break;
		}
 
		Sleep( 1000 );
	 
	}
 
	delete pInfo;
	pInfo = NULL;
 
	delete pChannel;
	pChannel = NULL;
 
	delete pSocket;
	pSocket = NULL;
 
	return -2;
}