109 lines
2.9 KiB
TypeScript
109 lines
2.9 KiB
TypeScript
// nacos.service.ts
|
||
import {
|
||
Injectable,
|
||
OnApplicationBootstrap,
|
||
OnApplicationShutdown,
|
||
} from '@nestjs/common';
|
||
import { ConfigService } from '@nestjs/config';
|
||
import { NacosConfigClient, NacosNamingClient } from 'nacos'; // ts
|
||
import * as os from 'os';
|
||
|
||
@Injectable()
|
||
export class NacosService
|
||
implements OnApplicationBootstrap, OnApplicationShutdown
|
||
{
|
||
private nacosNamingClient: NacosNamingClient;
|
||
private nacosConfigClient: NacosConfigClient;
|
||
serviceName: string;
|
||
instance: { ip: string; port: number };
|
||
group: string;
|
||
dataId: string;
|
||
constructor(private configService: ConfigService) {
|
||
this.nacosNamingClient = new NacosNamingClient({
|
||
logger: console,
|
||
serverList: configService.get('NACOS_ADDR'),
|
||
namespace: configService.get('NACOS_NAMESPACE'),
|
||
});
|
||
this.nacosConfigClient = new NacosConfigClient({
|
||
namespace: configService.get('NACOS_NAMESPACE'),
|
||
serverAddr: configService.get('NACOS_ADDR'),
|
||
});
|
||
this.serviceName = configService.get('NACOS_SERVICE_NAME');
|
||
this.dataId = configService.get('NACOS_DATAID');
|
||
this.group = configService.get('NACOS_GROUP');
|
||
this.instance = {
|
||
ip: this.getServerIP(),
|
||
port: configService.get('PORT'),
|
||
};
|
||
}
|
||
|
||
/**
|
||
* nestjs应用被关闭前
|
||
* @param {string} signal 'SIGTERM' | 'SIGINT' | 'SIGHUP' | 'SIGBREAK'
|
||
*/
|
||
onApplicationShutdown(signal?: string) {
|
||
if (signal) {
|
||
const { serviceName, instance, group } = this;
|
||
this.nacosNamingClient.deregisterInstance(serviceName, instance, group);
|
||
this.nacosConfigClient.close();
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 应用完全启动&微服务也被成功启动
|
||
*/
|
||
onApplicationBootstrap() {
|
||
const { serviceName, instance } = this;
|
||
this.nacosNamingClient.registerInstance(serviceName, instance);
|
||
}
|
||
|
||
/**
|
||
* 先于 onApplicationBootstrap
|
||
*/
|
||
async onModuleInit() {
|
||
this.nacosNamingClient.ready();
|
||
}
|
||
|
||
/**
|
||
* 从nacos获取最新的配置信息
|
||
*/
|
||
async getConfig() {
|
||
const { dataId, group } = this;
|
||
const configFromNacos = await this.nacosConfigClient.getConfig(
|
||
dataId,
|
||
group,
|
||
);
|
||
return configFromNacos;
|
||
}
|
||
|
||
/**
|
||
* 订阅配置中心，当远程修改nacos的配置时，触发
|
||
*/
|
||
async subscribeConfiguration() {
|
||
const { dataId, group } = this;
|
||
this.nacosConfigClient.subscribe(
|
||
{
|
||
dataId,
|
||
group,
|
||
},
|
||
(content) => console.log('content', content),
|
||
);
|
||
}
|
||
|
||
getServerIP(): string {
|
||
const networkInterfaces = os.networkInterfaces();
|
||
for (const name of Object.keys(networkInterfaces)) {
|
||
for (const iface of networkInterfaces[name]) {
|
||
// 跳过IPv6和内部地址
|
||
if ('IPv4' !== iface.family || iface.internal !== false) {
|
||
continue;
|
||
}
|
||
|
||
// 返回第一个找到的IPv4地址
|
||
return iface.address;
|
||
}
|
||
}
|
||
return 'localhost'; // 如果找不到外部IPv4地址，返回localhost
|
||
}
|
||
}
|