You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

67 lines
1.4 KiB

package storage
import (
"database/sql"
"fmt"
"github.com/davecgh/go-spew/spew"
"github.com/lib/pq"
)
type postgres struct {
db *sql.DB
}
func New(dsn string) (Service, error) {
pool, err := sql.Open("postgres", dsn)
if err != nil {
return nil, fmt.Errorf("could not open database connection: %v", err)
}
return &postgres{pool}, nil
}
func (s *postgres) Close() error {
return s.db.Close()
}
func (s *postgres) Subscribe(topic string, sub Subscriber) error {
_, err := s.db.Exec(
`INSERT INTO "subscribers" ("topic", "callback", "lease_seconds", "secret", "created") VALUES ($1, $2, $3, $4, now())`,
topic,
sub.Callback,
sub.LeaseSeconds,
sub.Secret,
)
if e, ok := err.(pq.Error); ok {
spew.Dump(e)
}
return err
}
func (s *postgres) Unsubscribe(topic, callback string) error {
_, err := s.db.Exec(
`DELETE FROM "subscribers" WHERE "topic" = $1 AND "callback" = $2`,
topic,
callback,
)
return err
}
func (s *postgres) Subscribers(topic string) ([]Subscriber, error) {
rows, err := s.db.Query(`SELECT callback, lease_seconds, secret, created FROM "subscribers" WHERE "topic" = $1`, topic)
if err != nil {
return nil, err
}
var subscribers []Subscriber
for rows.Next() {
var sub Subscriber
err := rows.Scan(&sub.Callback, &sub.LeaseSeconds, &sub.Secret, &sub.Created)
if err != nil {
return nil, err
}
subscribers = append(subscribers, sub)
}
return subscribers, nil
}