OpenKF/server/cmd/dbmigration/main.go

// Copyright © 2023 OpenIM open source community. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package main

import (
	"flag"
	"reflect"

	"github.com/OpenIMSDK/OpenKF/server/internal/config"
	"github.com/OpenIMSDK/OpenKF/server/internal/conn/db"
	systemroles "github.com/OpenIMSDK/OpenKF/server/internal/models/system_roles"
	"github.com/OpenIMSDK/OpenKF/server/internal/utils"
	"github.com/OpenIMSDK/OpenKF/server/pkg/log"
)

var (
	configPath string
	isDrop     bool
)

func init() {
	// arg
	flag.StringVar(&configPath, "c", "../../config.yaml", "config file path")
	flag.BoolVar(&isDrop, "d", false, "drop table if exist")
	flag.Parse()

	config.ConfigInit(configPath)
	utils.OpenKFBanner()
	log.InitLogger()
	db.InitMysqlDB()
}

// migrate table.
func main() {
	// get db instance.
	db := db.GetMysqlDB()

	// tables
	tables := []interface{}{
		systemroles.SysUser{},
		systemroles.SysCustomer{},
		systemroles.SysCommunity{},
		systemroles.SysBot{},
	}

	// drop tables if exist.
	if isDrop {
		for _, table := range tables {
			if db.Migrator().HasTable(&table) && db.Migrator().DropTable(&table) != nil {
				log.Panicf("OpenKF Table Migration", "Drop table %v... failed", reflect.TypeOf(table))
			}
			log.Infof("OpenKF Table Migration", "Drop table %v... ok", reflect.TypeOf(table))
		}
	}

	// migrate tables.
	for _, table := range tables {
		err := db.AutoMigrate(&table)
		if err != nil {
			log.Panicf("OpenKF Table Migration", "Migrate table %v... failed", reflect.TypeOf(table))
		}
		log.Infof("OpenKF Table Migration", "Migrate table %v... ok", reflect.TypeOf(table))
	}
}