mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-10 09:11:55 +00:00
694 lines
17 KiB
Go
694 lines
17 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"os"
|
|
"os/exec"
|
|
"regexp"
|
|
"runtime"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gin-contrib/cors"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/joho/godotenv"
|
|
"github.com/markbates/pkger"
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"github.com/c9s/bbgo/pkg/bbgo"
|
|
"github.com/c9s/bbgo/pkg/service"
|
|
"github.com/c9s/bbgo/pkg/types"
|
|
)
|
|
|
|
const DefaultBindAddress = "localhost:8080"
|
|
|
|
type Setup struct {
|
|
// Context is the trader context
|
|
Context context.Context
|
|
|
|
// Cancel is the trader context cancel function you want to cancel
|
|
Cancel context.CancelFunc
|
|
|
|
// Token is used for setup api authentication
|
|
Token string
|
|
}
|
|
|
|
type Server struct {
|
|
Config *bbgo.Config
|
|
Environ *bbgo.Environment
|
|
Trader *bbgo.Trader
|
|
Setup *Setup
|
|
Bind string
|
|
OpenInBrowser bool
|
|
}
|
|
|
|
func (s *Server) Run(ctx context.Context) error {
|
|
userConfig := s.Config
|
|
environ := s.Environ
|
|
|
|
r := gin.Default()
|
|
r.Use(cors.New(cors.Config{
|
|
AllowOrigins: []string{"*"},
|
|
AllowHeaders: []string{"Origin", "Content-Type"},
|
|
ExposeHeaders: []string{"Content-Length"},
|
|
AllowMethods: []string{"GET", "POST", "PUT", "DELETE"},
|
|
AllowWebSockets: true,
|
|
AllowCredentials: true,
|
|
MaxAge: 12 * time.Hour,
|
|
}))
|
|
|
|
r.GET("/api/ping", s.ping)
|
|
|
|
if s.Setup != nil {
|
|
r.POST("/api/setup/test-db", s.setupTestDB)
|
|
r.POST("/api/setup/configure-db", s.setupConfigureDB)
|
|
r.POST("/api/setup/strategy/single/:id/session/:session", s.setupAddStrategy)
|
|
r.POST("/api/setup/save", s.setupSaveConfig)
|
|
r.POST("/api/setup/restart", s.setupRestart)
|
|
}
|
|
|
|
r.GET("/api/trades", func(c *gin.Context) {
|
|
if environ.TradeService == nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "database is not configured"})
|
|
return
|
|
}
|
|
|
|
exchange := c.Query("exchange")
|
|
symbol := c.Query("symbol")
|
|
gidStr := c.DefaultQuery("gid", "0")
|
|
lastGID, err := strconv.ParseInt(gidStr, 10, 64)
|
|
if err != nil {
|
|
logrus.WithError(err).Error("last gid parse error")
|
|
c.Status(http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
trades, err := environ.TradeService.Query(service.QueryTradesOptions{
|
|
Exchange: types.ExchangeName(exchange),
|
|
Symbol: symbol,
|
|
LastGID: lastGID,
|
|
Ordering: "DESC",
|
|
})
|
|
if err != nil {
|
|
c.Status(http.StatusBadRequest)
|
|
logrus.WithError(err).Error("order query error")
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"trades": trades,
|
|
})
|
|
})
|
|
|
|
r.GET("/api/orders/closed", func(c *gin.Context) {
|
|
if environ.OrderService == nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "database is not configured"})
|
|
return
|
|
}
|
|
|
|
exchange := c.Query("exchange")
|
|
symbol := c.Query("symbol")
|
|
gidStr := c.DefaultQuery("gid", "0")
|
|
|
|
lastGID, err := strconv.ParseInt(gidStr, 10, 64)
|
|
if err != nil {
|
|
logrus.WithError(err).Error("last gid parse error")
|
|
c.Status(http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
orders, err := environ.OrderService.Query(service.QueryOrdersOptions{
|
|
Exchange: types.ExchangeName(exchange),
|
|
Symbol: symbol,
|
|
LastGID: lastGID,
|
|
Ordering: "DESC",
|
|
})
|
|
if err != nil {
|
|
c.Status(http.StatusBadRequest)
|
|
logrus.WithError(err).Error("order query error")
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"orders": orders,
|
|
})
|
|
})
|
|
|
|
r.GET("/api/trading-volume", func(c *gin.Context) {
|
|
if environ.TradeService == nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "database is not configured"})
|
|
return
|
|
}
|
|
|
|
period := c.DefaultQuery("period", "day")
|
|
segment := c.DefaultQuery("segment", "exchange")
|
|
startTimeStr := c.Query("start-time")
|
|
|
|
var startTime time.Time
|
|
|
|
if startTimeStr != "" {
|
|
v, err := time.Parse(time.RFC3339, startTimeStr)
|
|
if err != nil {
|
|
c.Status(http.StatusBadRequest)
|
|
logrus.WithError(err).Error("start-time format incorrect")
|
|
return
|
|
}
|
|
startTime = v
|
|
|
|
} else {
|
|
switch period {
|
|
case "day":
|
|
startTime = time.Now().AddDate(0, 0, -30)
|
|
|
|
case "month":
|
|
startTime = time.Now().AddDate(0, -6, 0)
|
|
|
|
case "year":
|
|
startTime = time.Now().AddDate(-2, 0, 0)
|
|
|
|
default:
|
|
startTime = time.Now().AddDate(0, 0, -7)
|
|
|
|
}
|
|
}
|
|
|
|
rows, err := environ.TradeService.QueryTradingVolume(startTime, service.TradingVolumeQueryOptions{
|
|
SegmentBy: segment,
|
|
GroupByPeriod: period,
|
|
})
|
|
if err != nil {
|
|
logrus.WithError(err).Error("trading volume query error")
|
|
c.Status(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"tradingVolumes": rows})
|
|
return
|
|
})
|
|
|
|
r.POST("/api/sessions/test", func(c *gin.Context) {
|
|
var sessionConfig bbgo.ExchangeSession
|
|
if err := c.BindJSON(&sessionConfig); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{
|
|
"error": err.Error(),
|
|
})
|
|
return
|
|
}
|
|
|
|
session, err := bbgo.NewExchangeSessionFromConfig(sessionConfig.ExchangeName, &sessionConfig)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{
|
|
"error": err.Error(),
|
|
})
|
|
return
|
|
}
|
|
|
|
var anyErr error
|
|
_, openOrdersErr := session.Exchange.QueryOpenOrders(ctx, "BTCUSDT")
|
|
if openOrdersErr != nil {
|
|
anyErr = openOrdersErr
|
|
}
|
|
|
|
_, balanceErr := session.Exchange.QueryAccountBalances(ctx)
|
|
if balanceErr != nil {
|
|
anyErr = balanceErr
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"success": anyErr == nil,
|
|
"error": anyErr,
|
|
"balance": balanceErr == nil,
|
|
"openOrders": openOrdersErr == nil,
|
|
})
|
|
})
|
|
|
|
r.GET("/api/sessions", func(c *gin.Context) {
|
|
var sessions = []*bbgo.ExchangeSession{}
|
|
for _, session := range environ.Sessions() {
|
|
sessions = append(sessions, session)
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"sessions": sessions})
|
|
})
|
|
|
|
r.POST("/api/sessions", func(c *gin.Context) {
|
|
var sessionConfig bbgo.ExchangeSession
|
|
if err := c.BindJSON(&sessionConfig); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{
|
|
"error": err.Error(),
|
|
})
|
|
return
|
|
}
|
|
|
|
session, err := bbgo.NewExchangeSessionFromConfig(sessionConfig.ExchangeName, &sessionConfig)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{
|
|
"error": err.Error(),
|
|
})
|
|
return
|
|
}
|
|
|
|
if userConfig.Sessions == nil {
|
|
userConfig.Sessions = make(map[string]*bbgo.ExchangeSession)
|
|
}
|
|
userConfig.Sessions[sessionConfig.Name] = session
|
|
|
|
environ.AddExchangeSession(sessionConfig.Name, session)
|
|
|
|
if err := session.Init(ctx, environ); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{
|
|
"error": err.Error(),
|
|
})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"success": true,
|
|
})
|
|
})
|
|
|
|
r.GET("/api/assets", func(c *gin.Context) {
|
|
totalAssets := types.AssetMap{}
|
|
|
|
for _, session := range environ.Sessions() {
|
|
balances := session.Account.Balances()
|
|
|
|
if err := session.UpdatePrices(ctx); err != nil {
|
|
logrus.WithError(err).Error("price update failed")
|
|
c.Status(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
assets := balances.Assets(session.LastPrices())
|
|
|
|
for currency, asset := range assets {
|
|
totalAssets[currency] = asset
|
|
}
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"assets": totalAssets})
|
|
})
|
|
|
|
r.GET("/api/sessions/:session", func(c *gin.Context) {
|
|
sessionName := c.Param("session")
|
|
session, ok := environ.Session(sessionName)
|
|
|
|
if !ok {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("session %s not found", sessionName)})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"session": session})
|
|
})
|
|
|
|
r.GET("/api/sessions/:session/trades", func(c *gin.Context) {
|
|
sessionName := c.Param("session")
|
|
session, ok := environ.Session(sessionName)
|
|
|
|
if !ok {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("session %s not found", sessionName)})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"trades": session.Trades})
|
|
})
|
|
|
|
r.GET("/api/sessions/:session/open-orders", func(c *gin.Context) {
|
|
sessionName := c.Param("session")
|
|
session, ok := environ.Session(sessionName)
|
|
|
|
if !ok {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("session %s not found", sessionName)})
|
|
return
|
|
}
|
|
|
|
marketOrders := make(map[string][]types.Order)
|
|
for symbol, orderStore := range session.OrderStores() {
|
|
marketOrders[symbol] = orderStore.Orders()
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"orders": marketOrders})
|
|
})
|
|
|
|
r.GET("/api/sessions/:session/account", func(c *gin.Context) {
|
|
sessionName := c.Param("session")
|
|
session, ok := environ.Session(sessionName)
|
|
|
|
if !ok {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("session %s not found", sessionName)})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"account": session.Account})
|
|
})
|
|
|
|
r.GET("/api/sessions/:session/account/balances", func(c *gin.Context) {
|
|
sessionName := c.Param("session")
|
|
session, ok := environ.Session(sessionName)
|
|
|
|
if !ok {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("session %s not found", sessionName)})
|
|
return
|
|
}
|
|
|
|
if session.Account == nil {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("the account of session %s is nil", sessionName)})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"balances": session.Account.Balances()})
|
|
})
|
|
|
|
r.GET("/api/sessions/:session/symbols", func(c *gin.Context) {
|
|
sessionName := c.Param("session")
|
|
session, ok := environ.Session(sessionName)
|
|
|
|
if !ok {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("session %s not found", sessionName)})
|
|
return
|
|
}
|
|
|
|
var symbols []string
|
|
for symbol := range session.Markets() {
|
|
symbols = append(symbols, symbol)
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"symbols": symbols})
|
|
})
|
|
|
|
r.GET("/api/sessions/:session/pnl", func(c *gin.Context) {
|
|
c.JSON(200, gin.H{"message": "pong"})
|
|
})
|
|
|
|
r.GET("/api/sessions/:session/market/:symbol/open-orders", func(c *gin.Context) {
|
|
c.JSON(200, gin.H{"message": "pong"})
|
|
})
|
|
|
|
r.GET("/api/sessions/:session/market/:symbol/trades", func(c *gin.Context) {
|
|
c.JSON(200, gin.H{"message": "pong"})
|
|
})
|
|
|
|
r.GET("/api/sessions/:session/market/:symbol/pnl", func(c *gin.Context) {
|
|
c.JSON(200, gin.H{"message": "pong"})
|
|
})
|
|
|
|
r.GET("/api/strategies/single", s.listStrategies)
|
|
r.NoRoute(s.pkgerHandler)
|
|
|
|
if s.OpenInBrowser {
|
|
if runtime.GOOS == "darwin" {
|
|
baseURL := "http://" + DefaultBindAddress
|
|
if len(s.Bind) > 0 {
|
|
baseURL = "http://" + s.Bind
|
|
}
|
|
|
|
go pingAndOpenURL(ctx, baseURL)
|
|
} else {
|
|
logrus.Warnf("%s is not supported for opening browser automatically", runtime.GOOS)
|
|
}
|
|
}
|
|
|
|
if len(s.Bind) > 0 {
|
|
return r.Run(s.Bind)
|
|
}
|
|
|
|
return r.Run() // listen and serve on 0.0.0.0:8080 (for windows "localhost:8080")
|
|
}
|
|
|
|
func (s *Server) setupTestDB(c *gin.Context) {
|
|
payload := struct {
|
|
DSN string `json:"dsn"`
|
|
}{}
|
|
|
|
if err := c.BindJSON(&payload); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "missing arguments"})
|
|
return
|
|
}
|
|
|
|
dsn := payload.DSN
|
|
if len(dsn) == 0 {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "missing dsn argument"})
|
|
return
|
|
}
|
|
|
|
db, err := bbgo.ConnectMySQL(dsn)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
if err := db.Close(); err != nil {
|
|
logrus.WithError(err).Error("db connection close error")
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"success": true})
|
|
}
|
|
|
|
func (s *Server) setupConfigureDB(c *gin.Context) {
|
|
payload := struct {
|
|
DSN string `json:"dsn"`
|
|
}{}
|
|
|
|
if err := c.BindJSON(&payload); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "missing arguments"})
|
|
return
|
|
}
|
|
|
|
dsn := payload.DSN
|
|
if len(dsn) == 0 {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "missing dsn argument"})
|
|
return
|
|
}
|
|
|
|
if err := s.Environ.ConfigureDatabase(c, dsn); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"success": true})
|
|
}
|
|
|
|
func (s *Server) ping(c *gin.Context) {
|
|
c.JSON(http.StatusOK, gin.H{"message": "pong"})
|
|
}
|
|
|
|
func (s *Server) setupAddStrategy(c *gin.Context) {
|
|
sessionName := c.Param("session")
|
|
strategyID := c.Param("id")
|
|
|
|
_, ok := s.Environ.Session(sessionName)
|
|
if !ok {
|
|
c.JSON(http.StatusNotFound, "session not found")
|
|
return
|
|
}
|
|
|
|
var conf map[string]interface{}
|
|
|
|
if err := c.BindJSON(&conf); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "missing arguments"})
|
|
return
|
|
}
|
|
|
|
strategy, err := bbgo.NewStrategyFromMap(strategyID, conf)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
mount := bbgo.ExchangeStrategyMount{
|
|
Mounts: []string{sessionName},
|
|
Strategy: strategy,
|
|
}
|
|
|
|
s.Config.ExchangeStrategies = append(s.Config.ExchangeStrategies, mount)
|
|
|
|
c.JSON(http.StatusOK, gin.H{"success": true})
|
|
}
|
|
|
|
func (s *Server) setupRestart(c *gin.Context) {
|
|
c.JSON(http.StatusOK, gin.H{"success": true})
|
|
}
|
|
|
|
func (s *Server) listStrategies(c *gin.Context) {
|
|
var stashes []map[string]interface{}
|
|
|
|
for _, mount := range s.Config.ExchangeStrategies {
|
|
stash, err := mount.Map()
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
stash["strategy"] = mount.Strategy.ID()
|
|
|
|
stashes = append(stashes, stash)
|
|
}
|
|
|
|
if len(stashes) == 0 {
|
|
c.JSON(http.StatusOK, gin.H{"strategies": []int{}})
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"strategies": stashes})
|
|
}
|
|
|
|
func (s *Server) setupSaveConfig(c *gin.Context) {
|
|
userConfig := s.Config
|
|
environ := s.Environ
|
|
|
|
if len(userConfig.Sessions) == 0 {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "session is not configured"})
|
|
return
|
|
}
|
|
|
|
envVars, err := collectSessionEnvVars(userConfig.Sessions)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
if len(s.Environ.MysqlURL) > 0 {
|
|
envVars["MYSQL_URL"] = environ.MysqlURL
|
|
}
|
|
|
|
dotenvFile := ".env.local"
|
|
if err := moveFileToBackup(dotenvFile); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
if err := godotenv.Write(envVars, dotenvFile); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
out, err := userConfig.YAML()
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
fmt.Println("config file")
|
|
fmt.Println("=================================================")
|
|
fmt.Println(string(out))
|
|
fmt.Println("=================================================")
|
|
|
|
filename := "bbgo.yaml"
|
|
if err := moveFileToBackup(filename); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
if err := ioutil.WriteFile(filename, out, 0666); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"success": true})
|
|
}
|
|
|
|
var pageRoutePattern = regexp.MustCompile("/[a-z]+$")
|
|
|
|
func (s *Server) pkgerHandler(c *gin.Context) {
|
|
fs := pkger.Dir("/frontend/out")
|
|
|
|
// redirect to .html page if the page exists
|
|
if pageRoutePattern.MatchString(c.Request.URL.Path) {
|
|
|
|
_, err := pkger.Stat("/frontend/out/" + c.Request.URL.Path + ".html")
|
|
if err == nil {
|
|
c.Request.URL.Path += ".html"
|
|
}
|
|
}
|
|
|
|
http.FileServer(fs).ServeHTTP(c.Writer, c.Request)
|
|
}
|
|
|
|
func moveFileToBackup(filename string) error {
|
|
stat, err := os.Stat(filename)
|
|
|
|
if err == nil && stat != nil {
|
|
err := os.Rename(filename, filename+"."+time.Now().Format("20060102_150405_07_00"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func collectSessionEnvVars(sessions map[string]*bbgo.ExchangeSession) (envVars map[string]string, err error) {
|
|
envVars = make(map[string]string)
|
|
|
|
for _, session := range sessions {
|
|
if len(session.Key) == 0 && len(session.Secret) == 0 {
|
|
err = fmt.Errorf("session %s key & secret is not empty", session.Name)
|
|
return
|
|
}
|
|
|
|
if len(session.EnvVarPrefix) > 0 {
|
|
envVars[session.EnvVarPrefix+"API_KEY"] = session.Key
|
|
envVars[session.EnvVarPrefix+"API_SECRET"] = session.Secret
|
|
} else if len(session.Name) > 0 {
|
|
sn := strings.ToUpper(session.Name)
|
|
envVars[sn+"API_KEY"] = session.Key
|
|
envVars[sn+"API_SECRET"] = session.Secret
|
|
} else {
|
|
err = fmt.Errorf("session %s name or env var prefix is not defined", session.Name)
|
|
return
|
|
}
|
|
|
|
// reset key and secret so that we won't marshal them to the config file
|
|
session.Key = ""
|
|
session.Secret = ""
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func getJSON(url string, data interface{}) error {
|
|
var client = &http.Client{Timeout: 500 * time.Millisecond}
|
|
r, err := client.Get(url)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer r.Body.Close()
|
|
|
|
return json.NewDecoder(r.Body).Decode(data)
|
|
}
|
|
|
|
func openURL(url string) error {
|
|
cmd := exec.Command("open", url)
|
|
return cmd.Start()
|
|
}
|
|
|
|
func pingAndOpenURL(ctx context.Context, baseURL string) {
|
|
pingURL := baseURL + "/api/ping"
|
|
setupURL := baseURL + "/setup"
|
|
|
|
ticker := time.NewTimer(time.Second)
|
|
defer ticker.Stop()
|
|
|
|
for {
|
|
select {
|
|
|
|
case <-ctx.Done():
|
|
return
|
|
|
|
case <-ticker.C:
|
|
var response map[string]interface{}
|
|
var err = getJSON(pingURL, &response)
|
|
if err == nil {
|
|
if err := openURL(setupURL); err != nil {
|
|
logrus.WithError(err).Errorf("can not call open command to open the web page")
|
|
}
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|