如果一个项目启动时(单机), 瞬间来了1000个访问, 如何确保db等资源不会压垮呢?

现在想想我当时回答的并不好, 而现在看公司框架才发现其实有针对于这一块做过专门的优化的。
下面就来分享下公司关于这个地方的处理, 一句话总结就是:项目启动时会先热身一段时间,概率性拒绝请求以保证服务的高可用。

下面直接说原理:
1, 根据自己的业务需求设置一个热身时间:warmupTime
2, 在服务启动接收请求的时候添加一个拦截器,如果项目还没有热身完就概率性停止对外服务。(这里使用random去计算时候返回正常的响应)
3, 等热身完毕的时候提供完整的响应。

使用场景:
特别适用于单机服务(双机因为有负载均衡所以不必考虑这个问题),且并发较高的服务。使用热身的主要原因是:当系统初始化时缓慢增加请求,防止系统开始压力过大导致db等资源出错。

下面直接贴代码, 用代码来讲解:

@Override
protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
    processRequest(req, resp);
}

@Override
protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
    processRequest(req, resp);
}

protected void processRequest(HttpServletRequest requestOld, HttpServletResponse response) throws ServletException, IOException {
    // 当系统初始化时缓慢增加请求,防止系统开始压力过大导致db等资源出错
if (!bootHandler.isOk() && bootHandler.handle(requestOld, response,               config)) {
        return;
    }
    
    //...
}

在接收到请求时会先去判断是否热身完毕, bootHandler.handle就是判断热身的方法,具体判断方式如下代码:

public boolean handle(HttpServletRequest request, HttpServletResponse response, MvcConfig config) throws IOException {
    long warmupTime = config.getWarmupTime();
    String startingUpJson = config.getStartingUpJson();
    if (warmupTime <= 0) {
        ok = true;
    } else {
        long now = System.currentTimeMillis();
        if (startTime == -1 || startTime == 0) {
            if (startTime == -1) {
                startTime = now;
                LOG.info("收到第一个请求,开始进行热身");
            }
            RenderUtils.renderJson(request, response, startingUpJson);
            return true;
        } else {
            //从开始时间到现在,以及配置的热身时间计算当前服务的比例
            if (now >= startTime + warmupTime) {
                ok = true;
            } else {
                //比率是从 [0-100)
                int rate = (int) ((now - startTime) * 100 / warmupTime);
                boolean rejected = random.nextInt(100) > rate;
                if (rejected) {
                    if (now - lastLogTime.get() > 1000L) {
                        lastLogTime.set(now);
                        LOG.warn("当前请求被拒绝,当前响应概率为:{}%", rate);
                    }
                    renderJson(request, response, startingUpJson);
                    return true;
                }
            }
        }
    }
    return false;
}

首先回去判断是否配置了热身时间, 如果没有配置那么直接返回。
接着就去用当前的时间和项目启动时间以及热身时间计算出比率, 然后通过random去判断当前请求是否返回响应。直到热身完, 比率会达到100%。

下面贴一张项目启动时的Log:

虽然这是一种很简单的方式处理项目启动时资源不可用的解决方法, 但是却对项目启动时带来很大的帮助, 希望这种方法能够对大家有用。