Skip to content

Commit e9ac7ce

Browse files
committed
fix: use dataTypeMap when creating loader property type declaration
1 parent 03cf87f commit e9ac7ce

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

src/utilities/generateDataLoaderFactory.js

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,12 @@ export default (
113113

114114
loaders.push(createLoaderByIdsDeclaration(loaderName, tableName, indexColumn.name, tableColumnSelector, false));
115115

116-
const loaderType = createLoaderTypePropertyDeclaration(loaderName, indexColumn, indexColumn.mappedTableName, false);
116+
const loaderType = createLoaderTypePropertyDeclaration(
117+
loaderName,
118+
dataTypeMap[indexColumn.dataType] ? dataTypeMap[indexColumn.dataType] : indexColumn.dataType,
119+
indexColumn.mappedTableName,
120+
false
121+
);
117122

118123
loaderTypes.push(loaderType);
119124

@@ -186,7 +191,12 @@ export default (
186191
throw new Error('Unexpected state.');
187192
}
188193

189-
const loaderType = createLoaderTypePropertyDeclaration(loaderName, keyColumn, relation.resource, true);
194+
const loaderType = createLoaderTypePropertyDeclaration(
195+
loaderName,
196+
dataTypeMap[keyColumn.dataType] ? dataTypeMap[keyColumn.dataType] : keyColumn.dataType,
197+
relation.resource,
198+
true
199+
);
190200

191201
loaderTypes.push(loaderType);
192202

0 commit comments

Comments
 (0)