node/card_info_svr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

只有生产环境才使用服务自发现功能

Browse Source
This commit is contained in:
zhl 2021-01-30 12:53:07 +08:00
parent b944d26044
commit 8bf7232da2
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/utils/system.util.ts
View File

@ -5,6 +5,7 @@ const signals: NodeJS.Signals[] = ['SIGINT', 'SIGTERM', 'SIGUSR2']
const NODES_SET = 'poker:infosvr' const NODES_SET = 'poker:infosvr'
const discovery_channel = 'poker:infosvr:discovery' const discovery_channel = 'poker:infosvr:discovery'
const isProd = process.env.NODE_ENV === 'production'
export function registerGracefulShutdown(callback: (err?: Error) => void) { export function registerGracefulShutdown(callback: (err?: Error) => void) {
/** /**
@ -26,6 +27,9 @@ async function getNodeAddress(port: number) {
} }
export async function registService(port: number) { export async function registService(port: number) {
if (!isProd) {
return
}
const address = await getNodeAddress(port) const address = await getNodeAddress(port)
const client = new RedisClient() const client = new RedisClient()
await client.sadd(NODES_SET, address) await client.sadd(NODES_SET, address)
@ -33,6 +37,9 @@ export async function registService(port: number) {
} }
export async function unRegistService(port: number) { export async function unRegistService(port: number) {
if (!isProd) {
return
}
const address = await getNodeAddress(port) const address = await getNodeAddress(port)
const client = new RedisClient() const client = new RedisClient()
await client.srem(NODES_SET, address) await client.srem(NODES_SET, address)