init
This commit is contained in:
@@ -0,0 +1,35 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/spf13/cobra"
|
||||
"trojan/core"
|
||||
"trojan/trojan"
|
||||
)
|
||||
|
||||
// upgradeCmd represents the update command
|
||||
var upgradeCmd = &cobra.Command{
|
||||
Use: "upgrade",
|
||||
Short: "升级数据库和trojan配置文件",
|
||||
}
|
||||
|
||||
func upgradeConfig() {
|
||||
domain, _ := core.GetValue("domain")
|
||||
if domain == "" {
|
||||
return
|
||||
}
|
||||
core.WriteDomain(domain)
|
||||
trojan.Restart()
|
||||
}
|
||||
|
||||
func init() {
|
||||
rootCmd.AddCommand(upgradeCmd)
|
||||
upgradeCmd.AddCommand(&cobra.Command{Use: "db", Short: "升级数据库", Run: func(cmd *cobra.Command, args []string) {
|
||||
if err := core.GetMysql().UpgradeDB(); err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
}})
|
||||
upgradeCmd.AddCommand(&cobra.Command{Use: "config", Short: "升级配置文件", Run: func(cmd *cobra.Command, args []string) {
|
||||
upgradeConfig()
|
||||
}})
|
||||
}
|
||||
Reference in New Issue
Block a user