From f4f00605fc9d4c7762fc6894b164eee812edb577 Mon Sep 17 00:00:00 2001
From: Zero <tobewhatwewant@gmail.com>
Date: Thu, 2 Nov 2023 13:05:56 +0800
Subject: [PATCH] chore(rule-engine): create add option.dataSourceKeys

---
 packages/rule-engine/src/core/create.ts      | 4 +++-
 packages/rule-engine/src/core/sync/create.ts | 4 +++-
 packages/rule-engine/src/core/sync/types.ts  | 2 ++
 packages/rule-engine/src/core/types.ts       | 1 +
 4 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/packages/rule-engine/src/core/create.ts b/packages/rule-engine/src/core/create.ts
index 3eada62a..e7ee7878 100644
--- 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
index 76b314d0..423c6acd 100644
--- 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
index ff3877c2..734d42ff 100644
--- 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
index 5cdc3560..1cc5fb13 100644
--- 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[];
 }