catface_backend_go/app/model/migrate/AutoMigrate.go

package main

// INFO 🐱 开发时测试 标迁移；

import (
	model "catface/app/model"
	"fmt"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

var DB *gorm.DB // 这种写法是方柏包外使用

// 自动迁移表
func autoMigrateTable() {
	err := DB.AutoMigrate(&model.Animal{}, &model.AnmBreed{}, &model.AnmSterilzation{}, &model.AnmStatus{}, &model.AnmGender{},
		&model.AnmVaccination{}, &model.AnmDeworming{})
	if err != nil {
		fmt.Println("autoMigrateTable error:", err)
	}
}

func testInsertSterilzation() {
	// 定义状态数据
	statusesZH := []string{"不明", "未绝育", "已绝育"}
	statusesEN := []string{"unknown", "unsterilized", "sterilized"}

	for i := 0; i < len(statusesZH); i++ {
		sterilzation := model.AnmSterilzation{
			NameZh: statusesZH[i],
			NameEn: statusesEN[i],
		}
		// 3.
		tx := DB.Create(&sterilzation)
		if tx.Error != nil {
			fmt.Println("insert sterilzation error:", tx.Error)
		}
	}
}

func testInsertBreed() {
	// INFO 为方便之后扩展，将 unknown 默认为  1
	colorsZH := []string{
		"不明", "橘白", "奶牛", "白猫", "黑猫", "橘猫", "狸花", "狸白", "三花", "玳瑁", "简州", "彩狸",
	}
	colorsEN := []string{
		"unknown", "orgwhite", "milk", "white", "black", "orange", "li", "liwhite", "flower", "tortoiseshell", "jianzhou", "color",
	}
	for i := 0; i < len(colorsZH); i++ {
		breed := model.AnmBreed{
			BriefModel: model.BriefModel{
				NameZh: colorsZH[i],
				NameEn: colorsEN[i],
			},
		}

		tx := DB.Create(&breed)
		if tx.Error != nil {
			fmt.Println("insert breed error:", tx.Error)
		}
	}
}

func testInsertAnmGender() {
	// 定义性别数据
	gendersZH := []string{"不明", "弟弟", "妹妹"}
	gendersEN := []string{"unknown", "boy", "gril"}

	for i := 0; i < len(gendersZH); i++ {
		anmGender := model.AnmGender{
			BriefModel: model.BriefModel{
				NameZh: gendersZH[i],
				NameEn: gendersEN[i],
			},
		}
		tx := DB.Create(&anmGender)
		if tx.Error != nil {
			fmt.Println("insert gender error:", tx.Error)
		}
	}
}

func testInsertStatus() {
	// 定义状态数据
	statusesZH := []string{"不明", "在校", "毕业", "退学", "喵星"}
	statusesEN := []string{"unknown", "inschool", "graduation", "missing", "catstar"}
	for i := 0; i < len(statusesZH); i++ {
		anmStatus := model.AnmStatus{
			BriefModel: model.BriefModel{
				NameZh: statusesZH[i],
				NameEn: statusesEN[i],
			},
		}

		tx := DB.Create(&anmStatus)
		if tx.Error != nil {
			fmt.Println("insertstatus error:", tx.Error)
		}
	}
}

func testInsertVaccination() {
	// 定义状态数据
	vaccinationsZH := []string{"不明", "未接种", "部分接种", "完全接种"}
	vaccinationsEN := []string{"unknown", "unvaccinated", "partially_vaccinated", "fully_vaccinated"}

	for i := 0; i < len(vaccinationsZH); i++ {
		vaccination := model.AnmVaccination{
			BriefModel: model.BriefModel{
				NameZh: vaccinationsZH[i],
				NameEn: vaccinationsEN[i],
			},
		}

		tx := DB.Create(&vaccination)
		if tx.Error != nil {
			fmt.Println("insert vaccination error:", tx.Error)
		}
	}
}

func testInsertDeworming() {
	// 定义状态数据
	dewormingZH := []string{"不明", "未驱虫", "已驱虫"}
	dewormingEN := []string{"unknown", "undewormed", "dewormed"}

	for i := 0; i < len(dewormingZH); i++ {
		deworming := model.AnmDeworming{
			BriefModel: model.BriefModel{
				NameZh: dewormingZH[i],
				NameEn: dewormingEN[i],
			},
		}

		tx := DB.Create(&deworming)
		if tx.Error != nil {
			fmt.Println("insert vaccination error:", tx.Error)
		}
	}
}

func insertData() {
	testInsertSterilzation()
	fmt.Println("testInsertSterilzation success.")

	testInsertBreed()
	fmt.Println("testInsertBreed success.")

	testInsertStatus()
	fmt.Println("testInsertStatus success.")

	testInsertAnmGender()
	fmt.Println("testInsertAnmGender success.")

	testInsertVaccination()
	fmt.Println("testInsertVaccination success.")

	testInsertDeworming()
	fmt.Println("testInsertDeworming success.")
}

func main() {
	// 1.
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		// "root", "Havocantelope412#", "113.44.68.213", "3306", "hav_cats") // ATT MySQL	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		"root", "havocantelope412", "127.0.0.1", "3306", "hav_cats") // ATT MySQL
	fmt.Println("dsn:", dsn)
	dbMySQL, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		panic(err)
	}
	DB = dbMySQL

	autoMigrateTable()
	fmt.Println("autoMigrateTable over.")

	// insertData() // INFO 记得用完注释掉
}