mirror of
https://github.com/hcengineering/platform.git
synced 2026-08-30 11:49:42 +02:00
93 lines
3.0 KiB
TypeScript
93 lines
3.0 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_MODEL_TX, DOMAIN_TX, withContext, type MeasureContext } from '@hcengineering/core'
|
|
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
|
|
}
|
|
}
|
|
|
|
@withContext('dbAdapter-middleware')
|
|
async init (ctx: MeasureContext): Promise<void> {
|
|
const adapters = new Map<string, DbAdapter>()
|
|
|
|
await ctx.with('create-adapters', {}, async (ctx) => {
|
|
for (const key in this.conf.adapters) {
|
|
const adapterConf = this.conf.adapters[key]
|
|
adapters.set(
|
|
key,
|
|
await adapterConf.factory(
|
|
ctx,
|
|
this.context.contextVars,
|
|
this.context.hierarchy,
|
|
adapterConf.url,
|
|
this.context.workspace,
|
|
this.context.modelDb,
|
|
this.context.storageAdapter
|
|
)
|
|
)
|
|
}
|
|
})
|
|
|
|
const metrics = ctx.newChild('📔 adapters', {})
|
|
|
|
const txAdapterName = this.conf.domains[DOMAIN_TX]
|
|
const txAdapter = adapters.get(txAdapterName) as TxAdapter
|
|
await txAdapter.init?.(metrics, this.context.contextVars, [DOMAIN_TX, DOMAIN_MODEL_TX])
|
|
|
|
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, metrics)
|
|
|
|
// 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()
|
|
}
|
|
}
|