ControlManager.cpp

/*
 * Copyright (c) 2012, chlaws <iamwljiang at gmail.com>
 * All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in the
 *       documentation and/or other materials provided with the distribution.
 *     * Neither the name of the <organization> nor the
 *       names of its contributors may be used to endorse or promote products
 *       derived from this software without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "ControlManager.h"
#include "Strutil.h"
#include "ssignal.h"
unsigned int CControlManager::THREAD_NUMBER = 1;
static unsigned int global_exit_flag = 0;

extern Logger run_logger;
extern Logger result_logger;


CControlManager::CControlManager()
{
	server_address = "";

	server_port = -1;

	hospital_name = "";

	department_name = "";

	doctor_name = "";

	run_log_file = "";

	result_log_file = "";

	day = "";
	
	thread_number = 1;

	connection_number = 1;
	apr_initialize();
}


CControlManager::~CControlManager()
{
	apr_terminate();
	for(int i = 0; i < thread_pools.size(); ++i){
		delete thread_pools.at(i);
	}
}

int CControlManager::Init(const char* address,int port)
{

	int r = 0;
	server_address.assign(address);
	server_port = port;
	if((r = first_connection.Init(address,port)) < 0){
		std::cout << "CControlManager::Init init first connection error\n";
		return r;
	}

	if(run_logger.init("error","running.log") != 0){
		return -11;
	}

	if(result_logger.init("info","success.log","./","./",NULL,NULL,true) != 0){
		return -12;
	}

	return 0;	
}


int CControlManager::GetHositalList(void *out)
{
	int rv = 0;
	rv = first_connection.RunOnceGetHosmap(out);
	return rv == 0 ? 0 : rv;
}

int CControlManager::GetDepartListOfHospital(const std::string& hos_name, void *out)
{
	int rv = 0;
	rv = first_connection.RunOnceGetDepmap(hos_name,out);
	return rv == 0 ? 0 : rv;
}

int CControlManager::GetDoctorListOfDepart(const std::string&   depart_name,void *out)
{
	int rv = 0;
	rv = first_connection.RunOnceGetDocmap(depart_name,out);
	return rv == 0 ? 0 : rv;
}

int CControlManager::TestLogin(const std::string& user,const std::string& passwd)
{
	int rv = 0;
	rv = first_connection.RunOnceTestLogin(user,passwd);
	return rv == 0 ? 0 : rv;
}
void CControlManager::SetWeekday(const std::string& day)
{
	this->day = day;
	first_connection.SetDay(day);
}

void CControlManager::SetLogfile(const std::string& run_log,const std::string& result_log)
{
	this->run_log_file = run_log;
	this->result_log_file = result_log;
}

void CControlManager::SetConcurrent(const std::string& thread_number,const std::string& connection_number)
{
	if(!thread_number.empty()){
		this->thread_number = atoi(thread_number.c_str());
	}

	if(!connection_number.empty()){
		this->connection_number = atoi(connection_number.c_str());
	}
}

int CControlManager::Run(const std::string& hos,const std::string& dep,const std::string &doc)
{
	int count_threads = 0;
	int last_time = time(NULL);
	
	CHttpProcesser::MAX_CONNECTION = connection_number;
	THREAD_NUMBER = thread_number;

	while(!apr_atomic_read32(&global_exit_flag)){
		count_threads = thread_pools.size();
		if(count_threads < THREAD_NUMBER){
			CHttpProcesser* http_proc = new CHttpProcesser();
			http_proc->Init(server_address.c_str(),server_port);
			http_proc->Clone(first_connection);
			http_proc->SetDatename(hos,dep,doc);
			http_proc->start_thread();
			thread_pools.push_back(http_proc);
		}

		time_t now = time(NULL);
		if(now - last_time > 60){
			//break;
			struct tm * ptm = localtime(&now);
			//放号前,增加并发量
			/*
			if(ptm->tm_hour == 13 && ptm->tm_min > 50){
				THREAD_NUMBER = 4;	
			}	
			//放号结束后,减少并发量
			if(ptm->tm_hour == 15 && ptm->tm_min > 10){
				THREAD_NUMBER = 4;
			}
			*/
			last_time = now;
		}

		sleep(100);
	}
	
	for(int i = 0; i < thread_pools.size(); ++i){
		thread_pools.at(i)->Stop();
	}

	//等待线程退出
	usleep(2000);
	return 0;
}

void CControlManager::Stop()
{
	//接受到退出的信号量
	apr_atomic_set32 (&global_exit_flag,1);
}