统一身份认证与试用功能的实现：一个程序员的实践分享

大家好，我是小明，一个普通的程序员。今天我想跟大家聊聊“统一身份认证”和“试用”这两个东西。听起来是不是有点高大上？其实啊，它们就是我们平时开发中经常用到的功能模块。如果你是做后端或者前端开发的，那你肯定对这两个概念不陌生。

先说说什么是“统一身份认证”。简单来说，它就是一种让不同系统或服务都能使用同一个账号登录的方式。比如你有一个电商网站，还有一个客服系统，如果这两个系统都用统一的身份认证，那用户就不用重复注册了，省事多了。

而“试用”呢，就是让用户在正式使用之前，可以先体验一下系统，看看有没有兴趣继续使用。比如你开发了一个新的SaaS产品，用户可以先免费试用几天，然后决定是否购买。

那问题来了，怎么把这两者结合起来呢？比如说，用户在试用的时候，能不能也用统一的身份认证？答案是肯定的。接下来我就用具体的代码来演示一下。

一、项目结构介绍

首先，我这里用的是Node.js + Express + MongoDB的组合。数据库用的是MongoDB，用来存储用户信息和试用状态。前端的话，你可以用任何你喜欢的框架，比如React或者Vue，不过这部分暂时不讲，重点还是后端逻辑。

二、统一身份认证的实现

统一身份认证的核心是“用户登录”，所以我们要先设计一个用户表。这个表里应该包含用户名、密码、邮箱、创建时间等字段。当然，为了安全，密码不能直接存，要用加密算法处理。

下面是一个简单的用户模型定义（用Mongoose）：

// models/User.js
const mongoose = require('mongoose');

const UserSchema = new mongoose.Schema({
  username: { type: String, unique: true, required: true },
  password: { type: String, required: true },
  email: { type: String, unique: true, required: true },
  createdAt: { type: Date, default: Date.now }
});

module.exports = mongoose.model('User', UserSchema);
    

接下来是登录接口。用户输入用户名和密码，后端验证是否正确，如果正确，就返回一个token，用于后续请求的鉴权。

这里用的是JWT（JSON Web Token），因为它简单又安全。下面是登录接口的代码：

// routes/auth.js
const express = require('express');
const router = express.Router();
const User = require('../models/User');
const jwt = require('jsonwebtoken');

router.post('/login', async (req, res) => {
  const { username, password } = req.body;

  try {
    const user = await User.findOne({ username });
    if (!user) return res.status(401).json({ message: '用户名或密码错误' });

    // 这里假设密码是明文存储的，实际应使用bcrypt加密
    if (user.password !== password) return res.status(401).json({ message: '用户名或密码错误' });

    const token = jwt.sign({ userId: user._id }, 'your-secret-key', { expiresIn: '1h' });
    res.json({ token });
  } catch (err) {
    res.status(500).json({ message: '服务器错误' });
  }
});

module.exports = router;
    

这样，用户就可以通过登录获取token，之后每次请求都带上这个token，后端就能知道是谁在访问了。

三、试用功能的实现

现在我们有了统一的身份认证，接下来要实现的是试用功能。也就是说，用户可以先免费试用一段时间，之后再付费。

首先，我们需要在用户表中添加一个字段，表示是否处于试用期，以及试用结束的时间。

修改一下之前的User模型：

// models/User.js
const UserSchema = new mongoose.Schema({
  username: { type: String, unique: true, required: true },
  password: { type: String, required: true },
  email: { type: String, unique: true, required: true },
  isTrial: { type: Boolean, default: true },
  trialExpiresAt: { type: Date, default: Date.now() + 7 * 24 * 60 * 60 * 1000 }, // 默认试用7天
  createdAt: { type: Date, default: Date.now }
});
    

这样，当用户第一次注册时，默认就是试用状态，试用期为7天。

接下来，我们可以在一些需要权限的接口中检查用户的试用状态。例如，某个高级功能只能在试用期结束后才能使用。

下面是一个中间件的例子，用来检查用户是否还在试用期内：

// middleware/checkTrial.js
const jwt = require('jsonwebtoken');
const User = require('../models/User');

function checkTrial(req, res, next) {
  const token = req.headers['authorization'];
  if (!token) return res.status(401).json({ message: '未提供token' });

  try {
    const decoded = jwt.verify(token, 'your-secret-key');
    User.findById(decoded.userId).then(user => {
      if (!user.isTrial || user.trialExpiresAt < Date.now()) {
        return res.status(403).json({ message: '您已超出试用期，请升级套餐' });
      }
      next();
    }).catch(err => {
      res.status(500).json({ message: '服务器错误' });
    });
  } catch (err) {
    res.status(401).json({ message: '无效token' });
  }
}

module.exports = checkTrial;
    

然后，在需要限制的路由中加入这个中间件：

// routes/feature.js
const express = require('express');
const router = express.Router();
const checkTrial = require('../middleware/checkTrial');

router.get('/premium-feature', checkTrial, (req, res) => {
  res.json({ message: '这是高级功能，只有试用期内的用户才能看到' });
});

module.exports = router;
    

这样，用户在试用期内可以访问高级功能，一旦过了试用期，就不能再用了。

四、试用期到期后的处理

试用期到期后，用户可能有两种选择：要么付费续费，要么停止使用。这时候，我们可以提供一个页面，引导用户进行付费操作。

不过，这涉及到支付系统的集成，比如支付宝、微信支付或者Stripe。这部分内容比较复杂，超出了今天的范围，但你可以根据自己的业务需求来扩展。

五、测试一下

现在我们已经实现了统一身份认证和试用功能，那么怎么测试呢？可以用Postman或者curl发送请求。

首先注册一个用户：

POST /api/users
{
  "username": "testuser",
  "password": "123456",
  "email": "test@example.com"
}
    

然后登录：

POST /api/auth/login
{
  "username": "testuser",
  "password": "123456"
}
    

登录成功后会得到一个token，接着用这个token访问受限制的接口：

GET /api/feature/premium-feature
Authorization: Bearer YOUR_TOKEN
    

如果在试用期内，就会返回“这是高级功能...”；如果过了试用期，就会提示“您已超出试用期，请升级套餐”。

六、总结

通过上面的代码示例，我们可以看到，统一身份认证和试用功能其实并不难实现。关键是要理解它们的原理，并合理地设计数据结构和接口。

统一身份认证保证了用户在多个系统中的一致性，而试用功能则提升了用户体验，降低了用户进入门槛。

当然，这只是最基础的实现方式。实际项目中，你还可能会遇到更多复杂的情况，比如多租户、第三方登录、自动续费等等。这些都需要进一步的设计和开发。

如果你正在开发一个SaaS产品，或者想给现有系统增加试用功能，不妨试试这套方案。希望这篇文章能对你有所帮助！