fix(all): use v2 module

This commit is contained in:
Yasuhiro Matsumoto
2023-05-08 23:55:48 +09:00
committed by fiatjaf_
parent 523b11d068
commit 2771d9b8ba
10 changed files with 140 additions and 110 deletions

View File

@@ -1,13 +1,14 @@
package main
import (
"context"
"encoding/json"
"fmt"
"log"
"time"
"github.com/fiatjaf/relayer"
"github.com/fiatjaf/relayer/storage/postgresql"
"github.com/fiatjaf/relayer/v2"
"github.com/fiatjaf/relayer/v2/storage/postgresql"
"github.com/kelseyhightower/envconfig"
"github.com/nbd-wtf/go-nostr"
)
@@ -22,12 +23,10 @@ func (r *Relay) Name() string {
return "BasicRelay"
}
func (r *Relay) Storage() relayer.Storage {
func (r *Relay) Storage(ctx context.Context) relayer.Storage {
return r.storage
}
func (r *Relay) OnInitialized(*relayer.Server) {}
func (r *Relay) Init() error {
err := envconfig.Process("", r)
if err != nil {
@@ -36,7 +35,7 @@ func (r *Relay) Init() error {
// every hour, delete all very old events
go func() {
db := r.Storage().(*postgresql.PostgresBackend)
db := r.Storage(context.TODO()).(*postgresql.PostgresBackend)
for {
time.Sleep(60 * time.Minute)
@@ -47,7 +46,7 @@ func (r *Relay) Init() error {
return nil
}
func (r *Relay) AcceptEvent(evt *nostr.Event) bool {
func (r *Relay) AcceptEvent(ctx context.Context, evt *nostr.Event) bool {
// block events that are too large
jsonb, _ := json.Marshal(evt)
if len(jsonb) > 10000 {
@@ -64,7 +63,11 @@ func main() {
return
}
r.storage = &postgresql.PostgresBackend{DatabaseURL: r.PostgresDatabase}
if err := relayer.Start(&r); err != nil {
server, err := relayer.NewServer(&r)
if err != nil {
log.Fatalf("failed to create server: %v", err)
}
if err := server.Start("0.0.0.0", 7447); err != nil {
log.Fatalf("server terminated: %v", err)
}
}