VC7Q6T03

1func CreateSchema(db *DB, modelsList []interface{}, dropExisting bool) error {
2    for _, model := range modelsList {
3        if dropExisting {
4            err := DropTable(db, model.(string))
5            if err != nil {
6                fmt.Println(err)
7            }
8        }
9 
10        err := CreateTable(db, model.(string))
11        if err != nil {
12            return err
13        }
14    }
15    return nil
16}
17 
18// CreateTable creates a new table in the database
19func CreateTable(db *DB, model interface{}) error {
20    _, err := db.Conn.Exec("CREATE TABLE IF NOT EXISTS "+model.(string))
21    if err != nil {
22        return err
23    }
24    return nil
25}
26 
27// DropTable deletes the existing tables
28func DropTable(db *DB, model interface{}) error {
29    _, err := db.Conn.Exec("DROP TABLE IF EXISTS "+model.(string))
30    if err != nil {
31        panic(err)
32    }
33    return nil
34}
35 
36 
37 
38type User struct {
39    ID                         int64      
40    Email                      string    
41    Name                       string    
42    LastName                   string    
43    Phone                      string    
44    PasswordHash               string    
45    Status                     UserStatus
46    VerificationCode           string    
47    VerificationCodeExpiration time.Time  
48}
49 
50func main () {
51 
52intfs := []interface{}{
53        models.User{},
54    }
55 
56 
57_ = database.CreateSchema(&db, intfs, dropExistingTables)
58    if err != nil {
59        log.Error("Failed to create database schema", "error", err)
60    }