Created
March 29, 2011 18:46
-
-
Save steadicat/892978 to your computer and use it in GitHub Desktop.
A reverse-proxy for CouchDB written in Node.js
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/local/bin/node | |
var sys = require('sys'); | |
var http = require('http'); | |
var url = require('url'); | |
var PREFIX = '/db/'; | |
var TARGET = 'http://example.cloudant.com'; | |
var PORT = 8001; | |
function error(response, error, reason, code) { | |
sys.log('Error '+code+': '+error+' ('+reason+').'); | |
response.writeHead(code, { 'Content-Type': 'application/json' }); | |
response.write(JSON.stringify({ error: error, reason: reason })); | |
response.end(); | |
} | |
function unknownError(response, e) { | |
sys.log(e.stack); | |
error(response, 'unknown', 'Unexpected error.', 500); | |
} | |
function forwardRequest(inRequest, inResponse, uri) { | |
sys.log(inRequest.method + ' ' + uri); | |
uri = url.parse(uri); | |
var out = http.createClient(uri.port||80, uri.hostname); | |
var path = uri.pathname + (uri.search || ''); | |
var headers = inRequest.headers; | |
headers['host'] = uri.hostname + ':' + (uri.port||80); | |
headers['x-forwarded-for'] = inRequest.connection.remoteAddress; | |
headers['referer'] = 'http://' + uri.hostname + ':' + (uri.port||80) + '/'; | |
var outRequest = out.request(inRequest.method, path, headers); | |
out.on('error', function(e) { unknownError(inResponse, e) }); | |
outRequest.on('error', function(e) { unknownError(inResponse, e) }); | |
inRequest.on('data', function(chunk) { outRequest.write(chunk) }); | |
inRequest.on('end', function() { | |
outRequest.on('response', function(outResponse) { | |
// nginx does not support chunked transfers for proxied requests | |
delete outResponse.headers['transfer-encoding']; | |
if (outResponse.statusCode == 503) { | |
return error(inResponse, 'db_unavailable', 'Database server not available.', 502); | |
} | |
inResponse.writeHead(outResponse.statusCode, outResponse.headers); | |
outResponse.on('data', function(chunk) { inResponse.write(chunk); }); | |
outResponse.on('end', function() { inResponse.end(); }); | |
}); | |
outRequest.end(); | |
}); | |
}; | |
function handleRequest(request, response) { | |
var u = url.parse(request.url); | |
// Only serve URLs that start with PREFIX | |
if (u.pathname.substring(0, PREFIX.length) != PREFIX) return error(response, 'not_found', 'Nothing found here.', 404); | |
u = TARGET + u.pathname.substring(PREFIX.length-1) + (u.search||''); | |
forwardRequest(request, response, u); | |
} | |
process.on('uncaughtException', function(e) { | |
sys.log(e.stack); | |
}); | |
http.createServer(handleRequest).listen(PORT); | |
sys.puts('Proxy ready on port '+PORT+'.') |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
An example using http-proxy as suggested by @lancecarlson.
Requests to
http://localhost:3000
are proxied to CouchDB running onhttp://localhost:5984
.