demo-db-benchmark's Introduction

demo-db-benchmark

数据库性能测试：

插入 500 万条数据
分组查询

10万分组

针对 10w 条数据

根据 category 分组
对 age 进行 sum 聚合
根据 rank 排序

import mongoose, { Schema, model } from "mongoose";

interface IUser {
  name: string;
  age: number;
  rank: number;
  category: string;
}

const userSchema = new Schema<IUser>({
  name: { type: String, required: true },
  age: { type: Number, required: true },
  rank: { type: Number, required: true },
  category: { type: String, required: true },
});

const UserModel = model<IUser>("TUser", userSchema);

async function query() {
  const start = Date.now();
  UserModel.aggregate(
    [
      {
        $group: {
          _id: "$name",
          count: { $sum: 1 },
          totalAge: { $sum: '$age' }
        }
      },
      {
        $sort: {
          totalAge: -1
        }
      },
      {
        $skip: 10000,
      },
      {
        $limit: 10
      },
    ],
    (_: any, data: any) => {
      # 345 稳定
      console.log(Date.now()  - start);
      console.log(data);
    }
  );
}

mongoose.connect("mongodb://47.112.180.188:27017").then(() => {
  query();
});

500 万分组

并发数 1 CPU 飙升到 50%
并发数 2 CPU 飙升到 100%

import mongoose, { Schema, model } from "mongoose";

interface IUser {
  name: string;
  age: number;
  rank: number;
  category: string;
}

const userSchema = new Schema<IUser>({
  name: { type: String, required: true },
  age: { type: Number, required: true },
  rank: { type: Number, required: true },
  category: { type: String, required: true },
});

const UserModel = model<IUser>("TUser", userSchema);

async function query() {
  const start = Date.now();
  UserModel.aggregate(
    [
      {
        $group: {
          _id: "$name",
          count: { $sum: 1 },
          totalAge: { $sum: '$age' }
        }
      },
      {
        $sort: {
          totalAge: -1
        }
      },
      {
        $skip: 0,
      },
      {
        $limit: 10
      },
    ],
    (_: any, data: any) => {
      # 8930
      console.log(Date.now()  - start);
      console.log(data);
    }
  );
}

mongoose.connect("mongodb://47.112.180.188:27017").then(() => {
  query();
});

Recommend Projects