使用 gin 框架写一个 blog项目

package flag

import sys_flag "flag"

type Option struct {
	DB bool
}

// Parse 解析命令行参数
func Parse() Option {
	db := sys_flag.Bool("db", false, "初始化数据库")
	// 解析命令行参数写入注册的flag里
	sys_flag.Parse()
	return Option{
		DB: *db,
	}
}

// IsWebStop 是否停止 web 项目
func IsWebStop(option Option) bool {
	if option.DB {
		return true
	}
	return false
}

// SwitchOption 根据命令执行不同的函数
func SwitchOption(option Option) {
	if option.DB {
		MakeMigrations()
	}
}

package flag

import (
	"gin_study_blog/common"
	"gin_study_blog/models"
)

// MakeMigrations models开头的都是建的模型文件,统一引用到这个文件执行
func MakeMigrations() {
	var err error
	common.DB.SetupJoinTable(&models.UserModel{}, "CollectsModels", &models.UserCollectsModel{})
	common.DB.SetupJoinTable(&models.MenuModel{}, "Banners", &models.MenuBannerModel{})
	// 生成表的表结构
	err = common.DB.Set("gorm:table_options", "ENGINE=InnoDB").
		AutoMigrate(
			&models.BannerModel{},
			&models.TagModel{},
			&models.MessageModel{},
			&models.AdvertModel{},
			&models.UserModel{},
			&models.CommentModel{},
			&models.ArticleModel{},
			&models.MenuModel{},
			&models.MenuBannerModel{},
			&models.FeedBackModel{},
			&models.LoginDataModel{},
		)
	if err != nil {
		common.Log.Error("生成数据库表结构失败")
		return
	}
	common.Log.Info("生成数据库表结构成功")
}