feat(@142vip/grpc): 支持单例模式，grpc客户端、服务端通用方法

Author: mmdapl

packages/grpc/CHANGELOG.md

@@ -0,0 +1,11 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+See [Conventional Commits](https://conventionalcommits.org) for commit guidelines.
+
+## v0.0.1-alpha.0 (2024-07-14)
+
+### ✨ Features
+
+- 新增`tsconfig`配置，统一模块编译 by . @chufan
+- 基本结构初始化  by . @chufan

packages/grpc/README.md

@@ -0,0 +1,9 @@
+# @142vip/grpc
+
+[![NPM version](https://img.shields.io/npm/v/@142vip/grpc?labelColor=0b3d52&color=1da469&label=version)](https://www.npmjs.com/package/@142vip/grpc)
+
+## 证书
+
+[MIT](https://opensource.org/license/MIT)
+
+Copyright (c) 2019-present, 142vip 储凡

packages/grpc/build.config.ts

@@ -0,0 +1,13 @@
+import { defineBuildConfig } from 'unbuild'
+
+export default defineBuildConfig({
+  entries: [
+    'src/index',
+  ],
+  declaration: true,
+  clean: true,
+  rollup: {
+    emitCJS: true,
+    inlineDependencies: true,
+  },
+})

packages/grpc/index.d.ts

@@ -0,0 +1 @@
+export * from './src'

packages/grpc/package.json

@@ -0,0 +1,47 @@
+{
+  "name": "@142vip/grpc",
+  "version": "0.0.1-alpha.0",
+  "private": false,
+  "description": "公众号搜：储凡",
+  "author": "mmdapl <mmdapl@163.com>",
+  "license": "MIT",
+  "homepage": "https://142vip.github.io/core-x/packages/oauth/",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/142vip/core-x.git",
+    "directory": "packages/grpc"
+  },
+  "keywords": [
+    "公众号搜：储凡",
+    "142vip",
+    "@142vip",
+    "@142vip/grpc",
+    "grpc"
+  ],
+  "sideEffects": false,
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.mjs",
+      "require": "./dist/index.cjs"
+    }
+  },
+  "main": "./dist/index.mjs",
+  "module": "./dist/index.mjs",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build": "unbuild",
+    "typecheck": "tsc --noEmit"
+  },
+  "dependencies": {
+    "@grpc/grpc-js": "^1.12.5",
+    "@grpc/proto-loader": "^0.7.13"
+  },
+  "publishConfig": {
+    "access": "public",
+    "registry": "https://registry.npmjs.org"
+  }
+}

packages/grpc/src/grpc-client.ts

@@ -0,0 +1,52 @@
+import grpc from '@grpc/grpc-js'
+import type { ServiceClient } from '@grpc/grpc-js/build/src/make-client'
+import { Singleton } from './singleton'
+import type { ServiceClaDefinition } from './grpc.interface'
+
+type GrpcServices = Map<string, ServiceClient>
+
+export class GrpcClient extends Singleton<GrpcClient> {
+  private readonly grpcServices: GrpcServices
+  constructor() {
+    super()
+    this.grpcServices = new Map<string, ServiceClient>()
+  }
+
+  /**
+   * 建立连接
+   */
+  public connect(connectUri: string, serviceClaDefinitions: ServiceClaDefinition[]): void {
+    // 创建服务实例
+    for (const { packageName, ServiceClass, serviceName } of serviceClaDefinitions) {
+      console.log(`包：${packageName} -- 类：${serviceName}`)
+      this.grpcServices.set(serviceName, new ServiceClass(connectUri, grpc.credentials.createInsecure()))
+    }
+  }
+
+  /**
+   * 获取连接Service
+   */
+  public getService<T>(serviceName: string): T {
+    if (!this.grpcServices.has(serviceName)) {
+      console.log('服务不存在')
+      throw new Error('服务不存在')
+    }
+    return this.grpcServices.get(serviceName) as T
+  }
+
+  /**
+   * 获取连接数
+   */
+  public getConnectSize(): number {
+    return this.grpcServices.size
+  }
+
+  /**
+   * 关闭gRPC连接
+   */
+  public close(): void {
+    for (const grpcService of this.grpcServices.values()) {
+      grpcService.close()
+    }
+  }
+}

packages/grpc/src/grpc-server.ts

@@ -0,0 +1,45 @@
+import type { ServiceDefinition, UntypedServiceImplementation } from '@grpc/grpc-js'
+import { Server, ServerCredentials } from '@grpc/grpc-js'
+import { Singleton } from './singleton'
+
+export class GrpcServer extends Singleton<GrpcServer> {
+  private readonly server: Server
+  constructor() {
+    super()
+    this.server = new Server()
+  }
+
+  /**
+   * 添加服务
+   */
+  public addService(service: ServiceDefinition, implementation: UntypedServiceImplementation): void {
+    this.server.addService(service, implementation)
+  }
+
+  /**
+   * 移除服务
+   * @param service
+   */
+  public removeService(service: ServiceDefinition): void {
+    this.server.removeService(service)
+  }
+
+  /**
+   * 监听端口，即启动
+   * @param connectUri
+   */
+  public async listen(connectUri: string): Promise<number> {
+    return new Promise((resolve, reject) => {
+      this.server.bindAsync(connectUri, ServerCredentials.createInsecure(), (error, port) => {
+        // 启动成功
+        if (error == null) {
+          console.log('启动成功')
+          resolve(port)
+        }
+        else {
+          reject(error)
+        }
+      })
+    })
+  }
+}

packages/grpc/src/grpc.interface.ts

@@ -0,0 +1,6 @@
+export interface ServiceClaDefinition {
+  // 构造函数
+  ServiceClass: any
+  serviceName: string
+  packageName: string
+}

packages/grpc/src/index.ts

@@ -0,0 +1,4 @@
+export * from './grpc.interface'
+export * from './proto-loader'
+export * from './grpc-client'
+export * from './grpc-server'

packages/grpc/src/proto-loader.ts

@@ -0,0 +1,58 @@
+import type protoLoader from '@grpc/proto-loader'
+import type { GrpcObject } from '@grpc/grpc-js/build/src/make-client'
+import { loadPackageDefinition } from '@grpc/grpc-js'
+import { loadSync } from '@grpc/proto-loader'
+import { Singleton } from './singleton'
+import type { ServiceClaDefinition } from './grpc.interface'
+
+/**
+ * proto文件加载器
+ */
+export class ProtoLoader extends Singleton<ProtoLoader> {
+  private readonly grpcObj: GrpcObject
+
+  private readonly protoPath: string
+  private readonly loaderOptions: protoLoader.Options
+
+  constructor(protoPath: string, loaderOptions: protoLoader.Options) {
+    super(protoPath, loaderOptions)
+    this.protoPath = protoPath
+    this.loaderOptions = loaderOptions
+
+    this.grpcObj = this.getGrpcObj()
+  }
+
+  public getPackageName(): string {
+    return Object.keys(this.grpcObj)[0]
+  }
+
+  /**
+   * 获取rpc Service类定义
+   */
+  public getServiceClassDefinition(): ServiceClaDefinition[] {
+    const packageName = this.getPackageName()
+
+    const services: ServiceClaDefinition[] = []
+    for (const [serviceName, cla] of Object.entries(this.grpcObj[packageName])) {
+      if (cla.service != null) {
+        services.push({
+          ServiceClass: cla,
+          serviceName,
+          packageName,
+        })
+      }
+    }
+
+    return services
+  }
+
+  /**
+   * 加载proto包文件
+   * @private
+   */
+  private getGrpcObj(): GrpcObject {
+    const packageDefinition = loadSync(this.protoPath, this.loaderOptions ?? {})
+
+    return loadPackageDefinition(packageDefinition)
+  }
+}