TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-06 10:00:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

改进脚本

Browse Source
This commit is contained in:
刘祥超
2020-11-16 23:33:46 +08:00
parent e76ba5cc6b
commit 2bda37f333
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
build/sql.sh
View File

@@ -1,3 +1,3 @@
#!/usr/bin/env bash #!/usr/bin/env bash
go run `dirname $0`/../../cmd/sql-dump/main.go -dir=`dirname $0` go run `dirname $0`/../cmd/sql-dump/main.go -dir=`dirname $0`

16
cmd/sql-dump/main.go
View File

@@ -2,10 +2,10 @@ package main
import ( import (
"encoding/json" "encoding/json"
"fmt"
"github.com/TeaOSLab/EdgeAPI/internal/setup" "github.com/TeaOSLab/EdgeAPI/internal/setup"
_ "github.com/iwind/TeaGo/bootstrap" _ "github.com/iwind/TeaGo/bootstrap"
"github.com/iwind/TeaGo/dbs" "github.com/iwind/TeaGo/dbs"
"github.com/iwind/TeaGo/logs"
"go/format" "go/format"
"io/ioutil" "io/ioutil"
"os" "os"
@@ -16,17 +16,17 @@ import (
func main() { func main() {
db, err := dbs.Default() db, err := dbs.Default()
if err != nil { if err != nil {
logs.Println("[ERROR]" + err.Error()) fmt.Println("[ERROR]" + err.Error())
return return
} }
results, err := setup.NewSQLDump().Dump(db) results, err := setup.NewSQLDump().Dump(db)
if err != nil { if err != nil {
logs.Println("[ERROR]" + err.Error()) fmt.Println("[ERROR]" + err.Error())
return return
} }
resultsJSON, err := json.Marshal(results) resultsJSON, err := json.Marshal(results)
if err != nil { if err != nil {
logs.Println("[ERROR]" + err.Error()) fmt.Println("[ERROR]" + err.Error())
return return
} }
dir, _ := os.Getwd() dir, _ := os.Getwd()
@@ -42,7 +42,7 @@ func main() {
} }
if len(sqlFile) == 0 { if len(sqlFile) == 0 {
logs.Println("[ERROR]can not find sql.go") fmt.Println("[ERROR]can not find sql.go")
return return
} }
content := []byte(`package setup content := []byte(`package setup
@@ -64,14 +64,14 @@ func init() {
`) `)
dst, err := format.Source(content) dst, err := format.Source(content)
if err != nil { if err != nil {
logs.Println("[ERROR]format code failed: " + err.Error()) fmt.Println("[ERROR]format code failed: " + err.Error())
return return
} }
err = ioutil.WriteFile(sqlFile, dst, 0666) err = ioutil.WriteFile(sqlFile, dst, 0666)
if err != nil { if err != nil {
logs.Println("[ERROR]write file failed: " + err.Error()) fmt.Println("[ERROR]write file failed: " + err.Error())
return return
} }
logs.Println("ok") fmt.Println("ok")
} }