nothing

// NMTaskInfo.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include "NMTaskInfo.h"
#ifdef _WINDOWS_
#include<time.h>
#undef _WINDOWS_
#endif
///////////////////////////全局变量定义区/////////////////////////////////////

CThreadMutux thMutux;  //数据库连接池所用到锁
CDbPool dbPool;    //数据库连接池
SYS_INFO gSysInfo;   //全局配置变量
//CHttpSock httpSocket;
HANDLE handleMutex;

//typedef CDataQueue<Task_Info> TTaskInfo;
//TTaskInfo tTaskInfo(1000);
//TSendInfo *vSendInfo=new TSendInfo();
typedef CDataQueue<TSendInfo> QueueSendInfo;
QueueSendInfo m_QueueSendInfo(5);//建立5个队列
///////////////////////////END END END //////////////////////////////////////

/////////////////////////常量定义///////////////////////////////////////////

#define CFG_FILE "NMTaskInfoCfg.ini"
#define MAX_THREAD_NUM 256

/////////////////////////////////////////////////////////////////////////////

//////////////////////库文件////////////////////////////////////////////////

#pragma comment(lib,"odbc32.lib")

/////////////////////////////////////////////////////////////////////////////

int _tmain(int argc, _TCHAR* argv[])
{
 printf("进入main\n");
 //预处理线程
 //HANDLE   _h_ProDataThread[MAX_THREAD_NUM];
 HANDLE   _h_ProDataThread;
 HANDLE  _h_GetDataThread;

 memset((void*)&gSysInfo,0,sizeof(SYS_INFO));
 gSysInfo.nExitFlag = 0;

 ReadCfgInfo();

 handleMutex =::CreateMutex(NULL,FALSE,(LPCWSTR)gSysInfo.szAppID);
 if(GetLastError()==ERROR_ALREADY_EXISTS){ 
  printf("应用程序已经在运行");
  return 0; 
 }

 //数据库连接
 //dbPool.SetConnInfo("UID=tnm;PWD=tnmadmin;DSN=test15",gSysInfo.nDataConnNum);
 dbPool.SetConnInfo("UID=tnm;PWD=tnmadmin;DSN=test15",10);
 
 if(dbPool.ConnectToPool()<0)
 {
  printf("连接失败!");
  return 0;
 }

 //开始进行处理程序线程，该程序包含一个线程，当我们队列里面的某一个数据时间到了的时候，它就开始执行这个操作
 _h_GetDataThread = CreateThread(NULL,NULL,GetDataThread,(LPVOID)NULL,0,NULL);
 if ( _h_GetDataThread==NULL )
 {
  printf("创建get数据线程失败!\r\n");
  return 0;
 }
 //开始新建读取数据库的线程
 //for(int i=0;i<gSysInfo.nMaxThread;i++)
 //{
 //_h_ProDataThread[i] = CreateThread(NULL,NULL,ProDataThread,(LPVOID)NULL,0,NULL);
 //}
    _h_ProDataThread = CreateThread(NULL,NULL,ProDataThread,(LPVOID)NULL,0,NULL);
 if ( _h_ProDataThread==NULL )
 {
  printf("创建pro数据线程失败!\r\n");
  return 0;
 }
 while(!gSysInfo.nExitFlag)
 {
  dbPool.ReconnectDb();  //重连数据库
  Sleep(1000);
 }

 /*if(_h_GetDataThread!=NULL)
 {
  printf("close getdata \n");
  WaitForSingleObject( _h_GetDataThread, INFINITE);
  CloseHandle(_h_GetDataThread);
 }*/

 /*for(int i=0;i<gSysInfo.nMaxThread;i++)
 {
  if(_h_ProDataThread[i]!=NULL)
  {
   printf("close prodata \n");
   WaitForSingleObject( _h_ProDataThread[i], INFINITE);
   CloseHandle(_h_ProDataThread[i]);
  }
 }*/
 /*if(_h_ProDataThread!=NULL)
  {
   printf("close prodata \n");
   WaitForSingleObject( _h_ProDataThread[i], INFINITE);
   CloseHandle(_h_ProDataThread[i]);
  }*/

 dbPool.DisconnectToPool();
 return 0;
}

DWORD WINAPI GetDataThread( LPVOID pdata)
{

 char sql[1024];
 char szUpdateSql[1024];
 int nPool = -1;
 //Task_Info taskInfo;

 //1.定义两个变量.这两个应该定义得不够
 /*long nLogicID;
 char szSendContent[500];
 char szSendAddr[100];
 int  nLogicType;
 char szExecTime[32];*/

 double  ltmptask_id;    //任务ID
 double ltmpic_id;   //逻辑ID
 //下面几个是配置时间用的
 char tmpSendType[3]; //采集类型 M按月 W按周 D按天
 char tmpSendTime[20]; //确定是某一天（一般是按照星期，比如星期二、星期三）
 char tmpBeginTime[20];   //开始时间，如早上4:00
 char tmpEndTime[20];    //结束时间，如 早上10:00
 //end of time config
 char tmpSendAdd[201]; //发送的地址
 char tmpContent[201]; //发送的内容
 int nLogicDataType;  //数据数据类型 10http   11FTP
 int ntmpIsCheck;   //是否校验
 int ntmpValid;   //是否有效

 while(!gSysInfo.nExitFlag)
 {
  printf("queue tail:%d\t",m_QueueSendInfo.rear);
  TSendInfo *vSendInfo=new TSendInfo();
  printf("进入getData线程\n");
  //数据提取
  /* if(tTaskInfo.GetQueueSize()>900)
  {
  Sleep(1000);
  continue;
  }*/
  if(m_QueueSendInfo.GetQueueSize()>100)
  {
   Sleep(1000);
   continue;
  }
  otl_stream  otl_GetData;
  memset((void*)sql,0,sizeof(sql));
  //sprintf(sql,"select logic_title from tnm_logic_data where rownum<100");

  sprintf(sql,"select tTask.Task_Id,tTask.Logic_Id,tTask.Send_Type,tTask.Send_Time,tTask.Start_Time,tTask.End_Time,tLogic.SEND_ADD,tLogic.Logic_Content,tLogic.LOGIC_DATA_TYPE,tLogic.Is_Check,tLogic.Is_Valid from tnm_logic_data tLogic, tnm_task_info tTask where tLogic.Logic_Id = tTask.logic_id and tLogic.tnm_ne_Id = tTask.ne_id and tLogic.is_valid = 1 and tTask.is_valid = 1 and tLogic.is_check = 1");

  try
  {
   __int64 nLogicID=0;
   nPool = dbPool.GetDbFromPool();
   if(nPool==-1)
   {
    return nPool;
   }

   otl_GetData.open(100,sql,dbPool.tDbInfo[nPool].db);
   while(!otl_GetData.eof())
   {
    TSendInfo *vSendInfo=new TSendInfo();
    memset(tmpSendType,0,sizeof(tmpSendType));
    memset(tmpSendTime,0,sizeof(tmpSendTime));
    memset(tmpBeginTime,0,sizeof(tmpBeginTime));
    memset(tmpEndTime,0,sizeof(tmpEndTime));
    memset(tmpSendAdd,0,sizeof(tmpSendAdd));
    memset(tmpContent,0,sizeof(tmpContent));
    otl_GetData>>ltmptask_id>>ltmpic_id>>tmpSendType>>tmpSendTime>>tmpBeginTime>>tmpEndTime>>tmpSendAdd>>tmpContent>>nLogicDataType>>ntmpIsCheck>>ntmpValid;
    CUtil::trim(tmpSendType);
    CUtil::trim(tmpSendTime);
    CUtil::trim(tmpBeginTime);
    CUtil::trim(tmpEndTime);
    CUtil::trim(tmpSendAdd);
    CUtil::trim(tmpContent);
    vSendInfo->ntask_id=(__int64)ltmptask_id;
    vSendInfo->nlogic_id=(__int64)ltmpic_id;
    sprintf(vSendInfo->szSendType,"%s",tmpSendType);
    sprintf(vSendInfo->szSendTime,"%s",tmpSendTime);
    sprintf(vSendInfo->szBeginTime,"%s",tmpBeginTime);
    sprintf(vSendInfo->szEndtIime,"%s",tmpEndTime);
    sprintf(vSendInfo->szSendAdd,"%s",tmpSendAdd);
    sprintf(vSendInfo->szContent,"%s",tmpContent);
    vSendInfo->nLogicDataType=nLogicDataType;
    vSendInfo->nIsCheck=ntmpIsCheck;
    vSendInfo->nIsValid=ntmpValid;
    printf("执行完取得数据线程!\n");
    //将数据插入到队列中
    if(0==m_QueueSendInfo.WriteDataToQueue(vSendInfo)) //写入成功，则
    {
     printf("...正在将数据读入队列\r\n");
     // delete vSendInfo;
    }
    else
    {
     printf("...数据读入队列出错\r\n");
     return -1;
    }
    //开始执行更新操作
    memset((void*)szUpdateSql,0,sizeof(szUpdateSql));
    nLogicID=vSendInfo->nlogic_id;
    sprintf(szUpdateSql,"update tnm_task_info set is_valid=2 where logic_id=%lld",nLogicID);
    dbPool.ExecSQL(szUpdateSql); //执行更新操作
    delete vSendInfo;
   }
   //执行完成，更新数据库状态
      otl_GetData.close();
       dbPool.FreeDbToPool(nPool);
  }
  catch(otl_exception& p)
  {
   char szErrMsg[1024];
   memset((void*)szErrMsg,0,sizeof(szErrMsg));
   sprintf(szErrMsg,"%s",(char*)p.msg);
   dbPool.tDbInfo[nPool].nConnFlag = 2;
   return (-2);
  }
  Sleep(500);
 }
 return 0;
}

DWORD WINAPI ProDataThread( LPVOID pdata)
{
 //TSendInfo *vSendInfo=new TSendInfo();
 printf("进入proData\n");
 char szUpdateSql[60];
 char szSql[600];
 int nPool=-1;
 char szTime[64];
 char szRecvData[1024];
 char szRetContent[1024]; //比对的信息
 //各个数据段对应的值
 while(!gSysInfo.nExitFlag)
 {
  TSendInfo *pSendInfo=new TSendInfo();
  static int i=0;
  __int64 nLogicId;
  __int64 lcheckId;
  otl_stream otl_ProData;
  int iFindRes;
  memset(szSql,0,sizeof(szSql));
  //线程运行，不断的取出 需要执行的任务
  sprintf(szSql,"select t.task_id as counts from tnm_task_info t where (t.is_valid = 1 and to_date(t.start_time,'yyyy-mm-dd hh24:mi:ss') <= sysdate and t.start_time is not null) and(to_date(t.end_time,'yyyy-mm-dd hh24:mi:ss')  >= sysdate and  t.end_time is not null) and ((t.send_type = 'D' or (t.send_type = 'M' and t.send_time = to_char(sysdate,'DD')) or (t.send_type = 'W' and t.send_time = to_char(to_number(to_char(sysdate,'D'))-1)))) order by t.insert_dt");
  //开始执行
  try
  {
   nPool = dbPool.GetDbFromPool();
   if(nPool==-1)
   {
    return nPool;
   }
   otl_ProData.open(100,szSql,dbPool.tDbInfo[nPool].db);
   while(!otl_ProData.eof())
   {
    //如果取出的数据不为空
    otl_ProData>>nLogicId;
    if(nLogicId>0)
    {
     //开始进行校验
     do
     {
      //m_QueueSendInfo.ReadNextDataFromQueue(pSendInfo,i++);
      iFindRes=m_QueueSendInfo.ReadDataFromQueue(pSendInfo);
     }
     while(pSendInfo->nlogic_id!=nLogicId);
     //{
      //开始执行发送等操作
      //判断
     if(iFindRes==0)
     {
      if(10==(pSendInfo->nLogicDataType))
      {
       /*
       httpSocket.InitEnv(NULL,vSendInfo->szSendAdd,80); //默认为80端口
       httpSocket.SetHttpTimeOut(5000000);
       if(httpSocket.SendPostData(vSendInfo->szContent))
       {
        printf("发送数据失败\n\r");
       }
       else
       {
        //取得返回的数据
        memset(szRecvData,0,sizeof(szRecvData));
        httpSocket.GetHeadData(szRecvData); //此时szRecvData保存服务器返回的数据
        //有数据返回
        if(strlen(szRecvData)>0)
        {
         //搜索数据库 
         otl_stream otl_logicCheck;
         memset((void*)szSql,0,sizeof(szSql));
         //sprintf(szSql,"select ret_content from tnm_logic_check where logic_id=%s and is_valid=1",vSendInfo->nlogic_id);
         sprintf(szSql,"select c1.check_id from tnm_log_check c1 where check_id not in(select c2.check_id from tnm_logic_check c2  where re_content like '%%s%' and  c1.logic_id=c2.logic_id and c2.logic_id=%d and c2.is_vaild=1",szRecvData,pSendInfo->nlogic_id);
         try
         {
          nPool=dbPool.GetDbFromPool();
          if(-1==nPool)
          {
           Sleep(2000);
           return nPool;
          }
          else
          {
           otl_logicCheck.open(100,szSql,dbPool.tDbInfo[nPool].db);
           if(!otl_logicCheck.eof()) //logic_id 一定要是唯一的，不然这里就会有错误
           {
            //说明有出错的情况存在
            otl_logicCheck>>lcheckId;
           }
           else
           {
            printf("检验未出错\n\r");
            break;
           }

           //进行消息校验，校验通过的时候就把任务采集表的数据更新为1
           //将数据插入到 报警接口表当中，暂时未做
           memset(szSql,0,sizeof(szSql));
           //返回的消息与校验库的消息一致
           if(strcmp(szRetContent,szRecvData))
           {
            memset((void*)szUpdateSql,0,sizeof(szUpdateSql));
            sprintf(szUpdateSql,"update tnm_task_info set is_valid=1,LAST_UPDATE_TIME=to_char(sysdate,'yyy-mm-dd hh24:mi:ss') where logic_id=%ld",vSendInfo->nlogic_id);
            dbPool.ExecSQL(szUpdateSql); //执行更新操作
           }
          }
         }
         catch(otl_exception &p)
         {
          char szMsg[1024];
          memset((void*)szMsg,0,sizeof(szMsg));
          sprintf(szMsg,"%s",(char*)p.msg);
          dbPool.tDbInfo[nPool].nConnFlag=2;
          return -2;
         }
        }
       }*/
      }
      else if(11==(int)(pSendInfo->szSendType))
      {
       //进行Ftp的发送操作
        printf("执行FTP操作\n");

      }
     //} //ZHANGXQ
     }
     else
     {
      printf("当前没有任务需要进行\n");
      Sleep(500);
     }
     }
    
   }
   ////TSendInfo *pSendInfo=new TSendInfo();
   ////第一步：取出时间，判断当前某一个时间段该执行操作，如果有，就发送并等待返回。返回的时候再把它和
   //memset(szTime,0,sizeof(szTime));
   ///*szTime=CUtil::getTime();*/
   //strcpy(szTime,CUtil::getTime());
   ////第二步：判断时间是否到了
   //m_QueueSendInfo.ReadDataFromQueue(vSendInfo); //获得队列的首元素
   ////时间到了

   //执行Sql语句判断是否有值
   //if(strcmp(vSendInfo->szExecTime,szTime))
   //{
    //为HTTP处理类型
    if(10==(pSendInfo->nLogicDataType))
    {
     /*
     httpSocket.InitEnv(NULL,vSendInfo->szSendAdd,80); //默认为80端口
     httpSocket.SetHttpTimeOut(5000000);
     if(httpSocket.SendPostData(vSendInfo->szContent))
     {
      printf("发送数据失败\n\r");
     }
     else
     {
      //取得返回的数据
      memset(szRecvData,0,sizeof(szRecvData));
      httpSocket.GetHeadData(szRecvData); //此时szRecvData保存服务器返回的数据
      //有数据返回
      if(strlen(szRecvData)>0)
      {
       //搜索数据库 
       otl_stream otl_logicCheck;
       memset((void*)szSql,0,sizeof(szSql));
       sprintf(szSql,"select ret_content from tnm_logic_check where logic_id=%s and is_valid=1",vSendInfo->nlogic_id);
       try
       {
        nPool=dbPool.GetDbFromPool();
        if(-1==nPool)
        {
         Sleep(2000);
         return nPool;
        }
        else
        {
         otl_logicCheck.open(100,szSql,dbPool.tDbInfo[nPool].db);
         while(!otl_logicCheck.eof()) //logic_id 一定要是唯一的，不然这里就会有错误
         {
          otl_logicCheck>>szRetContent;
         }
         //进行消息校验，校验通过的时候就把任务采集表的数据更新为1
         if(strlen(szRetContent))
         {
          //返回的消息与校验库的消息一致
          if(strcmp(szRetContent,szRecvData))
          {
           memset((void*)szUpdateSql,0,sizeof(szUpdateSql));
           sprintf(szUpdateSql,"update tnm_task_info set is_valid=1 where logic_id=%ld",vSendInfo->nlogic_id);
           dbPool.ExecSQL(szUpdateSql); //执行更新操作
          }
         }
        }
       }
       catch(otl_exception &p)
       {
        char szMsg[1024];
        memset((void*)szMsg,0,sizeof(szMsg));
        sprintf(szMsg,"%s",(char*)p.msg);
        dbPool.tDbInfo[nPool].nConnFlag=2;
        return -2;
       }
      }
     }*/
    }
   //} //zhangxq
   /*Sleep(10);
   otl_stream otl_getTaskData;
   memset((void*)szSql,0,sizeof(szSql));*/  //为何要变成空指针
   //sprintf(szSql,"select task_id,logic_id,ne_id,exec_time,is_valid from tnm_task_info where rownum<2");
   //开始执行
   //1.判断是否有空闲的数据库连接
   /*
   try
   {
   nPool=dbPool.GetDbFromPool();
   if((-1)==nPool)
   {
   Sleep(1000);
   return nPool;
   }
   else
   {
   //2.开始执行数据库操作
   otl_getTaskData.open(100,szSql,dbPool.tDbInfo[nPool].db);   
   //3.获取数据库的值
   while(!otl_getTaskData.eof())
   { //1.取出数据的值，注意数据类型要严格匹配
   long nTmpTaskID,nTmpLogicID,ntmpNeID;
   otl_getTaskData>>nTmpTaskID;
   //otl_getTaskData>>tmpTaskID>>tmpLogicID>>tmpNeID>>tmpExecTime>>nIsValid;
   //将这些数据添加到数据单元中
   int i=0;
   //itoa(nTmpTaskID,tmpTaskID,10); //把整形转为char*
   sprintf(tmpTaskID,"%ld",nTmpTaskID);
   Task_Info *vTaskInfo=new Task_Info;
   memset(vTaskInfo->szTaskID,0,sizeof(vTaskInfo->szTaskID));
   memset(vTaskInfo->szLogicID,0,sizeof(vTaskInfo->szLogicID));
   memset(vTaskInfo->szNeID,0,sizeof(vTaskInfo->szNeID));
   memset(vTaskInfo->)
   memset(vTaskInfo->szExecTime,0,sizeof(vTaskInfo->szExecTime));
   strcpy(vTaskInfo->szTaskID,tmpTaskID);
   strcpy(vTaskInfo->szLogicID,tmpLogicID);
   strcpy(vTaskInfo->szExecTime,tmpExecTime);
   strcpy(vTaskInfo->szNeID,tmpNeID);
   vTaskInfo->nIsValid=nIsValid;

   CUtil::trim(vTaskInfo->szTaskID);
   CUtil::trim(vTaskInfo->szLogicID);
   CUtil::trim(vTaskInfo->szExecTime);
   CUtil::trim(vTaskInfo->szNeID);
   //添加节点
   tTaskInfo.WriteDataToQueue(vTaskInfo);
   delete vTaskInfo;
   }
   //把数据放置到双向队列之后，我们将有处理的那些的标示更为成2
   //printf("操作结果:%s",szTaskData);
   }
   }*/

   otl_ProData.close();
   dbPool.FreeDbToPool(nPool);
   Sleep(100);
  }
  catch(otl_exception &p)
  {
  
  }
  delete pSendInfo;
 }
return 0;
}

int ReadCfgInfo()
{
 int nRetCode = 0;
 char szTmp[20];
 //DataBase
 nRetCode = ReadConfig(CFG_FILE,"[Database]","connstr",gSysInfo.szConnData);

 //ConnNum
 memset((void*)szTmp,0,sizeof(szTmp));
 nRetCode = ReadConfig(CFG_FILE,"[Database]","ConnNum",szTmp);
 gSysInfo.nDataConnNum = atoi(szTmp);
 if(gSysInfo.nDataConnNum<=0) gSysInfo.nDataConnNum = 1;

 //AppTitle
 nRetCode = ReadConfig(CFG_FILE,"[App]","AppTitle",gSysInfo.szAppTitle);
 nRetCode = ReadConfig(CFG_FILE,"[App]","AppID",gSysInfo.szAppID);

 //
 //ConnTimeOut
 memset((void*)szTmp,0,sizeof(szTmp));
 nRetCode = ReadConfig(CFG_FILE,"[TimeOut]","ConnTimeOut",szTmp);
 gSysInfo.nConnTimeOut = atoi(szTmp);
 if(gSysInfo.nConnTimeOut<=0) gSysInfo.nConnTimeOut = 20;

 //SendTimeOut
 memset((void*)szTmp,0,sizeof(szTmp));
 nRetCode = ReadConfig(CFG_FILE,"[TimeOut]","SendTimeOut",szTmp);
 gSysInfo.nSendTimeOut = atoi(szTmp);
 if(gSysInfo.nSendTimeOut<=0) gSysInfo.nSendTimeOut = 20;

 //SendTimeOut
 memset((void*)szTmp,0,sizeof(szTmp));
 nRetCode = ReadConfig(CFG_FILE,"[TimeOut]","RecvTimeOut",szTmp);
 gSysInfo.nRecvTimeOut = atoi(szTmp);
 if(gSysInfo.nRecvTimeOut<=0) gSysInfo.nRecvTimeOut = 20;

 //Thread
 memset((void*)szTmp,0,sizeof(szTmp));
 nRetCode = ReadConfig(CFG_FILE,"[Thread]","MAX_THREAD_NUM",szTmp);
 gSysInfo.nMaxThread = atoi(szTmp);
 if(gSysInfo.nMaxThread<=0) gSysInfo.nMaxThread = 1;
 if(gSysInfo.nMaxThread>MAX_THREAD_NUM) gSysInfo.nMaxThread = MAX_THREAD_NUM;

 nRetCode = ReadConfig(CFG_FILE,"[Log]","LogPath",gSysInfo.szLogPath);
 //SendTimeOut
 memset((void*)szTmp,0,sizeof(szTmp));
 nRetCode = ReadConfig(CFG_FILE,"[Log]","LogLevel",szTmp);
 gSysInfo.nLogLevel = atoi(szTmp);
 if(gSysInfo.nLogLevel<=0) gSysInfo.nLogLevel = 1;

 return 0;
}

int ReadConfig(char *file,char* title, char* token, char* value)
{
 char  buffer[1024];
 char  *readpoint,*ptr;
 char  tempstr[256];
 FILE  *fp;
 int   i;
 int   length;

 memset(tempstr,0,256);
 memset(buffer, 0,1024);
 if((fp=fopen(file,"r"))==NULL)
 {
  // printf("无法打开客户的配置文件,程序终止!");
  return -1;
 }
 fseek(fp,0l,SEEK_SET);
 i=fread(buffer,1024,1,fp);
 ptr = buffer;
 readpoint=(char*)strstr((const char *)ptr,title);
 if(readpoint!=NULL)
 {
  ptr = readpoint + strlen(title);
  readpoint=(char*)strstr((const char *)ptr,token);
  if(readpoint!=NULL)
  {
   length=strlen(token);
   for(i=0;i<256;i++)
   {
    if (*(readpoint+length+i)!='\n')
    {
     tempstr[i]=*(readpoint+length+i);
    }
    else
    {
     tempstr[i]='\0';
     break;
    }
   }
  }
 }
 strcpy(value,tempstr);
 fclose(fp);
 return 0;
}