spy-client 2.1.1 → 2.1.7

package/src/spy-local-cache.ts

@@ -6,7 +6,9 @@ interface Option {
     key: string;
     maxRecordLen: number;
     interval: number;
-    onFlush: (res: any) => void;
+    onFlush?: (res: any) => void;
+    onSave?: (res: any) => void;
+    onAdd?: (res: any) => boolean;
     storage: 'indexedDB' | 'localstorage' | 'empty';
 }
 
@@ -68,16 +70,78 @@ class LS extends Storage {
 }
 
 class IndexedDB extends Storage {
+    private readonly databaseName = 'spyLC';
+    private db: IDBDatabase | null = null;
+    private readonly setQueue: any[] = [];
+    private readonly getQueue: any[] = [];
+    constructor() {
+        super();
+
+        let request = window.indexedDB.open(this.databaseName);
+
+        request.onupgradeneeded = event => {
+            // @ts-ignore ts默认的EventTarget类型有问题，先ignore掉
+            this.db = event.target && event.target.result;
+            if (this.db && !this.db.objectStoreNames.contains(this.databaseName)) {
+                this.db.createObjectStore(this.databaseName, {keyPath: 'key'});
+            }
+            this.runQueueTask();
+        };
+
+        request.onsuccess = () => {
+            this.db = request.result;
+            this.runQueueTask();
+        };
+
+    }
     static isSupport() {
         return !!window.indexedDB;
     }
 
     set(key: string, value: any) {
+        // 因为indexDB是异步初始化的，如果set的时候未初始化，先缓存之后再执行set
+        if (!this.db) {
+            this.setQueue.push({
+                key,
+                value,
+            });
+            return;
+        }
+        this.db.transaction([this.databaseName], 'readwrite')
+            .objectStore(this.databaseName)
+            .add({
+                key,
+                value,
+            });
 
     }
 
     get(key: string, cb: any) {
+        // 因为indexDB是异步初始化的，如果get的时候未初始化，先缓存之后再执行get
+        if (!this.db) {
+            this.getQueue.push({
+                key,
+                cb,
+            });
+            return;
+        }
+        const transaction = this.db.transaction([this.databaseName]);
+        const objectStore = transaction.objectStore(this.databaseName);
+        const request = objectStore.get(key);
 
+        request.onsuccess = () => {
+            const result = request.result ? request.result : {value: ''};
+            cb(result.value);
+        };
+    }
+
+    private runQueueTask() {
+        this.setQueue.forEach(ele => {
+            this.set(ele.key, ele.value);
+        });
+        this.getQueue.forEach(ele => {
+            this.get(ele.key, ele.cb);
+        });
     }
 }
 
@@ -217,6 +281,10 @@ export default class SpyLocalCache {
             interval: 500,
             maxRecordLen: 30,
             onFlush: () => {},
+            onSave: () => {},
+            onAdd: () => {
+                return true;
+            },
             storage: IndexedDB.isSupport()
                 ? 'indexedDB'
                 : LS.isSupport()
@@ -255,6 +323,12 @@ export default class SpyLocalCache {
     }
 
     addLog(info: any) {
+        if (this.option.onAdd) {
+            if (!this.option.onAdd(info)) {
+                return;
+            }
+        }
+
         info = JSON.stringify(info);
 
         this.tmpList.push(info as string);
@@ -300,23 +374,42 @@ export default class SpyLocalCache {
     }
 
     save() {
+        const st = Date.now();
         this.getData((list: any[]) => {
             // 只保留最近的maxRecordLen条日志
             const reserveLen = this.option.maxRecordLen - 1;
             const originList = list.length > reserveLen ? list.slice(list.length - reserveLen, list.length) : list;
-            let content = originList.concat(this.tmpList).join('\n');
-            let data = this.zip(content);
-            let codesStr = '';
-            if (data.codes) {
-                codesStr = JSON.stringify(data.codes);
-                this.storage.set(this.option.key + 'Codes', codesStr);
+            let newList = originList.concat(this.tmpList);
+            let content = newList.join('\n');
+            let error: Error | null = null;
+            let len = 0;
+            try {
+                let data = this.zip(content);
+                let codesStr = '';
+                if (data.codes) {
+                    codesStr = JSON.stringify(data.codes);
+                    this.storage.set(this.option.key + 'Codes', codesStr);
+                }
+                else {
+                    this.storage.rm(this.option.key + 'Codes');
+                }
+
+                this.storage.set(this.option.key, data.result);
+                len = data.result.length;
             }
-            else {
-                this.storage.rm(this.option.key + 'Codes');
+            catch (e) {
+                error = e;
+                console.error(e);
             }
-            this.storage.set(this.option.key, data.result);
 
             this.tmpList = [];
+
+            this.option.onSave && this.option.onSave({
+                cost: Date.now() -  st,
+                length: len / 1024,
+                list: newList,
+                error: error,
+            });
         });
     }