Exemplo n.º 1
0
    def execute(self, req, keyspace=None):
        if self.my_proto is None:
            return defer.fail(error.ConnectionClosed(
                                    'Lost connection before %s request could be made'
                                    % (req.method,)))
        method = getattr(self.my_proto.client, req.method, None)
        if method is None:
            raise InvalidThriftRequest("don't understand %s request" % req.method)


        def wrap_thrift_request(method, *args):
            # convenience wrapper around thrift request with timeout support
            d = defer.Deferred()

            # institute a (fairly long) hard timeout
            def short_circuit():
                err = Exception('Timeout: %r %r' % (method, args))
                if not d.called:
                    d.errback(err)
            timeout = reactor.callLater(10, short_circuit)

            req_d = defer.succeed(True)
            # the actual thrift call
            req_d.addCallback(lambda _ : method(*args))

            def pipe_callback(res):
                if timeout.active():
                    timeout.cancel()
                if d.called:
                    log.msg('REQ_TIMEOUT: (%r, %r) -> %r' % (method, args, res))
                else:
                    d.callback(res)

            def pipe_errback(e):
                if timeout.active():
                    timeout.cancel()
                if d.called:
                    log.err(e, 'REQ_TIMEOUT: Error (%r, %r)' % (method, args))
                else:
                    d.errback(e)

            # make sure the results gets wired to the right place
            req_d.addCallbacks(pipe_callback, pipe_errback)
            return d

        d = defer.succeed(0)
        if req.method == 'set_keyspace':
            newksname = req.args[0]
            d.addCallback(lambda _: wrap_thrift_request(method, newksname))
            d.addCallback(self.store_successful_keyspace_set, newksname)
        else:
            if keyspace is not None and keyspace != self.keyspace:
                d.addCallback(lambda _: self.my_set_keyspace(keyspace))
            args = translateArgs(req, self.api_version)
            d.addCallback(lambda _: wrap_thrift_request(method, *args))
            d.addCallback(lambda results: postProcess(results, req.method))
        return d
Exemplo n.º 2
0
 def submitRequest(self, request):
     if not self.deferred:
         fun = getattr(self.client, request.method, None)
         if not fun:
             raise InvalidThriftRequest
         else:
             args = translate.translateArgs(request, self.api_version)
             d = fun(*args)
             d.addCallback(lambda results: translate.postProcess(results, request.method))
         self.deferred = d
         d.addBoth(self._complete)
         return d
     else:
         raise ClientBusy
Exemplo n.º 3
0
 def submitRequest(self, request):
     if not self.deferred:
         fun = getattr(self.client, request.method, None)
         if not fun:
             raise InvalidThriftRequest
         else:
             args = translate.translateArgs(request, self.api_version)
             d = fun(*args)
             d.addCallback(lambda results: translate.postProcess(
                 results, request.method))
         self.deferred = d
         d.addBoth(self._complete)
         return d
     else:
         raise ClientBusy
Exemplo n.º 4
0
    def execute(self, req, keyspace=None):
        if self.my_proto is None:
            return defer.errback(error.ConnectionClosed(
                                    'Lost connection before %s request could be made'
                                    % (req.method,)))
        method = getattr(self.my_proto.client, req.method, None)
        if method is None:
            raise InvalidThriftRequest("don't understand %s request" % req.method)

        d = defer.succeed(0)

        if req.method == 'set_keyspace':
            newksname = req.args[0]
            d.addCallback(lambda _: method(newksname))
            d.addCallback(self.store_successful_keyspace_set, newksname)
        else:
            if keyspace is not None and keyspace != self.keyspace:
                d.addCallback(lambda _: self.my_set_keyspace(keyspace))
            args = translateArgs(req, self.api_version)
            d.addCallback(lambda _: method(*args))
            d.addCallback(lambda results: postProcess(results, req.method))
        return d
Exemplo n.º 5
0
    def execute(self, req, keyspace=None):
        if self.my_proto is None:
            return defer.errback(
                error.ConnectionClosed(
                    'Lost connection before %s request could be made' %
                    (req.method, )))
        method = getattr(self.my_proto.client, req.method, None)
        if method is None:
            raise InvalidThriftRequest("don't understand %s request" %
                                       req.method)

        d = defer.succeed(0)

        if req.method == 'set_keyspace':
            newksname = req.args[0]
            d.addCallback(lambda _: method(newksname))
            d.addCallback(self.store_successful_keyspace_set, newksname)
        else:
            if keyspace is not None and keyspace != self.keyspace:
                d.addCallback(lambda _: self.my_set_keyspace(keyspace))
            args = translateArgs(req, self.api_version)
            d.addCallback(lambda _: method(*args))
            d.addCallback(lambda results: postProcess(results, req.method))
        return d