feat: init

This commit is contained in:
Rogee
2024-09-30 11:02:26 +08:00
parent 679759846b
commit 694dfd2a4f
90 changed files with 6046 additions and 15 deletions

37
cmd/init.go Normal file
View File

@@ -0,0 +1,37 @@
/*
Copyright © 2024 NAME HERE <EMAIL ADDRESS>
*/
package cmd
import (
"os"
"dyproxy/providers/db"
"github.com/spf13/cobra"
)
// initCmd represents the init command
var initCmd = &cobra.Command{
Use: "init",
Short: "init db",
RunE: func(cmd *cobra.Command, args []string) error {
db, err := db.Connect("")
if err != nil {
return err
}
defer db.Close()
initSql, _ := os.ReadFile("init.sql")
_, err = db.Exec(string(initSql))
if err != nil {
return err
}
return nil
},
}
func init() {
rootCmd.AddCommand(initCmd)
}