在老的Node.js服务器里“加点Rust”,我的服务性能飙升近 80%

liftword2个月前 (01-07)技术文章19

你有没有遇到过这样的情况?服务器跑着跑着就卡了,明明只是一些普通的操作,却让资源“飚红”,甚至快撑不住了。特别是当你用JavaScript或者Python这些脚本语言写的服务器,遇到CPU密集型任务时,性能瓶颈似乎更是无可避免。这时候,是不是觉得有点力不从心?

今天,我们安利一个解决方案——Rust!一种速度快、效率高的编程语言。它有点像是给你的Node.js或者Python服务器加了“肌肉”,尤其适合处理高强度的运算任务。下面,我就给大家讲讲如何一步步把Rust“融入”到现有的服务器里,用简单的策略大幅度提升性能。


引入Rust的三步策略

在这个策略中,我们从“0”开始,逐步引入Rust,分别通过Rust CLI工具和Wasm模块来提升服务器的性能。总的原则是:每一步都不搞大改动,让你的老服务器既能“焕发新生”,又能保持现有的代码框架。


第0步:从Node.js服务器开始

假设我们现在有一个Node.js服务器,用来生成二维码。这个需求其实并不复杂,但在高并发的情况下,这样的CPU密集型任务会让JavaScript显得吃力。

const express = require('express');
const generateQrCode = require('./generate-qr.js');

const app = express();
app.get('/qrcode', async (req, res) => {
    const { text } = req.query;

    if (!text) {
        return res.status(400).send('missing "text" query param');
    }

    if (text.length > 512) {
        return res.status(400).send('text must be <= 512 bytes');
    }

    try {
        const qrCode = await generateQrCode(text);
        res.setHeader('Content-Type', 'image/png');
        res.send(qrCode);
    } catch (err) {
        res.status(500).send('failed generating QR code');
    }
});

app.listen(42069, '127.0.0.1');

基准测试:在纯Node.js的情况下,这个服务每秒能处理1464个请求,内存占用也不小。虽然勉强能跑起来,但一旦用户多了,可能会明显感觉到卡顿。


第1步:引入Rust CLI工具,效率提升近80%

这里的策略是保留Node.js的框架不变,把处理二维码生成的那段代码用Rust写成一个独立的命令行工具(CLI)。在Node.js中,我们直接调用这个CLI工具,分担高强度的计算工作。

/** qr_lib/lib.rs **/

use qrcode::{QrCode, EcLevel};
use image::Luma;
use image::codecs::png::{CompressionType, FilterType, PngEncoder};

pub type StdErr = Box<dyn std::error::Error>;

pub fn generate_qr_code(text: &str) -> Result<Vec<u8>, StdErr> {
    let qr = QrCode::with_error_correction_level(text, EcLevel::L)?;
    let img_buf = qr.render::<Luma<u8>>()
        .min_dimensions(200, 200)
        .build();
    let mut encoded_buf = Vec::with_capacity(512);
    let encoder = PngEncoder::new_with_quality(
        &mut encoded_buf,
        // these options were chosen since
        // they offered the best balance
        // between speed and compression
        // during testing
        CompressionType::Default,
        FilterType::NoFilter,
    );
    img_buf.write_with_encoder(encoder)?;
    Ok(encoded_buf)
}

效果:重写后,我们的处理性能直接飙升到了每秒2572个请求!这是一个显著的提升,更让人欣慰的是,内存占用也跟着降了下来。Rust的高效编译和内存管理,确实比JavaScript强太多了。

实现步骤

  1. 首先,用Rust编写二维码生成的核心逻辑代码。
  2. 将这段Rust代码编译成一个可执行的CLI工具。
  3. 在Node.js代码中,通过子进程调用CLI工具,直接拿到生成的结果。

在Node.js中调用Rust CLI工具的代码示例如下:

const { exec } = require('child_process');
exec('./qr_generator_cli', (error, stdout, stderr) => {
  if (error) {
    console.error(`执行出错: ${error}`);
    return;
  }
  console.log(`生成的二维码数据: ${stdout}`);
});

这个方法就像是给Node.js加了一个“外挂”,而且几乎不需要改动现有代码。也就是说,你可以在不动大框架的情况下,得到Rust的性能优势。


第2步:编译Rust到WebAssembly(Wasm),性能提升再进一步

在第1步中,我们通过CLI工具调用了Rust,但依旧会产生一定的通信开销。所以,接下来,我们可以进一步优化,将Rust代码编译成WebAssembly(Wasm)模块,并在Node.js中直接调用它。这样,整个过程就在内存中运行,不用通过子进程调用CLI,速度进一步提升。

效果:使用Wasm后,处理性能再上升到了每秒2978个请求,而内存使用依旧维持在较低水平。

实现步骤

  1. 将Rust代码编译为Wasm模块。可以使用wasm-pack这样的工具来帮助生成。
  2. 在Node.js中,通过wasm-bindgen等工具直接加载并调用Wasm模块。

Node.js中加载Wasm模块的代码示例如下:

const fs = require('fs');
const wasmBuffer = fs.readFileSync('./qr_generator_bg.wasm');
WebAssembly.instantiate(wasmBuffer).then(wasmModule => {
  const qrGenerator = wasmModule.instance.exports.qr_generate;
  console.log(qrGenerator('Hello, Rust with Wasm!'));
});

这种方法让我们完全绕过了CLI的通信环节,直接把Rust的性能用在Node.js中。这不仅提升了效率,还让代码更加紧凑,减少了延迟。


思考

通过以上三步策略,我们可以在不完全推翻现有代码的前提下,逐步引入Rust,极大地提升服务器的性能。这个过程既适用于Node.js,也可以推广到其他语言和环境中。

为什么这个方法特别值得尝试呢?首先,它成本低。你不需要重写整个系统,只需要对瓶颈部分进行改进。其次,效果明显,尤其是对那些经常“吃力”的功能。最后,这个方法是可扩展的,你可以根据实际情况,灵活选择用CLI还是Wasm的方式来引入Rust。

所以,如果你的服务器正被性能问题困扰,不妨试试这个三步引Rust法。正如一位资深开发者所说:“Rust不仅让你的服务器跑得更快,还让代码变得更加优雅。”

学习交流

最后,把我的座右铭送给你:投资自己才是最大的财富。 如果你觉得本文章对你有帮助,点赞,收藏不迷路

公众号:我不是架构师,持续为你输出更多的硬核文章和面试经验。

相关文章

用Python进行短视频二次创作,网红必备

无论是抖音还是快手等视频平台,一旦一个视频火了后,很多 UP 主都会争先抢后去模仿拍摄或剪辑,然后上传到平台,最后都能带来不错的流量。对于一般的短视频,完全可以通过裁剪、特效转场、加入混合图层和字幕等...

老男孩教育2016年第一期免费系列课程

新的一年新的分享老男孩教育2016年首次在腾讯课程发布免费系统课程,每周三晚8点30至晚10点,不见不散!1.老师阵容:老男孩:老男孩教育的创始人,有实力不解释,你懂得~Alex:帅的不要不要的金角大...

两个日本人已经支持华语音乐20年了!原来我们从小就听日语歌长

大家伙早上好,不知道小编今天要说的这位明星大家有没有印象呢?而华语音乐的开始相对较晚,所以很多著名的国王或天后,他们的许多原创名著都是外文封面歌曲。即使到目前为止,也有很多歌曲来自外语歌曲,比如筷子哥...

如果编程语言是种武器,你使用的武器是什么?

对程序员来说,编程语言就是武器。你使用的是哪种武器,杀伤力又是如何呢?不如听老男孩老师来分析分析~ C语言是M1式加兰德步枪,很老但可靠。C++是双截棍,挥舞起来很强悍,很吸引人,但需要你多年的磨练来...

搞了半天,终于弄懂了TCP Socket数据的接收和发送,太难

本文将从上层介绍Linux上的TCP/IP栈是如何工作的,特别是socket系统调用和内核数据结构的交互、内核和实际网络的交互。写这篇文章的部分原因是解释监听队列溢出(listen queue ove...