package main import ( "fmt" "github.com/gin-gonic/gin" "github.com/jessevdk/go-flags" _ "github.com/lib/pq" "os" ) var options struct { Url string `long:"url" description:"Database connection string"` Host string `long:"host" description:"Server hostname or IP" default:"localhost"` Port int `long:"port" description:"Server port" default:"5432"` User string `long:"user" description:"Database user" default:"postgres"` DbName string `long:"db" description:"Database name" default:"postgres"` Ssl string `long:"ssl" description:"SSL option" default:"disable"` } var dbClient *Client func exitWithMessage(message string) { fmt.Println("Error:", message) os.Exit(1) } func getConnectionString() string { if options.Url != "" { return options.Url } return fmt.Sprintf( "host=%s port=%d user=%s dbname=%s sslmode=disable", options.Host, options.Port, options.User, options.DbName, ) } func initClient() { client, err := NewClient() if err != nil { exitWithMessage(err.Error()) } fmt.Println("Connecting to server...") err = client.Test() if err != nil { exitWithMessage(err.Error()) } fmt.Println("Checking tables...") tables, err := client.Tables() if err != nil { exitWithMessage(err.Error()) } if len(tables) == 0 { exitWithMessage("Database does not have any tables") } dbClient = client } func initOptions() { _, err := flags.ParseArgs(&options, os.Args) if err != nil { os.Exit(1) } } func main() { initOptions() initClient() defer dbClient.db.Close() router := gin.Default() router.GET("/", API_Home) router.GET("/info", API_Info) router.GET("/tables", API_GetTables) router.GET("/tables/:table", API_GetTable) router.GET("/tables/:table/indexes", API_TableIndexes) router.GET("/query", API_RunQuery) router.POST("/query", API_RunQuery) router.GET("/explain", API_ExplainQuery) router.POST("/explain", API_ExplainQuery) router.GET("/history", API_History) router.GET("/static/:type/:name", API_ServeAsset) fmt.Println("Starting server...") fmt.Println("Once started you can view application at http://localhost:8080") router.Run(":8080") }