senset/reCoreD-UI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use DAO

Browse Source
This commit is contained in:
Sense T
2024-04-09 16:28:18 +08:00
parent 2369734230
commit 7dd3af3707
20 changed files with 419 additions and 392 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
cmd/config/database.go
View File

@@ -2,6 +2,7 @@ package config
import (
"reCoreD-UI/controllers"
"reCoreD-UI/database"
"github.com/urfave/cli/v2"
)
@@ -26,11 +27,9 @@ func init() {
}
func migrateDatabase(c *cli.Context) error {
controller, err := controllers.NewController(c.String("mysql-dsn"))
if err != nil {
if err := database.Connect(c.String("mysql-dsn")); err != nil {
return err
}
defer controller.Close()
return controller.Migrate()
return controllers.Migrate()
}

15
cmd/config/dns.go
View File

@@ -2,6 +2,7 @@ package config
import (
"reCoreD-UI/controllers"
"reCoreD-UI/database"
"github.com/urfave/cli/v2"
)
@@ -10,13 +11,13 @@ var DNSCommand *cli.Command
func init() {
DNSCommand = &cli.Command{
Name: "dns",
Name: "dns",
Usage: "Config DNS Settings",
Flags: []cli.Flag{
&cli.StringSliceFlag{
Name: "servers",
Usage: "dns servers",
Aliases: []string{"s"},
Name: "servers",
Usage: "dns servers",
Aliases: []string{"s"},
Required: true,
},
},
@@ -25,11 +26,9 @@ func init() {
}
func setDNS(c *cli.Context) error {
controller, err := controllers.NewController(c.String("mysql-dsn"))
if err != nil {
if err := database.Connect(c.String("mysql-dsn")); err != nil {
return err
}
defer controller.Close()
return controller.SetupDNS(c.StringSlice("servers")...)
return controllers.SetupDNS(c.StringSlice("servers")...)
}

7
cmd/config/user.go
View File

@@ -2,6 +2,7 @@ package config
import (
"reCoreD-UI/controllers"
"reCoreD-UI/database"
"github.com/urfave/cli/v2"
)
@@ -33,10 +34,8 @@ func init() {
}
func setUser(c *cli.Context) error {
controller, err := controllers.NewController(c.String("mysql-dsn"))
if err != nil {
if err := database.Connect(c.String("mysql-dsn")); err != nil {
return err
}
defer controller.Close()
return controller.SetupAdmin(c.String("username"), c.String("password"))
return controllers.SetupAdmin(c.String("username"), c.String("password"))
}