Question

我有XML-RPC服务器：

import time

import xmlrpclib
from SimpleXMLRPCServer import SimpleXMLRPCServer

class Worker(object):
    def start_work(self):
        # is it possible do return value to client here?
        self.do_work()
        return 'we started!'

    def do_work(self):
        while(True):
            print 'I\'m doing work...'
            time.sleep(3)


if __name__ == '__main__':
    port = 8080
    server = SimpleXMLRPCServer(("localhost", port))
    print "Listening on port %s..." % port

    w = Worker()

    server.register_function(w.start_work)
    while(1):
        server.handle_request()

# vim: filetype=python syntax=python expandtab shiftwidth=4 softtabstop=4 encoding=utf8

轻松的客户：

import xmlrpclib
c = xmlrpclib.ServerProxy('http://localhost:8080/')
print c.start_work()

当然，start_work函数返回的值永远不会打印出来。

我的问题是如何重写服务器代码，以便在完成工作之前返回值。我知道我可以使用线程，但我想确保没有更简单的方法。

Answer 1

如果您希望XML-RPC具有长时间运行的早期返回任务，您可能需要将服务器重写为异步框架，例如twisted

Answer 2

我不确定在没有线程或没有重新实现SimpleXMLRPCServer的某些部分或没有破坏XMLRPC协议的情况下做这样的事情是个好主意，但是你可以通过使用yield语句得到早期答案在你的被叫方法中：

class Worker(object):
    def start_work(self):
        # is it possible do return value to client here?
        # yes, with a yield
        yield "the_return_value"
        self.do_work()
        # but yielding a first value and returning it to client will disable any next response
        # return 'we started!'

    def do_work(self):
        while(True):
            print 'I\'m doing work...'
            time.sleep(3)

覆盖SimpleXMLRPCRequestHandler的do_POST方法调用两次被调用的方法（###部分周围的所有内容都是python代码的标准部分）

class MySimpleXMLRPCRequestHandler(SimpleXMLRPCRequestHandler):

    def do_POST(self):
        """Handles the HTTP POST request.
        Attempts to interpret all HTTP POST requests as XML-RPC calls,
        which are forwarded to the server's _dispatch method for handling.
        """
        # Check that the path is legal
        if not self.is_rpc_path_valid():
            self.report_404()
            return

        try:
            # Get arguments by reading body of request.
            # We read this in chunks to avoid straining
            # socket.read(); around the 10 or 15Mb mark, some platforms
            # begin to have problems (bug #792570).
            max_chunk_size = 10*1024*1024
            size_remaining = int(self.headers["content-length"])
            L = []
            while size_remaining:
                chunk_size = min(size_remaining, max_chunk_size)
                L.append(self.rfile.read(chunk_size))
                size_remaining -= len(L[-1])
            data = ''.join(L)

            # In previous versions of SimpleXMLRPCServer, _dispatch
            # could be overridden in this class, instead of in
            # SimpleXMLRPCDispatcher. To maintain backwards compatibility,
            # check to see if a subclass implements _dispatch and dispatch
            # using that method if present.
            response = self.server._marshaled_dispatch(
                    data, getattr(self, '_dispatch', None)
                )
            ##############################################
            ##############################################
            ## Here you'll have the first part of your response
            response = response.next() # to get the first item of the generator (yes, a little bit uggly)
            ##############################################
            ##############################################
        except Exception, e: # This should only happen if the module is buggy
            # internal error, report as HTTP server error
            self.send_response(500)

            # Send information about the exception if requested
            if hasattr(self.server, '_send_traceback_header') and \
                    self.server._send_traceback_header:
                self.send_header("X-exception", str(e))
                self.send_header("X-traceback", traceback.format_exc())

            self.end_headers()
        else:
            # got a valid XML RPC response
            self.send_response(200)
            self.send_header("Content-type", "text/xml")
            self.send_header("Content-length", str(len(response)))
            self.end_headers()
            self.wfile.write(response)

            # shut down the connection
            self.wfile.flush()
            self.connection.shutdown(1)

            ##############################################
            ##############################################
            ## Here you've send the the first part of your response to the client, relaunch the method
            self.server._marshaled_dispatch(
                    data, getattr(self, '_dispatch', None)
            )
            ##############################################
            ##############################################

然后，主要将成为：

if __name__ == '__main__':
    port = 8080
    server = SimpleXMLRPCServer(("localhost", port), requestHandler=MySimpleXMLRPCRequestHandler)
    print "Listening on port %s..." % port

    w = Worker()

    server.register_function(w.start_work)
    while(1):
        server.handle_request()

XML-RPC在执行所有函数代码之前返回值

2 个答案: