Create your Gitee Account
Explore and code with more than 6 million developers,Free private repositories !:)
Sign up
Clone or download
main.go 3.64 KB
Copy Edit Web IDE Raw Blame History
张臻 authored 2020-06-05 11:22 . U 初始化默认值
package main
import (
"bufio"
"fmt"
"github.com/joho/godotenv"
"goploy/core"
"goploy/model"
"goploy/route"
"goploy/ws"
"log"
"net/http"
"os"
"os/exec"
"strings"
"time"
_ "github.com/go-sql-driver/mysql"
)
func main() {
install()
godotenv.Load(core.GlobalPath + ".env")
println("应用启动")
println("http://localhost:" + os.Getenv("PORT"))
core.CreateValidator()
model.Init()
ws.Init()
route.Init()
err := http.ListenAndServe(":"+os.Getenv("PORT"), nil) //设置监听的端口
if err != nil {
log.Fatal("ListenAndServe: ", err)
}
}
func install() {
println("检测是否第一次安装")
_, err := os.Stat(".env")
if err == nil || os.IsExist(err) {
println("配置文件已存在,无需重新安装(如果需要重新安装,请先备份好数据库goploy,删除.env文件)")
return
}
inputReader := bufio.NewReader(os.Stdin)
println("安装指引(回车确认输入)")
println("请输入mysql的用户:")
mysqlUser, err := inputReader.ReadString('\n')
if err != nil {
println("There were errors reading, exiting program.")
return
}
println("请输入mysql的密码:")
mysqlPassword, err := inputReader.ReadString('\n')
if err != nil {
println("There were errors reading, exiting program.")
return
}
println("请输入mysql的主机(默认127.0.0.1,不带端口):")
mysqlHost, err := inputReader.ReadString('\n')
if err != nil {
println("There were errors reading, exiting program.")
return
}
if len(mysqlHost) == 0 {
mysqlHost = "127.0.0.1"
}
println("请输入mysql的端口(默认3306):")
mysqlPort, err := inputReader.ReadString('\n')
if err != nil {
println("There were errors reading, exiting program.")
return
}
if len(mysqlPort) == 0 {
mysqlPort = "3306"
}
println("请输入日志目录的绝对路径(默认/tmp/):")
logPath, err := inputReader.ReadString('\n')
if err != nil {
println("There were errors reading, exiting program.")
return
}
if len(logPath) == 0 {
logPath = "/tmp/"
}
println("请输入sshkey的绝对路径(默认/root/.ssh/id_rsa):")
sshFile, err := inputReader.ReadString('\n')
if err != nil {
println("There were errors reading, exiting program.")
return
}
if len(sshFile) == 0 {
sshFile = "/root/.ssh/id_rsa"
}
println("请输入监听端口(默认80,打开网页时的端口):")
port, err := inputReader.ReadString('\n')
if err != nil {
println("There were errors reading, exiting program.")
return
}
if len(port) == 0 {
port = "80"
}
println("开始安装数据库...")
mysqlUser = strings.TrimRight(mysqlUser, "\n")
mysqlPassword = strings.TrimRight(mysqlPassword, "\n")
mysqlHost = strings.TrimRight(mysqlHost, "\n")
mysqlPort = strings.TrimRight(mysqlPort, "\n")
cmd := exec.Command("mysql",
"-h"+mysqlHost,
"-P"+mysqlPort,
"-u"+mysqlUser,
"-p"+mysqlPassword,
"-e",
"source ./goploy.sql")
err = cmd.Run()
if err != nil {
println(err.Error())
println("检查是否有安装mysql客户端")
return
}
println("安装数据库完成")
envContent := ""
envContent += "DB_TYPE=mysql\n"
envContent += fmt.Sprintf(
"DB_CONN=%s:%s@tcp(%s:%s)/goploy?charset=utf8\n",
mysqlUser,
mysqlPassword,
mysqlHost,
mysqlPort)
envContent += fmt.Sprintf("SIGN_KEY=%d\n", time.Now().Unix())
envContent += fmt.Sprintf("LOG_PATH=%s\n", strings.TrimRight(logPath, "\n"))
envContent += fmt.Sprintf("SSHKEY_PATH=%s\n", strings.TrimRight(sshFile, "\n"))
envContent += "ENV=production\n"
envContent += fmt.Sprintf("PORT=%s\n", strings.TrimRight(port, "\n"))
println("开始写入配置文件")
file, err := os.Create(".env")
if err != nil {
panic(err)
}
defer file.Close()
file.WriteString(envContent)
println("写入配置文件完成")
}

Comment ( 0 )

Sign in for post a comment