fix(util): OrmUtils

1 files changed, 9 insertions, 3 deletions

diff --git a/util/scripts/benchmark.js b/util/scripts/benchmark.js

index f4f34e9e..e7435191 100644
--- a/util/scripts/benchmark.js
+++ b/util/scripts/benchmark.js
@@ -1,22 +1,28 @@
 const typeorm = require("typeorm");
-const Database = require("../dist/util/Database");
 const Models = require("../dist/entities");
+const { PrimaryColumn } = require("typeorm");
 
 function shouldIncludeEntity(name) {
-	return ![BaseClassWithoutId, PrimaryColumn, BaseClass, PrimaryGeneratedColumn].map((x) => x.name).includes(name);
+	return ![Models.BaseClassWithoutId, PrimaryColumn, Models.BaseClass, Models.PrimaryGeneratedColumn]
+		.map((x) => x?.name)
+		.includes(name);
 }
 
 async function main() {
+	console.log("starting");
 	const db = new typeorm.DataSource({
-		driver: "sqlite",
+		type: "sqlite",
 		database: ":memory:",
 		entities: Object.values(Models).filter((x) => x.constructor.name == "Function" && shouldIncludeEntity(x.name)),
+		synchronize: true,
 	});
 	await db.initialize();
+	console.log("Initialized database");
 
 	for (var i = 0; i < 100; i++) {
 		await Models.User.register({ username: "User" + i });
 		console.log("registered user " + i);
 	}
 }
+
 main();