NestJS配置环境变量、读取Yaml配置的保姆级教程

前言

哈喽,大家好,我是 webxue ,也可以叫我 前端小帅 ,今天来分享一个nestjs学习中绕不过的东西,环境变量与配置 ,用过env配置,最终还是选择了 yaml

这里我先声明一下

  • 本地运行命令使用的 nest start --watch
  • 线上部署命令使用的 pm2 start dist/main.js --name nest-demo

编写环境变量

在项目根目录下新建了一个 yaml 的目录,里面存放 yaml 配置文件,我的 yaml 配置文件格式如下:

bash 复制代码
# [环境]-[用户].yaml
dev-zhangsan.yaml # 张三开发时使用的配置文件
dev-lisi.yaml # 李四开发时使用的配置文件
dev-wangwu.yaml # 王五开发时使用的配置文件
pro-120.yaml # 部署在120服务器上时使用的配置文件
pro-130.yaml # 部署在130服务器上时使用的配置文件

这里以 张三开发环境配置 为例,其他配置同理

yaml 复制代码
# dev-zhangsan.yaml
# 以下为示例,随便写一些配置,能读到就行
ENV_NAME: 开发环境-zhangsan
PORT: 3000
MYSQL:
  CONFIG:
    HOST: 127.0.0.1

安装依赖

执行 npm i @nestjs/config cross-env yaml 来安装依赖

  • @nestjs/config: 用来很方便的读取配置文件
  • cross-env: 用来设置当前环境
  • yaml: 用来解析yaml配置

修改启动脚本

json5 复制代码
// package.json
{
    // ...
    "scripts":{
        // ...
        "start:dev-zhangsan": "cross-env NODE_ENV=dev-zhangsan nest start --watch",
        "start:dev-lisi": "cross-env NODE_ENV=dev-lisi nest start --watch",
        "pm2:start:pro-120": "cross-env NODE_ENV=pro-120 pm2 start dist/main.js --name nest-demo"
        // ...
    }
}

当然,如果你觉得我的脚本命令太长了,这个你可以随意,而且我这个也只是个示例,莫要照抄...

接下来就可以在工程中读取到环境变量了

app.module.ts注入配置文件

先来看一下我的目录结构

typescript 复制代码
// app.module.ts
import { Module } from '@nestjs/common';
import { ConfigModule } from '@nestjs/config';
import * as yaml from 'yaml';
import * as path from 'path';
import * as fs from 'fs';

@Module({
  imports: [
    ConfigModule.forRoot({
      load: [
        () => {
          // 获取当前环境标识
          const env = process.env.NODE_ENV || 'development';
          // 获取yaml配置文件路径
          const configPath = path.join(__dirname, `../yaml/${env}.yaml`);
          // 校验yaml路径不存在抛错
          if (!fs.existsSync(configPath)) {
            throw new Error(`Configuration file not found: ${configPath}`);
          }
          // 将yaml配置解析
          return yaml.parse(fs.readFileSync(configPath, 'utf8'));
        },
      ],
      isGlobal: true,
    }),
  ],
})
export class AppModule {}

在main.ts读取环境配置

typescript 复制代码
import { NestFactory } from '@nestjs/core';
import { AppModule } from './app.module';
import { ConfigService } from '@nestjs/config';
import { Logger } from '@nestjs/common';

async function bootstrap() {
  const app = await NestFactory.create(AppModule);

  // 配置文件
  const configService = app.get(ConfigService);
  Logger.verbose(`当前环境:${configService.get('ENV_NAME')}`);
  Logger.verbose(`当前环境端口:${configService.get('PORT')}`);
  Logger.verbose(`Mysql的Host:${configService.get('MYSQL.CONFIG.HOST')}`);

  await app.listen(configService.get('PORT'));
}
bootstrap();

接下来,启动之后,就可以在控制台打印出来环境配置了。

到这里我相信你已经完全会了,如果你还不会在其他ts文件中读取config,在下面评论吧,多的话,我再出一篇文章,我相信应该不会有评论的~~~

结余

每天进步亿点点,主业前端,副业后端,做过自媒体,但没坚持下去,欢迎志同道合的朋友一起吹吹牛逼。

相关推荐
plusone2 小时前
【Nest指北系列-源码】(七)请求生命周期
nestjs
小飞悟3 小时前
浏览器和服务器是怎么“认出你”的?揭秘 Cookie 认证
后端·node.js
关山月3 小时前
什么是 Node.js ABI :它为什么重要
node.js
会飞的鱼先生18 小时前
Node.js-path模块
node.js
企鹅侠客20 小时前
实践篇:14-构建 Node.js 应用程序镜像
docker·node.js·dockerfile
爱分享的程序员1 天前
前端面试专栏-算法篇:18. 查找算法(二分查找、哈希查找)
前端·javascript·node.js
YongGit1 天前
探索 AI + MCP 渲染前端 UI
前端·后端·node.js
小山不高1 天前
nest中如何对typeorm 的repo设置总的center
nestjs
ncj3934379061 天前
vscode中对node项目进行断点调试
vscode·node.js
abigale031 天前
webpack+vite前端构建工具 -11实战中的配置技巧
前端·webpack·node.js