From 02d1edc6f041b968a0f02d01c767f37eda3fab27 Mon Sep 17 00:00:00 2001
From: Mark Slee <mcslee@apache.org>
Date: Wed, 7 Mar 2007 05:17:25 +0000
Subject: [PATCH] Some thrift test code improvements

git-svn-id: https://svn.apache.org/repos/asf/incubator/thrift/trunk@665050 13f79535-47bb-0310-9956-ffa450edef68
---
 test/rb/TestClient.rb          |  2 +-
 test/rb/TestServer.rb          |  2 +-
 test/threads/ThreadsClient.cpp |  1 +
 test/threads/ThreadsServer.cpp | 18 ++++++++++++++----
 4 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/test/rb/TestClient.rb b/test/rb/TestClient.rb
index beb5af1d..a4aecb8d 100755
--- a/test/rb/TestClient.rb
+++ b/test/rb/TestClient.rb
@@ -1,4 +1,4 @@
-#!/usr/bin/ruby
+#!/usr/bin/env ruby
 
 $:.push('gen-rb')
 $:.push('../../lib/rb/lib')
diff --git a/test/rb/TestServer.rb b/test/rb/TestServer.rb
index 62b73f76..6e083135 100755
--- a/test/rb/TestServer.rb
+++ b/test/rb/TestServer.rb
@@ -1,4 +1,4 @@
-#!/usr/bin/ruby
+#!/usr/bin/env ruby
 
 $:.push('gen-rb')
 $:.push('../../lib/rb/lib')
diff --git a/test/threads/ThreadsClient.cpp b/test/threads/ThreadsClient.cpp
index d8f67669..7ad6f00b 100644
--- a/test/threads/ThreadsClient.cpp
+++ b/test/threads/ThreadsClient.cpp
@@ -10,6 +10,7 @@
 #include <thrift/concurrency/ThreadManager.h>
 #include <thrift/concurrency/PosixThreadFactory.h>
 
+using boost::shared_ptr;
 using namespace facebook::thrift;
 using namespace facebook::thrift::protocol;
 using namespace facebook::thrift::transport;
diff --git a/test/threads/ThreadsServer.cpp b/test/threads/ThreadsServer.cpp
index f5bd665c..7d17ecab 100644
--- a/test/threads/ThreadsServer.cpp
+++ b/test/threads/ThreadsServer.cpp
@@ -4,12 +4,14 @@
 #include "ThreadsTest.h"
 #include <protocol/TBinaryProtocol.h>
 #include <server/TThreadPoolServer.h>
+#include <server/TThreadedServer.h>
 #include <transport/TServerSocket.h>
 #include <transport/TTransportUtils.h>
 #include <thrift/concurrency/Monitor.h>
 #include <thrift/concurrency/ThreadManager.h>
 #include <thrift/concurrency/PosixThreadFactory.h>
 
+using boost::shared_ptr;
 using namespace facebook::thrift;
 using namespace facebook::thrift::protocol;
 using namespace facebook::thrift::transport;
@@ -84,23 +86,31 @@ int main(int argc, char **argv) {
   shared_ptr<TTransportFactory> transportFactory(new TBufferedTransportFactory());
   shared_ptr<TProtocolFactory> protocolFactory(new TBinaryProtocolFactory());
 
+  /*
   shared_ptr<ThreadManager> threadManager = 
     ThreadManager::newSimpleThreadManager(10);
   shared_ptr<PosixThreadFactory> threadFactory =
     shared_ptr<PosixThreadFactory>(new PosixThreadFactory());
   threadManager->threadFactory(threadFactory);
   threadManager->start();
-  
-  shared_ptr<TServer> threadPoolServer =
+ 
+  shared_ptr<TServer> server =
     shared_ptr<TServer>(new TThreadPoolServer(processor,
                                               serverTransport,
                                               transportFactory,
                                               protocolFactory,
                                               threadManager));
+  */
 
-  handler->setServer(threadPoolServer);
+  shared_ptr<TServer> server =
+    shared_ptr<TServer>(new TThreadedServer(processor,
+                                            serverTransport,
+                                            transportFactory,
+                                            protocolFactory));
 
-  threadPoolServer->serve();
+  handler->setServer(server);
+
+  server->serve();
 
   fprintf(stderr, "done.\n");
 
-- 
2.17.1