migrate all built-in storage backends.

storage/sqlite3/delete.go

@@ -1,6 +1,8 @@
 package sqlite3
 
-func (b SQLite3Backend) DeleteEvent(id string, pubkey string) error {
-	_, err := b.DB.Exec("DELETE FROM event WHERE id = $1 AND pubkey = $2", id, pubkey)
+import "context"
+
+func (b SQLite3Backend) DeleteEvent(ctx context.Context, id string, pubkey string) error {
+	_, err := b.DB.ExecContext(ctx, "DELETE FROM event WHERE id = $1 AND pubkey = $2", id, pubkey)
 	return err
 }

storage/sqlite3/init.go

@@ -1,11 +1,14 @@
 package sqlite3
 
 import (
+	"github.com/fiatjaf/relayer/v2"
 	"github.com/jmoiron/sqlx"
 	"github.com/jmoiron/sqlx/reflectx"
 	_ "github.com/mattn/go-sqlite3"
 )
 
+var _ relayer.Storage = (*SQLite3Backend)(nil)
+
 func (b *SQLite3Backend) Init() error {
 	db, err := sqlx.Connect("sqlite3", b.DatabaseURL)
 	if err != nil {

storage/sqlite3/query.go

@@ -1,18 +1,20 @@
 package sqlite3
 
 import (
+	"context"
 	"database/sql"
 	"encoding/hex"
 	"errors"
 	"fmt"
 	"strconv"
 	"strings"
-	"time"
 
 	"github.com/nbd-wtf/go-nostr"
 )
 
-func (b SQLite3Backend) QueryEvents(filter *nostr.Filter) (events []nostr.Event, err error) {
+func (b SQLite3Backend) QueryEvents(ctx context.Context, filter *nostr.Filter) (ch chan *nostr.Event, err error) {
+	ch = make(chan *nostr.Event)
+
 	var conditions []string
 	var params []any
 
@@ -110,11 +112,11 @@ func (b SQLite3Backend) QueryEvents(filter *nostr.Filter) (events []nostr.Event,
 
 	if filter.Since != nil {
 		conditions = append(conditions, "created_at > ?")
-		params = append(params, filter.Since.Unix())
+		params = append(params, filter.Since)
 	}
 	if filter.Until != nil {
 		conditions = append(conditions, "created_at < ?")
-		params = append(params, filter.Until.Unix())
+		params = append(params, filter.Until)
 	}
 	if filter.Search != "" {
 		conditions = append(conditions, "content LIKE ?")
@@ -143,19 +145,21 @@ func (b SQLite3Backend) QueryEvents(filter *nostr.Filter) (events []nostr.Event,
 		return nil, fmt.Errorf("failed to fetch events using query %q: %w", query, err)
 	}
 
-	defer rows.Close()
-
-	for rows.Next() {
-		var evt nostr.Event
-		var timestamp int64
-		err := rows.Scan(&evt.ID, &evt.PubKey, &timestamp,
-			&evt.Kind, &evt.Tags, &evt.Content, &evt.Sig)
-		if err != nil {
-			return nil, fmt.Errorf("failed to scan row: %w", err)
+	go func() {
+		defer rows.Close()
+		defer close(ch)
+		for rows.Next() {
+			var evt nostr.Event
+			var timestamp int64
+			err := rows.Scan(&evt.ID, &evt.PubKey, &timestamp,
+				&evt.Kind, &evt.Tags, &evt.Content, &evt.Sig)
+			if err != nil {
+				return
+			}
+			evt.CreatedAt = nostr.Timestamp(timestamp)
+			ch <- &evt
 		}
-		evt.CreatedAt = time.Unix(timestamp, 0)
-		events = append(events, evt)
-	}
+	}()
 
-	return events, nil
+	return ch, nil
 }

storage/sqlite3/save.go

@@ -1,6 +1,7 @@
 package sqlite3
 
 import (
+	"context"
 	"encoding/json"
 	"fmt"
 
@@ -8,30 +9,30 @@ import (
 	"github.com/nbd-wtf/go-nostr"
 )
 
-func (b *SQLite3Backend) SaveEvent(evt *nostr.Event) error {
+func (b *SQLite3Backend) SaveEvent(ctx context.Context, evt *nostr.Event) error {
 	// react to different kinds of events
 	if evt.Kind == nostr.KindSetMetadata || evt.Kind == nostr.KindContactList || (10000 <= evt.Kind && evt.Kind < 20000) {
 		// delete past events from this user
-		b.DB.Exec(`DELETE FROM event WHERE pubkey = $1 AND kind = $2`, evt.PubKey, evt.Kind)
+		b.DB.ExecContext(ctx, `DELETE FROM event WHERE pubkey = $1 AND kind = $2`, evt.PubKey, evt.Kind)
 	} else if evt.Kind == nostr.KindRecommendServer {
 		// delete past recommend_server events equal to this one
-		b.DB.Exec(`DELETE FROM event WHERE pubkey = $1 AND kind = $2 AND content = $3`,
+		b.DB.ExecContext(ctx, `DELETE FROM event WHERE pubkey = $1 AND kind = $2 AND content = $3`,
 			evt.PubKey, evt.Kind, evt.Content)
 	} else if evt.Kind >= 30000 && evt.Kind < 40000 {
 		// NIP-33
 		d := evt.Tags.GetFirst([]string{"d"})
 		if d != nil {
 			tagsLike := fmt.Sprintf(`%%"d","%s"%%`, d.Value())
-			b.DB.Exec(`DELETE FROM event WHERE pubkey = $1 AND kind = $2 AND tags LIKE $3`, evt.PubKey, evt.Kind, tagsLike)
+			b.DB.ExecContext(ctx, `DELETE FROM event WHERE pubkey = $1 AND kind = $2 AND tags LIKE $3`, evt.PubKey, evt.Kind, tagsLike)
 		}
 	}
 
 	// insert
 	tagsj, _ := json.Marshal(evt.Tags)
-	res, err := b.DB.Exec(`
+	res, err := b.DB.ExecContext(ctx, `
         INSERT INTO event (id, pubkey, created_at, kind, tags, content, sig)
         VALUES ($1, $2, $3, $4, $5, $6, $7)
-    `, evt.ID, evt.PubKey, evt.CreatedAt.Unix(), evt.Kind, tagsj, evt.Content, evt.Sig)
+    `, evt.ID, evt.PubKey, evt.CreatedAt, evt.Kind, tagsj, evt.Content, evt.Sig)
 	if err != nil {
 		return err
 	}
@@ -48,7 +49,7 @@ func (b *SQLite3Backend) SaveEvent(evt *nostr.Event) error {
 	return nil
 }
 
-func (b *SQLite3Backend) BeforeSave(evt *nostr.Event) {
+func (b *SQLite3Backend) BeforeSave(ctx context.Context, evt *nostr.Event) {
 	// do nothing
 }