diff --git a/src/interface.ts b/src/interface.ts
index aab2526c8..42157c78d 100644
--- a/src/interface.ts
+++ b/src/interface.ts
@@ -115,6 +115,41 @@ export interface ColumnGroupType<RecordType> extends ColumnSharedType<RecordType
 
 export type AlignType = 'start' | 'end' | 'left' | 'right' | 'center' | 'justify' | 'match-parent';
 
+type IsExactlyAny<T> = boolean extends (T extends never ? true : false) ? true : false;
+
+type ExtractIndex<RecordType> = Extract<
+    {
+        [key in Extract<keyof RecordType, string | number>]: key extends never ? [] : IsExactlyAny<RecordType[key]> extends true ? [key, ...DataIndexArray] : [key, ...DataIndexArrayType<RecordType[key]>];
+    }[Extract<keyof RecordType, string | number>],
+    readonly (string | number)[]
+>;
+
+type Unwrap<TArr extends readonly (string | number)[]> = TArr extends { length: 0 }
+    ? []
+    : number extends TArr['length']
+    ? TArr
+    : (string | number)[] extends TArr ? TArr :
+    TArr
+    | (TArr extends { length: 1 }
+        ? TArr
+        : TArr extends { length: 2 }
+        ? [TArr[0]]
+        : TArr extends [...infer U, unknown]
+        ? U extends readonly (string | number)[] ? Unwrap<U> : []
+        : []);
+
+type DataIndexArrayType<RecordType> = IsExactlyAny<RecordType> extends true
+    ? []
+    : RecordType extends string
+    ? [number]
+    : RecordType extends Record<string | number, any>
+    ? Unwrap<ExtractIndex<RecordType>>
+    : RecordType extends any[]
+    ? [number, ...Unwrap<ExtractIndex<RecordType[number]>>]
+    : [];
+
+type DataIndexType<RecordType> = Readonly<DataIndexArrayType<RecordType>> extends { length: 0 } ? DataIndex : Readonly<DataIndexArrayType<RecordType>> | Readonly<DataIndexArrayType<RecordType>>[0]
+
 export interface ColumnType<RecordType> extends ColumnSharedType<RecordType> {
   colSpan?: number;
   dataIndex?: DataIndex<RecordType>;