chore(rule-engine): create add option.dataSourceKeys

zcorky · Nov 2, 2023 · f4f0060 · f4f0060
1 parent 51496ac
commit f4f0060
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 2 deletions.
diff --git a/packages/rule-engine/src/core/create.ts b/packages/rule-engine/src/core/create.ts
@@ -23,6 +23,8 @@ export function create<DataSource>(
 
   // real runner
   async function run(dataSource: Partial<DataSource>) {
+    const dataSourceKeys = options?.dataSourceKeys || Object.keys(dataSource);
+
     let shows: IShowData<DataSource> = {} as any;
 
     const allRuleKeys: Record<string, boolean> = {};
@@ -48,7 +50,7 @@ export function create<DataSource>(
     function initializeShows() {
       traverse(rules);
 
-      shows = Object.keys(dataSource).reduce((all, key) => {
+      shows = dataSourceKeys.reduce((all, key) => {
         // 所有未参与的 key 都应该是 true
         if (!allRuleKeys[key]) {
           all[key] = true;

diff --git a/packages/rule-engine/src/core/sync/create.ts b/packages/rule-engine/src/core/sync/create.ts
@@ -21,6 +21,8 @@ export function create<DataSource>(
 
   // real runner
   function run(dataSource: Partial<DataSource>) {
+    const dataSourceKeys = options?.dataSourceKeys || Object.keys(dataSource);
+
     let shows: IShowData<DataSource> = {} as any;
 
     const allRuleKeys: Record<string, boolean> = {};
@@ -46,7 +48,7 @@ export function create<DataSource>(
     function initializeShows() {
       traverse(rules);
 
-      shows = Object.keys(dataSource).reduce((all, key) => {
+      shows = dataSourceKeys.reduce((all, key) => {
         // 所有未参与的 key 都应该是 true
         if (!allRuleKeys[key]) {
           all[key] = true;

diff --git a/packages/rule-engine/src/core/sync/types.ts b/packages/rule-engine/src/core/sync/types.ts
@@ -110,4 +110,6 @@ export type IOnHitAttr<DataSource> = (
 export interface Options<DataSource> {
   defaultOnScaleTo?: IOnScaleTo<DataSource>;
   defaultOnHitAttr?: IOnHitAttr<DataSource>;
+  //
+  dataSourceKeys?: string[];
 }
diff --git a/packages/rule-engine/src/core/types.ts b/packages/rule-engine/src/core/types.ts
@@ -110,4 +110,5 @@ export type IOnHitAttr<DataSource> = (
 export interface Options<DataSource> {
   defaultOnScaleTo?: IOnScaleTo<DataSource>;
   defaultOnHitAttr?: IOnHitAttr<DataSource>;
+  dataSourceKeys?: string[];
 }