Extend the type of dataKey to support function

aisin · tangbc · commit 7f38b8c875df · 2020-06-09T14:23:00.000+08:00
diff --git a/src/index.js b/src/index.js
@@ -205,7 +205,9 @@ const VirtualList = Vue.component('virtual-list', {
     },
 
     getUniqueIdFromDataSources () {
-      return this.dataSources.map((dataSource) => dataSource[this.dataKey])
+      return this.dataSources.map((dataSource) => {
+        return typeof this.dataKey === 'function' ? this.dataKey(dataSource) : dataSource[this.dataKey]
+      })
     },
 
     // event called when each item mounted or size changed
@@ -267,15 +269,16 @@ const VirtualList = Vue.component('virtual-list', {
       for (let index = start; index <= end; index++) {
         const dataSource = dataSources[index]
         if (dataSource) {
-          if (Object.prototype.hasOwnProperty.call(dataSource, dataKey)) {
+          const uniqueKey = typeof dataKey === 'function' ? dataKey(dataSource) : dataSource[dataKey]
+          if (uniqueKey) {
             slots.push(h(Item, {
               // key: dataSource[dataKey],
               props: {
                 index,
                 tag: itemTag,
                 event: EVENT_TYPE.ITEM,
                 horizontal: isHorizontal,
-                uniqueKey: dataSource[dataKey],
+                uniqueKey: uniqueKey,
                 source: dataSource,
                 extraProps: extraProps,
                 component: dataComponent,
diff --git a/src/props.js b/src/props.js
@@ -4,7 +4,7 @@
 
 export const VirtualProps = {
   dataKey: {
-    type: String,
+    type: [String, Function],
     required: true
   },
   dataSources: {
