mirror of
https://github.com/toeverything/AFFiNE.git
synced 2026-02-14 05:14:54 +00:00
refactor(editor): introduce store container to make implement doc easier (#12146)
<!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit - **New Features** - Introduced a new store management system for handling document stores, improving efficiency and flexibility when working with document data. - **Refactor** - Updated internal store handling to use a centralized store container, simplifying store retrieval and removal across various components. - Renamed and updated several store-related method signatures for consistency and clarity. - Replaced editor extension loading logic with a new local implementation for better modularity. - **Chores** - Improved and streamlined the export of store-related modules for better maintainability. - Removed obsolete and redundant code related to previous store management approaches. <!-- end of auto-generated comment: release notes by coderabbit.ai -->
This commit is contained in:
@@ -3,9 +3,8 @@ import {
|
||||
AwarenessStore,
|
||||
type Doc,
|
||||
type ExtensionType,
|
||||
type GetBlocksOptions,
|
||||
type Query,
|
||||
Store,
|
||||
type GetStoreOptions,
|
||||
StoreContainer,
|
||||
type Workspace,
|
||||
type YBlock,
|
||||
} from '@blocksuite/affine/store';
|
||||
@@ -21,7 +20,7 @@ type DocOptions = {
|
||||
export class DocImpl implements Doc {
|
||||
private readonly _collection: Workspace;
|
||||
|
||||
private readonly _storeMap = new Map<string, Store>();
|
||||
private readonly _storeContainer: StoreContainer;
|
||||
|
||||
private readonly _initSpaceDoc = () => {
|
||||
{
|
||||
@@ -105,19 +104,15 @@ export class DocImpl implements Doc {
|
||||
|
||||
this._yBlocks = this._ySpaceDoc.getMap('blocks');
|
||||
this._collection = collection;
|
||||
}
|
||||
|
||||
private _getReadonlyKey(readonly?: boolean): 'true' | 'false' {
|
||||
return (readonly?.toString() as 'true' | 'false') ?? 'false';
|
||||
this._storeContainer = new StoreContainer(this);
|
||||
}
|
||||
|
||||
clear() {
|
||||
this._yBlocks.clear();
|
||||
}
|
||||
|
||||
clearQuery(query: Query, readonly?: boolean) {
|
||||
const key = this._getQueryKey({ readonly, query });
|
||||
this._storeMap.delete(key);
|
||||
get removeStore() {
|
||||
return this._storeContainer.removeStore;
|
||||
}
|
||||
|
||||
private _destroy() {
|
||||
@@ -134,58 +129,26 @@ export class DocImpl implements Doc {
|
||||
}
|
||||
}
|
||||
|
||||
private readonly _getQueryKey = (
|
||||
idOrOptions: string | { readonly?: boolean; query?: Query }
|
||||
) => {
|
||||
if (typeof idOrOptions === 'string') {
|
||||
return idOrOptions;
|
||||
}
|
||||
const { readonly, query } = idOrOptions;
|
||||
const readonlyKey = this._getReadonlyKey(readonly);
|
||||
const key = JSON.stringify({
|
||||
readonlyKey,
|
||||
query,
|
||||
});
|
||||
return key;
|
||||
};
|
||||
|
||||
getStore({
|
||||
readonly,
|
||||
query,
|
||||
provider,
|
||||
extensions,
|
||||
id,
|
||||
}: GetBlocksOptions = {}) {
|
||||
let idOrOptions: string | { readonly?: boolean; query?: Query };
|
||||
if (readonly || query) {
|
||||
idOrOptions = { readonly, query };
|
||||
} else if (!id) {
|
||||
idOrOptions = this.workspace.idGenerator();
|
||||
} else {
|
||||
idOrOptions = id;
|
||||
}
|
||||
const key = this._getQueryKey(idOrOptions);
|
||||
|
||||
if (this._storeMap.has(key)) {
|
||||
return this._storeMap.get(key) as Store;
|
||||
}
|
||||
|
||||
}: GetStoreOptions = {}) {
|
||||
const storeExtensions = getStoreManager().get('store');
|
||||
const extensionSet = new Set(
|
||||
storeExtensions.concat(extensions ?? []).concat(this.storeExtensions)
|
||||
);
|
||||
const exts = storeExtensions
|
||||
.concat(extensions ?? [])
|
||||
.concat(this.storeExtensions);
|
||||
const extensionSet = new Set(exts);
|
||||
|
||||
const doc = new Store({
|
||||
doc: this,
|
||||
return this._storeContainer.getStore({
|
||||
id,
|
||||
readonly,
|
||||
query,
|
||||
provider,
|
||||
extensions: Array.from(extensionSet),
|
||||
});
|
||||
|
||||
this._storeMap.set(key, doc);
|
||||
|
||||
return doc;
|
||||
}
|
||||
|
||||
load(initFn?: () => void): this {
|
||||
|
||||
Reference in New Issue
Block a user