import "reflect-metadata"
import { DataSource } from "../../../src/data-source/DataSource"
import {
    closeTestingConnections,
    createTestingConnections,
    reloadTestingDatabases,
} from "../../utils/test-utils"
import { Document } from "./entity/Document"

describe("benchmark > bulk-save > case2", () => {
    let connections: DataSource[]
    before(
        async () =>
            (connections = await createTestingConnections({
                __dirname,
                enabledDrivers: ["postgres"],
            })),
    )
    beforeEach(() => reloadTestingDatabases(connections))
    after(() => closeTestingConnections(connections))

    it("testing bulk save of 10000 objects", () =>
        Promise.all(
            connections.map(async (connection) => {
                const documents: Document[] = []
                for (let i = 0; i < 10000; i++) {
                    const document = new Document()

                    document.id = i.toString()
                    document.docId = "label/" + i
                    document.context =
                        "Lorem ipsum dolor sit amet, consectetur adipiscing elit. Praesent vel faucibus nunc. Etiam volutpat vel urna in scelerisque. Cras a erat ipsum. "
                    document.label = "label/" + i
                    document.distributions = [
                        {
                            weight: "0.9",
                            id: i,
                            docId: i,
                        },
                        {
                            weight: "0.23123",
                            id: i,
                            docId: i,
                        },
                        {
                            weight: "0.12312",
                            id: i,
                            docId: i,
                        },
                    ]
                    document.date = new Date()

                    documents.push(document)
                    // await connection.manager.save(document);
                    // await connection.manager.insert(Document, document);
                }

                await connection.manager.save(documents)
                // await connection.manager.insert(Document, documents);
            }),
        ))
})
