added wallet controller

This commit is contained in:
Fran Jurmanović
2021-05-15 22:23:44 +02:00
parent cc98d0cf49
commit a37386bb36
6 changed files with 148 additions and 1 deletions

View File

@@ -0,0 +1,40 @@
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()
}