package main

import (
	"fmt"
	k8s_manager "k8s-manager/src"
	"k8s-manager/src/model"
)

func CreateTables() {
	db := k8s_manager.DB()
	err := db.Migrator().AutoMigrate(
		//&model.ProjectModel{},
		//&model.ImageModel{},
		//&model.DataModel{},
		&model.ProjectTrainDataModel{},
	)
	if err != nil {
		fmt.Println(err)
		return
	}
}

func main() {
	k8s_manager.InitConfig()
	CreateTables()
	// db := k8s_manager.DB()
	// p := model.ProjectModel{}
	// db.Preload("TrainData").Where("user_name = ?", "wmj").Find(&p)
	// j, _ := json.Marshal(&p)
	// fmt.Println(string(j))
}