valitydev/thrift
14
0
Fork 0
mirror of https://github.com/valitydev/thrift.git synced 2024-11-07 02:45:22 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

THRIFT-658. python: Add THttpClient.setTimeout()

Browse Source 
git-svn-id: https://svn.apache.org/repos/asf/incubator/thrift/trunk@920660 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
David Reiss 2010-03-09 05:19:16 +00:00
parent bec15e5105
commit ff3d249d41
1 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
lib/py/src/transport/THttpClient.py
View File

@ -23,6 +23,7 @@ from cStringIO import StringIO
import urlparse import urlparse
import httplib import httplib
import warnings import warnings
import socket
class THttpClient(TTransportBase): class THttpClient(TTransportBase):
@ -55,6 +56,7 @@ class THttpClient(TTransportBase):
self.path = parsed.path self.path = parsed.path
self.__wbuf = StringIO() self.__wbuf = StringIO()
self.__http = None self.__http = None
self.__timeout = None
def open(self): def open(self):
if self.scheme == 'http': if self.scheme == 'http':
@ -69,12 +71,30 @@ class THttpClient(TTransportBase):
def isOpen(self): def isOpen(self):
return self.__http != None return self.__http != None
def setTimeout(self, ms):
if not hasattr(socket, 'getdefaulttimeout'):
raise NotImplementedError
if ms is None:
self.__timeout = None
else:
self.__timeout = ms/1000.0
def read(self, sz): def read(self, sz):
return self.__http.file.read(sz) return self.__http.file.read(sz)
def write(self, buf): def write(self, buf):
self.__wbuf.write(buf) self.__wbuf.write(buf)
def __withTimeout(f):
def _f(*args, **kwargs):
orig_timeout = socket.getdefaulttimeout()
socket.setdefaulttimeout(args[0].__timeout)
result = f(*args, **kwargs)
socket.setdefaulttimeout(orig_timeout)
return result
return _f
def flush(self): def flush(self):
if self.isOpen(): if self.isOpen():
self.close() self.close()
@ -98,3 +118,7 @@ class THttpClient(TTransportBase):
# Get reply to flush the request # Get reply to flush the request
self.code, self.message, self.headers = self.__http.getreply() self.code, self.message, self.headers = self.__http.getreply()
# Decorate if we know how to timeout
if hasattr(socket, 'getdefaulttimeout'):
flush = __withTimeout(flush)