海外纯英文在线客服系统高效可靠的数据库初始化方案设计

package cmd

import (
    "github.com/spf13/cobra"
    "github.com/taoshihan1991/imaptool/models"
    "github.com/taoshihan1991/imaptool/tools"
    "log"
    "os"
    "strings"
)

var installCmd = &cobra.Command{
    Use:   "install",
    Short: "Initialize database and import data", // More precise description
    Run: func(cmd *cobra.Command, args []string) {
        install()
    },
}

func install() {
    // Check if already installed
    if ok, _ := tools.IsFileNotExist("./install.lock"); !ok {
        log.Println("Please remove ./install.lock file to reinstall")
        os.Exit(1)
    }

    // Verify required files exist
    sqlFile := "import.sql"
    dataExists, _ := tools.IsFileExist(sqlFile)
    if !dataExists {
        log.Println("Configuration file config/mysql.json or database import file import.sql not found")
        os.Exit(1)
    }

    // Execute SQL statements
    sqls, err := os.ReadFile(sqlFile)
    if err != nil {
        log.Printf("Failed to read SQL file %s: %v\n", sqlFile, err)
        os.Exit(1)
    }
    sqlArr := strings.Split(string(sqls), ";")
    for _, sql := range sqlArr {
        sql = strings.TrimSpace(sql)
        if sql == "" {
            continue
        }
        err := models.Execute(sql)
        if err != nil {
            log.Printf("SQL execution failed: %s\nError: %v\n", sql, err)
            log.Println("Database initialization failed - please check SQL statements")
            os.Exit(1)
        }
        log.Printf("Executed successfully: %s\n", sql)
    }

    // Create installation lock file
    installFile, err := os.OpenFile("./install.lock", os.O_RDWR|os.O_CREATE, os.ModePerm)
    if err != nil {
        log.Printf("Failed to create lock file: %v\n", err)
        os.Exit(1)
    }
    defer installFile.Close()

    _, err = installFile.WriteString("gofly live chat installation complete")
    if err != nil {
        log.Printf("Failed to write lock file: %v\n", err)
        os.Exit(1)
    }

    log.Println("Database initialization completed successfully")
}