SQLForge/src/main/java/io/github/mrarcode/sqlforge/Main.java at master · MrArCode/SQLForge · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
package io.github.mrarcode.sqlforge;

import io.github.mrarcode.sqlforge.generators.FakerDataGenerator;
import io.github.mrarcode.sqlforge.generators.ScriptBuilder;
import io.github.mrarcode.sqlforge.models.Column;
import io.github.mrarcode.sqlforge.models.Table;
import io.github.mrarcode.sqlforge.models.Types;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;

public class Main {
    public static void main(String[] args) {


        Table usersTable = new Table("Users");

        usersTable.addColumn(new Column("ID", Types.INTEGER)
                .setPrimaryKey(true)
                .setNotNull(true));

        usersTable.addColumn(new Column("FirstName", Types.FIRSTNAME)
                .setNotNull(true));

        usersTable.addColumn(new Column("LastName", Types.LASTNAME)
                .setNotNull(true));

        usersTable.addColumn(new Column("Email", Types.EMAIL)
                .setUnique(true)
                .setNotNull(true));


        Table ordersTable = new Table("Orders");
        ordersTable.addColumn(new Column("OrderID", Types.INTEGER)
                .setPrimaryKey(true)
                .setNotNull(true));

        ordersTable.addColumn(new Column("UserID", Types.INTEGER)
                .setForeignKey("Users(ID)")
                .setNotNull(true));

        ordersTable.addColumn(new Column("OrderDate", Types.DATE_TIME)
                .setNotNull(true));


        FakerDataGenerator dataGenerator = new FakerDataGenerator();


        String script = ScriptBuilder.builder()
                .addTable(usersTable)
                .addTable(ordersTable)
                .includeCreate()
                .includeInsert( 10)
                .includeRead()
                .includeUpdate(10)
                .includeDelete(10)
                .withDataGenerator(dataGenerator)
                .buildScript();

        System.out.println(script);

        Path outputFile = Path.of("output", "script.sql");
        try {
            Files.createDirectories(outputFile.getParent());
            Files.writeString(outputFile, script);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}