oneproxyserver.cpp

/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements.  See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership.  The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License.  You may obtain a copy of the License at
*
*   http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied.  See the License for the
* specific language governing permissions and limitations
* under the License.
*
* @ClassName: oneproxyserver.cpp
* @Description: TODO
* All rights Reserved, Designed By huih
* @Company: onexsoft
* @Author: hui
* @Version: V1.0
* @Date: 2016年7月29日
*
*/

#include "tool.h"
#include "record.h"
#include "oneproxyserver.h"
#include "connectmanager.h"
//#include "google/profiler.h"

thread_start_func(OneproxyServer::start)
{
	OneproxyServer *os = (OneproxyServer*)args;

	if (os->create_tcpServer()) {
		logs(Logger::FATAL, "create tcp server error");
		return 0;
	}
	os->start_success();

	os->get_ioEvent().regester_checkQuit();
	os->get_ioEvent().run_loop();

	return 0;
}

void OneproxyServer::accept_clientRequest(NetworkSocket *clientSocket)
{
	if (this->connectManager == NULL)
		return;

	unsigned int clientHashCode = Tool::quick_hash_code(clientSocket->get_address().c_str(), clientSocket->get_address().length());
	clientSocket->set_addressHashCode(clientHashCode);
	clientSocket->get_attachData().set_connTime(config()->get_globalMillisecondTime());
	unsigned int connHashCode = Tool::quick_conn_hash_code(clientSocket->get_address(),
			clientSocket->get_port(), clientSocket->get_attachData().get_listenAddr(),
			clientSocket->get_attachData().get_listenPort(), clientSocket->get_attachData().get_connTime());
	clientSocket->get_attachData().set_connHashCode(connHashCode);

	record()->record_clientQueryAddNewClient(connHashCode,
			clientSocket->get_addressHashCode(),
			clientSocket->get_address(),
			clientSocket->get_port(),
			clientSocket->get_attachData().get_listenAddr(),
			clientSocket->get_attachData().get_listenPort());

	logs(Logger::DEBUG, "accept fd: %d", clientSocket->get_fd());
	if (this->connectManager->get_taskSize() > 0) {
		this->connectManager->add_task(clientSocket);
	} else {
		ClientThread* ct = this->connectManager->get_minTaskThread();
		if (ct != NULL) {
			if (ct->write_socketPair((void*)(&clientSocket), sizeof(clientSocket))) {
				logs(Logger::ERR, "write client connection to thread(%llu) error", ct->get_threadId());
				delete clientSocket;
			}
		} else {
			this->connectManager->add_task(clientSocket);
		}
	}
	record()->record_acceptClientConn();
}