platform/server/middleware/src/dbAdapter.ts
Andrey Sobolev 1e0edd4eb5
UBERF-8185: Fix duplicate hierarchy clases (#6660)
Signed-off-by: Andrey Sobolev <haiodo@gmail.com>
2024-09-23 11:15:14 +07:00

104 lines
3.3 KiB
TypeScript

//
// Copyright © 2022 Hardcore Engineering Inc.
//
// Licensed under the Eclipse Public License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. You may
// obtain a copy of the License at https://www.eclipse.org/legal/epl-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//
// See the License for the specific language governing permissions and
// limitations under the License.
//
import { DOMAIN_TX, type MeasureContext } from '@hcengineering/core'
import { PlatformError, unknownStatus } from '@hcengineering/platform'
import type {
DbAdapter,
DbConfiguration,
Middleware,
MiddlewareCreator,
PipelineContext,
TxAdapter
} from '@hcengineering/server-core'
import { BaseMiddleware, createServiceAdaptersManager, DbAdapterManagerImpl } from '@hcengineering/server-core'
/**
* @public
*/
export class DBAdapterMiddleware extends BaseMiddleware implements Middleware {
constructor (
context: PipelineContext,
next: Middleware | undefined,
readonly conf: DbConfiguration
) {
super(context, next)
}
static create (conf: DbConfiguration): MiddlewareCreator {
return async (ctx, context, next): Promise<Middleware> => {
const middleware = new DBAdapterMiddleware(context, next, conf)
await middleware.init(ctx)
return middleware
}
}
async init (ctx: MeasureContext): Promise<void> {
const adapters = new Map<string, DbAdapter>()
await ctx.with('create-adapters', {}, async (ctx) => {
if (this.context.storageAdapter == null) {
throw new PlatformError(unknownStatus('StorageSdapter is not specified'))
}
for (const key in this.conf.adapters) {
const adapterConf = this.conf.adapters[key]
adapters.set(
key,
await adapterConf.factory(
ctx,
this.context.hierarchy,
adapterConf.url,
this.context.workspace,
this.context.modelDb,
this.context.storageAdapter
)
)
}
})
const txAdapterName = this.conf.domains[DOMAIN_TX]
const txAdapter = adapters.get(txAdapterName) as TxAdapter
const txAdapterDomains: string[] = []
for (const key in this.conf.domains) {
if (this.conf.domains[key] === txAdapterName) {
txAdapterDomains.push(key)
}
}
await txAdapter.init?.(txAdapterDomains)
const metrics = this.conf.metrics.newChild('📔 server-storage', {})
const defaultAdapter = adapters.get(this.conf.defaultAdapter)
if (defaultAdapter === undefined) {
throw new Error(`No default Adapter for ${this.conf.defaultAdapter}`)
}
this.context.serviceAdapterManager = await createServiceAdaptersManager(
this.conf.serviceAdapters,
this.conf.metrics.newChild('🔌 service adapters', {})
)
// We need to init all next, since we will use model
const adapterManager = new DbAdapterManagerImpl(metrics, this.conf, this.context, defaultAdapter, adapters)
this.context.adapterManager = adapterManager
}
async close (): Promise<void> {
await this.context.adapterManager?.close()
await this.context.serviceAdapterManager?.close()
}
}