Add CardtraderToken field to config file
This commit is contained in:
parent
93c2b55754
commit
429732d076
|
@ -1,8 +1,11 @@
|
||||||
{
|
{
|
||||||
|
"DatabaseConfig": {
|
||||||
"User": "<your username>",
|
"User": "<your username>",
|
||||||
"Passwd": "<your password>",
|
"Passwd": "<your password>",
|
||||||
"Net": "tcp",
|
"Net": "tcp",
|
||||||
"Addr": "127.0.0.1:3306",
|
"Addr": "127.0.0.1:3306",
|
||||||
"DBName": "<your database name>",
|
"DBName": "<your database name>",
|
||||||
"AllowNativePasswords": true
|
"AllowNativePasswords": true
|
||||||
|
},
|
||||||
|
"CardtraderToken": "<your cardtrader token>"
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,29 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"github.com/go-sql-driver/mysql"
|
||||||
|
)
|
||||||
|
|
||||||
|
type SevenkeysConfig struct {
|
||||||
|
DatabaseConfig mysql.Config `json:"DatabaseConfig"`
|
||||||
|
CardtraderToken string `json:"CardtraderToken"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func ReadConfigFromFile(filename string) SevenkeysConfig {
|
||||||
|
bytes, err := os.ReadFile(filename)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var config SevenkeysConfig
|
||||||
|
err = json.Unmarshal(bytes, &config)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return config
|
||||||
|
}
|
|
@ -2,31 +2,12 @@ package database
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"encoding/json"
|
|
||||||
"log"
|
"log"
|
||||||
"os"
|
|
||||||
|
|
||||||
"github.com/go-sql-driver/mysql"
|
"github.com/go-sql-driver/mysql"
|
||||||
)
|
)
|
||||||
|
|
||||||
func readConfigFromFile(filename string) mysql.Config {
|
func GetDatabaseFromConfig(config mysql.Config) *sql.DB {
|
||||||
bytes, err := os.ReadFile(filename)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var config mysql.Config
|
|
||||||
err = json.Unmarshal(bytes, &config)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return config
|
|
||||||
}
|
|
||||||
|
|
||||||
func GetDatabaseFromConfig(filename string) *sql.DB {
|
|
||||||
config := readConfigFromFile(filename)
|
|
||||||
|
|
||||||
db, err := sql.Open("mysql", config.FormatDSN())
|
db, err := sql.Open("mysql", config.FormatDSN())
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
|
@ -8,6 +8,7 @@ import (
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"log"
|
"log"
|
||||||
"os"
|
"os"
|
||||||
|
"sevenkeys/config"
|
||||||
"sevenkeys/database"
|
"sevenkeys/database"
|
||||||
"sevenkeys/delverlens"
|
"sevenkeys/delverlens"
|
||||||
"sevenkeys/logic"
|
"sevenkeys/logic"
|
||||||
|
@ -37,7 +38,8 @@ func main() {
|
||||||
|
|
||||||
flag.Parse()
|
flag.Parse()
|
||||||
|
|
||||||
db := database.GetDatabaseFromConfig("config." + profile + ".json")
|
config := config.ReadConfigFromFile("config." + profile + ".json")
|
||||||
|
db := database.GetDatabaseFromConfig(config.DatabaseConfig)
|
||||||
|
|
||||||
if len(flag.Args()) == 0 {
|
if len(flag.Args()) == 0 {
|
||||||
fmt.Fprintln(os.Stderr, "Please specify a subcommand.")
|
fmt.Fprintln(os.Stderr, "Please specify a subcommand.")
|
||||||
|
|
Loading…
Reference in New Issue