nodejs实用示例 缩址还原

请看以下攻略：

nodejs实用示例：缩址还原

在网站或者移动应用开发中，我们经常需要将长网址转化为短网址，以提高用户体验。本文将介绍如何使用 Node.js 实现一个简单的缩址功能。

要点

• 缩址算法：将长网址通过散列算法转化为短网址。
• 数据库保存：使用 MongoDB 数据库保存长网址和短网址的映射关系。
• 路由设置：将短链接重定向到长链接，需要根据短链接从数据库中查询对应的长链接。

缩址算法

现在有很多缩址算法，比如 Base62、ShortID、NanoID 等等。在本实例中，我们使用 shortid 模块来生成短网址。它是一个简单易用的模块，允许我们设置短码的长度、字符集等参数。

安装 shortid 模块：

npm install shortid

生成短码的示例代码如下：

const shortid = require('shortid');

function generateShortUrl(url) {
  const id = shortid.generate();
  const shorturl = `http://example.com/${id}`;
  // 将长网址和短网址存入数据库
  return shorturl;
}

数据库保存

我们使用 MongoDB 作为数据库，使用 mongoose 模块来访问数据库。在本实例中，我们需要创建一个名为 url 的集合来存储长网址和短网址的映射关系。

安装 mongoose 模块：

npm install mongoose

创建 mongoose 模型：

const mongoose = require('mongoose');

const UrlSchema = new mongoose.Schema({
  longurl: { type: String, required: true },
  shorturl: { type: String, required: true },
  created_at: { type: Date, default: Date.now }
});

const UrlModel = mongoose.model('Url', UrlSchema);

module.exports = UrlModel;

将长网址和短网址保存到数据库的示例代码如下：

const mongoose = require('mongoose');
const UrlModel = require('./models/url');

mongoose.connect('mongodb://localhost:27017/url_shortener');

function saveToDatabase(longurl, shorturl) {
  const url = new UrlModel({
    longurl: longurl,
    shorturl: shorturl
  });
  url.save(function(err) {
    if (err) {
      console.error(`Failed to save ${longurl}, ${shorturl}: ${err}`);
    } else {
      console.log(`Saved ${longurl}, ${shorturl} to database.`);
    }
  });
}

路由设置

我们需要使用 Express 模块来设置短链接的路由，将短链接重定向到长链接。在路由处理函数中，根据短链接从数据库中查找对应的长链接，并使用 res.redirect() 方法重定向到长链接。

安装 Express 模块：

npm install express

设置路由的示例代码如下：

const express = require('express');
const mongoose = require('mongoose');
const UrlModel = require('./models/url');

const app = express();

mongoose.connect('mongodb://localhost:27017/url_shortener');
const db = mongoose.connection;
db.on('error', console.error.bind(console, 'connection error:'));
db.once('open', function() {
  console.log('Connected to database.');
});

// 短链接路由
app.get('/:id', function(req, res) {
  const id = req.params.id;
  UrlModel.findOne({ shorturl: `http://example.com/${id}` }, function(err, url) {
    if (err || !url) {
      console.error(`Failed to find longurl for ${id}: ${err}`);
      res.status(404).send('Not found.');
    } else {
      console.log(`Redirecting ${id} to ${url.longurl}`);
      res.redirect(url.longurl);
    }
  });
});

app.listen(3000, function() {
  console.log('URL shortener listening on port 3000.');
});

示例说明

以下是两个示例说明：

示例一：生成短链接

const shortUrlService = require('./short-url-service');

// 假设长链接为：https://www.google.com/search?q=abc
const longurl = 'https://www.google.com/search?q=abc';
const shorturl = shortUrlService.generateShortUrl(longurl);
console.log(`Short URL for ${longurl}: ${shorturl}`);

// 输出：Short URL for https://www.google.com/search?q=abc: http://example.com/-B1zj-WpW

示例二：使用短链接

访问 http://example.com/-B1zj-WpW，会自动重定向到 https://www.google.com/search?q=abc。

总结

本实例介绍了使用 Node.js、shortid、mongoose、Express 等模块实现一个简单的缩址功能。我们使用 shortid 模块生成短码，使用 mongoose 模块访问 MongoDB 数据库，使用 Express 模块设置路由并重定向短链接。