Node.js URL 还原示例：解析短链接真实地址

实现原理

短链接服务通常使用 HTTP 301 或 302 状态码配合 Location 响应头进行跳转。还原流程的核心逻辑如下：

1. 接收请求：启动一个 HTTP 服务器，接收用户传入的短链接参数。
2. 发起探测：使用 Node.js http.createClient 或 request对象向短链接地址发送 GET 请求。
3. 状态判断：检查响应状态码。若为 301 或 302，读取 headers.location 作为新的目标 URL，继续下一轮请求；若为非重定向状态码（如 200、404），则当前的 URL 即为还原后的真实地址。
4. 返回结果：将最终得到的原始 URL 返回给客户端，可选支持 JSONP 格式以解决跨域问题。

代码实现

以下代码构建了一个简单的本地服务，监听端口 1235。当访问 /api?u=... 时，会自动执行上述还原逻辑。

var net = require('net'),
http = require('http'),
url = require('url'),
fs = require('fs');

// 定义常见协议的默认端口
var DEFAULT_PORTS = {
'http:': 80,
'https:': 443
};

// 读取静态页面模板（需确保 index.html 存在）
var INDEX_TPL = fs.readFileSync('index.html');

// 统一响应写入函数
function _write(str, res, content_type) {
if(res.jsonp_cb) {
str = res.jsonp_cb + '("' + str + '")';
}
res.writeHead(200, {
'Content-Length': str.length,
'Content-Type': content_type || 'text/plain'
});
res.end(str);
};

// 核心递归展开函数
function expand(short_url, res) {
var info = url.parse(short_url);

// 简单限制：当前示例仅处理 http 协议，https 需更复杂的证书处理
if(info.protocol != 'http:') {
_write(short_url, res);
return;
}

// 创建 HTTP 客户端连接
var client = http.createClient(info.port || DEFAULT_PORTS[info.protocol], info.hostname);
var path = info.pathname || '/';
if(info.search) {
path += info.search;
}

// 构造请求头
var headers = {
host: info.hostname,
'User-Agent': 'NodejsSpider/1.0'
};

var request = client.request('GET', path, headers);
request.end();

// 监听响应事件
request.on('response', function (response) {
// 如果是重定向状态码，递归获取下一个 location
if(response.statusCode == 302 || response.statusCode == 301) {
expand(response.headers.location, res);
} else {
// 最终的非重定向地址即为原 URL
_write(short_url, res);
}
});
};

// 启动 HTTP 服务器
http.createServer(function(req, res){
if(req.url.indexOf('/api?') == 0) {
var params = url.parse(req.url, true);
if(params.query && params.query.u) {
// 支持 JSONP 跨域
if(params.query.cb) {
res.jsonp_cb = params.query.cb;
}
expand(params.query.u, res);
} else {
_write('', res);
}
} else {
// 否则返回默认页面
_write(INDEX_TPL, res, 'text/html');
}
}).listen(1235);

process.on('uncaughtException', function (err) {
console.log('Caught exception: ' + err);
});

运行测试

保存上述代码至 urlexpand.js 文件，并准备一个简单的 index.html 文件在同一目录下。在终端中启动服务：

随后在浏览器或 curl 中访问以下地址进行测试：

• 基本测试：http://127.0.0.1:1235/api?u=http://is.gd/imWyT
• JSONP 测试：http://127.0.0.1:1235/api?u=http://is.gd/imWyT&cb=foo

该服务将根据输入的短链接递归追踪，最终返回真实的长链接地址。

以上就是 Node.js 实现短链接还原的详细代码与步骤。通过理解 HTTP 状态码与 Location 头的关系，你可以轻松扩展此脚本以支持 HTTPS 或其他重定向规则。更多关于前端自动化处理的技巧与资料，请关注本站其它相关文章！