vthreadpoolconnecttest.cxx

RTP协议

/* ====================================================================
 * The Vovida Software License, Version 1.0 
 * 
 * Copyright (c) 2000 Vovida Networks, Inc.  All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 
 * 2. 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.
 * 
 * 3. The names "VOCAL", "Vovida Open Communication Application Library",
 *    and "Vovida Open Communication Application Library (VOCAL)" must
 *    not be used to endorse or promote products derived from this
 *    software without prior written permission. For written
 *    permission, please contact [e-mail].
 *
 * 4. Products derived from this software may not be called "VOCAL", nor
 *    may "VOCAL" appear in their name, without prior written
 *    permission of Vovida Networks, Inc.
 * 
 * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESSED OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, TITLE AND
 * NON-INFRINGEMENT ARE DISCLAIMED.  IN NO EVENT SHALL VOVIDA
 * NETWORKS, INC. OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT DAMAGES
 * IN EXCESS OF $1,000, NOR FOR ANY 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.
 * 
 * ====================================================================
 * 
 * This software consists of voluntary contributions made by Vovida
 * Networks, Inc. and many individuals on behalf of Vovida Networks,
 * Inc.  For more information on Vovida Networks, Inc., please see
 * <http://www.vovida.org/>.
 *
 */

static const char* const VThreadPoolConcurrencyTest_cxx_Version =
    "$Id: VThreadPoolConnectTest.cxx,v 1.1 2001/07/20 02:29:50 bko Exp $";

#include "VThreadPool.cxx"
#include "Condition.hxx"
#include "cpLog.h"
#include "Tcp_ServerSocket.hxx"
#include <map>
#include "VMutex.h"
#include "Lock.hxx"

using Vocal::Threads::Lock;
VMutex myMutex;


class ConnContainer
{
    public:
	Connection conn;
	bool active;

	ConnContainer()
	    : active(false)
	{
	}
	ConnContainer(const Connection& c)
	    : conn(c),
	      active(false)
	{
	}

	ConnContainer(const ConnContainer& c)
	    : conn(c.conn),
	      active(c.active)
	{
	}
	const ConnContainer& operator=(const ConnContainer& c)
	{
	    if(&c != this)
	    {
		conn = c.conn;
		active = c.active;
	    }
	    return *this;
	}
};

map <int, ConnContainer> myMap;

void doWork(void* msg)
{
    ConnContainer* conn;
    conn = reinterpret_cast<ConnContainer*>(msg);
    assert(conn);
    cpLog(LOG_ERR, "got connection");

    char buf[256];
    int num;
    conn->conn.readLine(buf, 256, num);
    vusleep(100);
    int fd = conn->conn.getConnId();
    conn->conn.close();
    Lock myLock(myMutex);
    myMap.erase(fd);
}


int main()
{
    VThreadPool pool(10);
    TcpServerSocket server(5000);

    for(;;)
    {
	if(server.getServerConn().isReadReady())
	{
	    ConnContainer conn;
	    server.accept(conn.conn);
	    myMutex.lock();
	    myMap[conn.conn.getConnId()] = conn;
	    myMutex.unlock();
	}

	myMutex.lock();
	for(map <int, ConnContainer>::iterator i = myMap.begin(); 
	    i != myMap.end() ; 
	    ++i)
	{
	    ConnContainer* ptr = &((*i).second);
	    if(ptr->conn.isLive() && ptr->conn.isReadReady())
	    {
		VFunctor func(doWork, ptr);
		pool.addFunctor(func);
	    }
	}
	myMutex.unlock();

    }
}