initial commit

1 files changed, 45 insertions, 0 deletions

diff --git a/db/market_config.go b/db/market_config.go
new file mode 100644
index 0000000..b26c092
--- /dev/null
+++ b/db/market_config.go
@@ -0,0 +1,45 @@
+package db
+
+import "github.com/go-pg/pg"
+
+type MarketConfig struct {
+        Id         int64
+        MarketName string `sql:",notnull"`
+        UserId     int64  `sql:",notnull"`
+        Config     map[string]string
+}
+
+func InsertMarketConfig(config *MarketConfig) error {
+        return DB.Insert(config)
+}
+
+func GetUserMarketConfig(userId int64, market string) (*MarketConfig, error) {
+        var config MarketConfig
+
+        err := DB.Model(&config).Where("user_id = ?", userId).Where("market_name = ?", market).First()
+
+        if err != nil && err != pg.ErrNoRows {
+                return nil, err
+        }
+
+        if err == pg.ErrNoRows {
+                return nil, nil
+        } else {
+                return &config, nil
+        }
+}
+
+func SetUserMarketConfig(userId int64, market string, newConfig map[string]string) (*MarketConfig, error) {
+        config := MarketConfig{
+                UserId:     userId,
+                MarketName: market,
+                Config:     newConfig,
+        }
+
+        _, err := DB.Model(&config).
+                OnConflict("(user_id, market_name) DO UPDATE").
+                Set("config = ?", newConfig).
+                Insert()
+
+        return &config, err
+}