128 lines
3.4 KiB
Go
128 lines
3.4 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/joho/godotenv"
|
|
_ "github.com/lib/pq"
|
|
"github.com/namsral/flag"
|
|
)
|
|
|
|
var (
|
|
DbUrl string
|
|
db *DB
|
|
)
|
|
|
|
type DB struct {
|
|
*sql.DB
|
|
}
|
|
|
|
func init() {
|
|
err := godotenv.Load()
|
|
if err != nil {
|
|
log.Fatal("Error loading .env file")
|
|
}
|
|
flag.StringVar(&DbUrl, "DATABASE_URL", "", "Database URL")
|
|
flag.Parse()
|
|
validateFlags()
|
|
db = initDb()
|
|
_, err = db.Exec("SELECT 1")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func initDb() *DB {
|
|
db, err := sql.Open("postgres", DbUrl)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
return &DB{DB: db}
|
|
}
|
|
|
|
func validateFlags() {
|
|
if DbUrl == "" {
|
|
log.Fatal("DATABASE_URL not set")
|
|
}
|
|
}
|
|
|
|
func (db *DB) FetchMarket(marketId int, market *Market) error {
|
|
if err := db.QueryRow("SELECT id, description FROM markets WHERE id = $1", marketId).Scan(&market.Id, &market.Description); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (db *DB) FetchShares(marketId int, shares *[]Share) error {
|
|
rows, err := db.Query("SELECT id, market_id, description FROM shares WHERE market_id = $1 ORDER BY description DESC", marketId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
var share Share
|
|
rows.Scan(&share.Id, &share.MarketId, &share.Description)
|
|
*shares = append(*shares, share)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (db *DB) FetchOrders(marketId int, orders *[]Order) error {
|
|
rows, err := db.Query(""+
|
|
"SELECT o.id, share_id, o.pubkey, o.side, o.quantity, o.price, s.description, o.order_id "+
|
|
"FROM orders o "+
|
|
"JOIN invoices i ON o.invoice_id = i.id "+
|
|
"JOIN shares s ON o.share_id = s.id "+
|
|
"WHERE share_id = ANY(SELECT id FROM shares WHERE market_id = $1) "+
|
|
"AND i.confirmed_at IS NOT NULL "+
|
|
"ORDER BY price DESC", marketId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
var order Order
|
|
rows.Scan(&order.Id, &order.ShareId, &order.Pubkey, &order.Side, &order.Quantity, &order.Price, &order.Share.Description, &order.OrderId)
|
|
*orders = append(*orders, order)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (db *DB) CreateOrder(order *Order) error {
|
|
if _, err := db.Exec(""+
|
|
"INSERT INTO orders(share_id, pubkey, side, quantity, price, invoice_id) "+
|
|
"VALUES ($1, $2, $3, $4, $5, $6)",
|
|
order.ShareId, order.Pubkey, order.Side, order.Quantity, order.Price, order.InvoiceId); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (db *DB) CreateInvoice(invoice *Invoice) error {
|
|
if err := db.QueryRow(""+
|
|
"INSERT INTO invoices(pubkey, msats, preimage, hash, bolt11, created_at, expires_at) "+
|
|
"VALUES($1, $2, $3, $4, $5, $6, $7) "+
|
|
"RETURNING id",
|
|
invoice.Pubkey, invoice.Msats, invoice.Preimage, invoice.PaymentHash, invoice.PaymentRequest, invoice.CreatedAt, invoice.ExpiresAt).Scan(&invoice.Id); err != nil {
|
|
panic(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (db *DB) FetchInvoice(invoiceId string, invoice *Invoice) error {
|
|
if err := db.QueryRow(""+
|
|
"SELECT id, pubkey, msats, preimage, hash, bolt11, created_at, expires_at, confirmed_at, held_since FROM invoices WHERE id = $1", invoiceId).Scan(&invoice.Id, &invoice.Pubkey, &invoice.Msats, &invoice.Preimage, &invoice.PaymentHash, &invoice.PaymentRequest, &invoice.CreatedAt, &invoice.ExpiresAt, &invoice.ConfirmedAt, &invoice.HeldSince); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (db *DB) ConfirmInvoice(hash string, confirmedAt time.Time, msatsReceived int) error {
|
|
if _, err := db.Exec("UPDATE invoices SET confirmed_at = $2, msats_received = $3 WHERE hash = $1", hash, confirmedAt, msatsReceived); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|