iframe-proxy | Sunbelt Computer Software

93 lines (78 loc) · 2.47 KB

#!/usr/bin/python
import traceback, json, sys, socket, os
import rethinkdb
import tornado.ioloop
import tornado.web
import tornado.httpserver
import tornado.netutil
from subprocess import check_output
HOST_PATH = '/host'
SOCK_PATH = '%s/ol.sock' % HOST_PATH
STDOUT_PATH = '%s/stdout' % HOST_PATH
STDERR_PATH = '%s/stderr' % HOST_PATH
PROCESSES_DEFAULT = 10
initialized = False
config = None
db_conn = None
# run once per process
def init():
    global initialized, config, db_conn, lambda_func
    if initialized:
        return
    sys.stdout = open(STDOUT_PATH, 'w')
    sys.stderr = open(STDERR_PATH, 'w')
    if len(sys.argv) == 1:
        config = json.loads(os.environ['ol.config'])
        if config.get('db', None) == 'rethinkdb':
            host = config.get('rethinkdb.host', 'localhost')
            port = config.get('rethinkdb.port', 28015)
            print 'Connect to %s:%d' % (host, port)
            db_conn = rethinkdb.connect(host, port)
    sys.path.append('/handler')
    import lambda_func # assume submitted .py file is /handler/lambda_func.py
    initialized = True
class SockFileHandler(tornado.web.RequestHandler):
    def post(self):
        try:
            init()
            data = self.request.body
            try :
                event = json.loads(data)
            except:
                self.set_status(400)
                self.write('bad POST data: "%s"'%str(data))
                return
            self.write(json.dumps(lambda_func.handler(db_conn, event)))
        except Exception:
            self.set_status(500) # internal error
            self.write(traceback.format_exc())
tornado_app = tornado.web.Application([
    (r".*", SockFileHandler),
# listen on sock file with Tornado
def lambda_server():
    server = tornado.httpserver.HTTPServer(tornado_app)
    socket = tornado.netutil.bind_unix_socket(SOCK_PATH)
    server.add_socket(socket)
    tornado.ioloop.IOLoop.instance().start()
    server.start(PROCESSES_DEFAULT)
# listen for fds to forkenter
def fdlisten(path):
    import ns
    r = ns.fdlisten(path)
    # parent
    if r > 0:
        print('Parent should never escape from fdlisten')
        sys.exit(1)
    # child
    if r == 0:
        lambda_server()
if __name__ == '__main__':
    if len(sys.argv) == 1:
        lambda_server()
    elif len(sys.argv) == 2:
        fdlisten(sys.argv[1])
        print('Usage (nofork): python %s' % sys.argv[0])
        print('Usage (fork): python %s <fifo>' % sys.argv[0])
        sys.exit(1)

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

server.py

Latest commit

History

server.py

File metadata and controls