mirror of
https://github.com/FJurmanovic/wallet-go-api.git
synced 2026-02-06 06:08:16 +00:00
41 lines
780 B
Go
41 lines
780 B
Go
package migrations
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"wallet-api/pkg/models"
|
|
|
|
"github.com/go-pg/pg/v10"
|
|
"github.com/go-pg/pg/v10/orm"
|
|
)
|
|
|
|
type WalletsMigration struct {
|
|
Db *pg.DB
|
|
}
|
|
|
|
func (am *WalletsMigration) Create() {
|
|
models := []interface{}{
|
|
(*models.WalletTypeModel)(nil),
|
|
(*models.WalletModel)(nil),
|
|
}
|
|
|
|
for _, model := range models {
|
|
err := am.Db.Model(model).CreateTable(&orm.CreateTableOptions{
|
|
IfNotExists: false,
|
|
FKConstraints: true,
|
|
})
|
|
if err != nil {
|
|
log.Printf("Error Creating Table: %s", err)
|
|
} else {
|
|
fmt.Println("Table created successfully")
|
|
}
|
|
}
|
|
}
|
|
|
|
func (am *WalletsMigration) PopulateTypes() {
|
|
walletTypeModel := new(models.WalletTypeModel)
|
|
walletTypeModel.Init()
|
|
walletTypeModel.Name = "Test"
|
|
am.Db.Model(walletTypeModel).Insert()
|
|
}
|