37 lines
1.7 KiB
TypeScript
37 lines
1.7 KiB
TypeScript
import { Effect, Layer, ManagedRuntime } from "effect"
|
|
import * as Context from "effect/Context"
|
|
import { Instance } from "@/project/instance"
|
|
import { LocalContext } from "@/util/local-context"
|
|
import { InstanceRef, WorkspaceRef } from "./instance-ref"
|
|
import { Observability } from "./oltp"
|
|
import { WorkspaceContext } from "@/control-plane/workspace-context"
|
|
|
|
export const memoMap = Layer.makeMemoMapUnsafe()
|
|
|
|
export function attach<A, E, R>(effect: Effect.Effect<A, E, R>): Effect.Effect<A, E, R> {
|
|
try {
|
|
const ctx = Instance.current
|
|
const workspaceID = WorkspaceContext.workspaceID
|
|
return effect.pipe(Effect.provideService(InstanceRef, ctx), Effect.provideService(WorkspaceRef, workspaceID))
|
|
} catch (err) {
|
|
if (!(err instanceof LocalContext.NotFound)) throw err
|
|
}
|
|
return effect
|
|
}
|
|
|
|
export function makeRuntime<I, S, E>(service: Context.Service<I, S>, layer: Layer.Layer<I, E>) {
|
|
let rt: ManagedRuntime.ManagedRuntime<I, E> | undefined
|
|
const getRuntime = () => (rt ??= ManagedRuntime.make(Layer.merge(layer, Observability.layer), { memoMap }))
|
|
|
|
return {
|
|
runSync: <A, Err>(fn: (svc: S) => Effect.Effect<A, Err, I>) => getRuntime().runSync(attach(service.use(fn))),
|
|
runPromiseExit: <A, Err>(fn: (svc: S) => Effect.Effect<A, Err, I>, options?: Effect.RunOptions) =>
|
|
getRuntime().runPromiseExit(attach(service.use(fn)), options),
|
|
runPromise: <A, Err>(fn: (svc: S) => Effect.Effect<A, Err, I>, options?: Effect.RunOptions) =>
|
|
getRuntime().runPromise(attach(service.use(fn)), options),
|
|
runFork: <A, Err>(fn: (svc: S) => Effect.Effect<A, Err, I>) => getRuntime().runFork(attach(service.use(fn))),
|
|
runCallback: <A, Err>(fn: (svc: S) => Effect.Effect<A, Err, I>) =>
|
|
getRuntime().runCallback(attach(service.use(fn))),
|
|
}
|
|
}
|