史上最简略的带流控功能的http server

史上最简单的带流控功能的http server

s.py

import time
import SimpleHTTPServer
import SocketServer

BYTES_PER_SECOND=160*1024

class MyHTTPRequestHandler(SimpleHTTPServer.SimpleHTTPRequestHandler):
    def do_GET(self):
        """Serve a GET request."""
        f = self.send_head()
        if f:
            self.copyfileobj(f, self.wfile)
            f.close()
    def copyfileobj(self,fsrc, fdst, length=16*1024):
        """copy data from file-like object fsrc to file-like object fdst"""
        count = 0
        t1 = time.time()
        while 1:
            buf = fsrc.read(length)
            if not buf:
                break
            fdst.write(buf)
            count += len(buf)
            if count >= BYTES_PER_SECOND:
                count = 0
                delay = 1.0 - (time.time() - t1)
                if delay > 0.0:
                    time.sleep(delay)
                t1 = time.time()                
PORT = 8000
Handler = MyHTTPRequestHandler
httpd = SocketServer.TCPServer(("127.0.0.1", PORT), Handler)
print "serving at port", PORT
httpd.serve_forever()

yes,that's it! 总共三十几行代码。BYTES_PER_SECOND用于指定每秒传输的最大字节数。127.0.0.1 是服务器监听的IP。8000为http server监听的端口。

这就是为什么写python程序是一种享受的原因。号称battery include的python，人用人爱的python。

但是python也由此给人一种误解，只能写功能简单的程序。写不了高大上的程序，其实真的是一种很大的误解。

配合上DNS服务器，这个30行的程序可以作为一个不错的升级测试环境。

另外，顺便说一句，如果是在linux下面，端口号如果小于1000的话运行程序是要系统权限的，要不然会抛异常。

史上最简略的带流控功能的http server

相关推荐