commit f47c3cbfb00fcabc461fbf05f0cce3f33c4d9d52 · edavis.dev/bsky-feeds

+27 -22

cmd/mostliked/main.go

···

       6
        
       	"os"

     

       7
        
       	"os/signal"

     

       8
        
       	"syscall"

     

       0
        
       
     

       9
        
       

     

       10
        
       	"github.com/edavis/bsky-feeds/pkg/mostliked"

     

       11
        
       	"github.com/gorilla/websocket"

     

       12
        
       	_ "github.com/mattn/go-sqlite3"

     

       13
        
       )

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       14
        
       

     

       15
        
       func main() {

     

       16
       -
       	conn, _, err := websocket.DefaultDialer.Dial("ws://localhost:6008/subscribe?wantedCollections=app.bsky.feed.post&wantedCollections=app.bsky.feed.like", nil)

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       17
        
       	if err != nil {

     

       18
       -
       		log.Fatal("websocket connection error:", err)

     

       19
        
       	}

     

       20
       -
       	defer conn.Close()

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       21
        
       

     

       22
        
       	dbCnx, err := sql.Open("sqlite3", "data/mostliked.db?_journal=WAL&_fk=on")

     

       23
        
       	if err != nil {

     

       24
       -
       		log.Fatal("error opening db")

     

       25
        
       	}

     

       26
       -
       	defer dbCnx.Close()

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       27
        
       

     

       28
        
       	jetstreamEvents := make(chan []byte)

     

       29
       -
       	go mostliked.Handler(jetstreamEvents, dbCnx)

     

       30
       -
       

     

       31
       -
       	signalChan := make(chan os.Signal, 1)

     

       32
       -
       	cleanupDone := make(chan struct{})

     

       33
       -
       	signal.Notify(signalChan, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)

     

       34
       -
       

     

       35
       -
       	go func(conn *websocket.Conn, dbCnx *sql.DB, jetstreamEvents chan []byte) {

     

       36
       -
       		<-signalChan

     

       37
       -
       		log.Println("shutting down...")

     

       38
       -
       		conn.Close()

     

       39
       -
       		dbCnx.Close()

     

       40
       -
       		close(cleanupDone)

     

       41
       -
       	}(conn, dbCnx, jetstreamEvents)

     

       42
        
       

     

       43
        
       	log.Println("starting up")

     

       44
       -
       	go func(conn *websocket.Conn) {

     

       45
        
       		for {

     

       46
        
       			_, message, err := conn.ReadMessage()

     

       47
        
       			if err != nil {

     

       48
       -
       				log.Println("error reading from websocket:", err)

     

       49
        
       				break

     

       50
        
       			}

     

       51
        
       			jetstreamEvents <- message

     

       52
        
       		}

     

       53
       -
       	}(conn)

     

       54
        
       

     

       55
       -
       	<-cleanupDone

     

       0
        
       
     

       56
        
       }

···

       6
        
       	"os"

     

       7
        
       	"os/signal"

     

       8
        
       	"syscall"

     

       9
       +
       	"context"

     

       10
        
       

     

       11
        
       	"github.com/edavis/bsky-feeds/pkg/mostliked"

     

       12
        
       	"github.com/gorilla/websocket"

     

       13
        
       	_ "github.com/mattn/go-sqlite3"

     

       14
        
       )

     

       15
       +
       

     

       16
       +
       //const JetstreamUrl = `wss://jetstream1.us-west.bsky.network/subscribe?wantedCollections=app.bsky.feed.post&wantedCollections=app.bsky.feed.like&cursor=1728846514000000`

     

       17
       +
       const JetstreamUrl = `ws://localhost:6008/subscribe?wantedCollections=app.bsky.feed.post&wantedCollections=app.bsky.feed.like`

     

       18
        
       

     

       19
        
       func main() {

     

       20
       +
       	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGINT, syscall.SIGTERM)

     

       21
       +
       	defer stop()

     

       22
       +
       

     

       23
       +
       	conn, _, err := websocket.DefaultDialer.Dial(JetstreamUrl, nil)

     

       24
        
       	if err != nil {

     

       25
       +
       		log.Fatalf("failed to open websocket: %v\n", err)

     

       26
        
       	}

     

       27
       +
       	defer func() {

     

       28
       +
       		if err := conn.Close(); err != nil {

     

       29
       +
       			log.Printf("failed to close websocket: %v\n", err)

     

       30
       +
       		}

     

       31
       +
       		log.Println("websocket closed")

     

       32
       +
       	}()

     

       33
        
       

     

       34
        
       	dbCnx, err := sql.Open("sqlite3", "data/mostliked.db?_journal=WAL&_fk=on")

     

       35
        
       	if err != nil {

     

       36
       +
       		log.Fatalf("failed to open database: %v\n", err)

     

       37
        
       	}

     

       38
       +
       	defer func() {

     

       39
       +
       		if err := dbCnx.Close(); err != nil {

     

       40
       +
       			log.Printf("failed to close db: %v", err)

     

       41
       +
       		}

     

       42
       +
       		log.Println("db closed")

     

       43
       +
       	}()

     

       44
        
       

     

       45
        
       	jetstreamEvents := make(chan []byte)

     

       46
       +
       	go mostliked.Handler(ctx, jetstreamEvents, dbCnx)

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       47
        
       

     

       48
        
       	log.Println("starting up")

     

       49
       +
       	go func() {

     

       50
        
       		for {

     

       51
        
       			_, message, err := conn.ReadMessage()

     

       52
        
       			if err != nil {

     

       0
        
       
     

       53
        
       				break

     

       54
        
       			}

     

       55
        
       			jetstreamEvents <- message

     

       56
        
       		}

     

       57
       +
       	}()

     

       58
        
       

     

       59
       +
       	<-ctx.Done()

     

       60
       +
       	log.Println("shutting down")

     

       61
        
       }

+1 -1

pkg/mostliked/generator.go

···

       59
        
       	ctx := context.Background()

     

       60
        
       	dbCnx, err := sql.Open("sqlite3", "data/mostliked.db?_journal=WAL&_fk=on&mode=ro")

     

       61
        
       	if err != nil {

     

       62
       -
       		log.Fatal("error opening db")

     

       63
        
       	}

     

       64
        
       	defer dbCnx.Close()

     

       65

···

       59
        
       	ctx := context.Background()

     

       60
        
       	dbCnx, err := sql.Open("sqlite3", "data/mostliked.db?_journal=WAL&_fk=on&mode=ro")

     

       61
        
       	if err != nil {

     

       62
       +
       		log.Printf("error opening db: %v\n", err)

     

       63
        
       	}

     

       64
        
       	defer dbCnx.Close()

     

       65

+7 -9

pkg/mostliked/handler.go

···

       62
        
       	return ""

     

       63
        
       }

     

       64
        
       

     

       65
       -
       func Handler(events <-chan []byte, dbCnx *sql.DB) {

     

       66
       -
       	ctx := context.Background()

     

       67
       -
       

     

       68
        
       	if _, err := dbCnx.ExecContext(ctx, ddl); err != nil {

     

       69
       -
       		log.Fatal("couldn't create tables")

     

       70
        
       	}

     

       71
        
       	queries := db.New(dbCnx)

     

       72
        
       

     
···

       100
        
       		if !txOpen {

     

       101
        
       			dbTx, err = dbCnx.BeginTx(ctx, nil)

     

       102
        
       			if err != nil {

     

       103
       -
       				log.Fatal(err)

     

       104
        
       			}

     

       105
        
       			txOpen = true

     

       106
        
       			queriesTx = queries.WithTx(dbTx)

     
···

       158
        
       				continue

     

       159
        
       			}

     

       160
        
       			drafts.Delete(like.Subject.Uri)

     

       161
       -
       			log.Println("storing", like.Subject.Uri, "in database")

     

       162
        
       			err := queriesTx.InsertPost(ctx, db.InsertPostParams{

     

       163
        
       				Uri:      like.Subject.Uri,

     

       164
        
       				CreateTs: draftPost.Created,

     
···

       184
        
       		}

     

       185
        
       

     

       186
        
       		eventCount += 1

     

       187
       -
       		if eventCount % 500 == 0 {

     

       188
        
       			if err := dbTx.Commit(); err != nil {

     

       189
       -
       				log.Fatalf("commit failed: %v\n", err)

     

       190
        
       			} else {

     

       191
        
       				txOpen = false

     

       192
        
       			}

     

       193
       -
       			log.Println("db committed")

     

       194
        
       		}

     

       195
        
       	}

     

       196
        
       }

···

       62
        
       	return ""

     

       63
        
       }

     

       64
        
       

     

       65
       +
       func Handler(ctx context.Context, events <-chan []byte, dbCnx *sql.DB) {

     

       0
        
       
     

       0
        
       
     

       66
        
       	if _, err := dbCnx.ExecContext(ctx, ddl); err != nil {

     

       67
       +
       		log.Printf("couldn't create tables: %v\n", err)

     

       68
        
       	}

     

       69
        
       	queries := db.New(dbCnx)

     

       70
        
       

     
···

       98
        
       		if !txOpen {

     

       99
        
       			dbTx, err = dbCnx.BeginTx(ctx, nil)

     

       100
        
       			if err != nil {

     

       101
       +
       				log.Printf("failed to begin transaction: %v\n", err)

     

       102
        
       			}

     

       103
        
       			txOpen = true

     

       104
        
       			queriesTx = queries.WithTx(dbTx)

     
···

       156
        
       				continue

     

       157
        
       			}

     

       158
        
       			drafts.Delete(like.Subject.Uri)

     

       159
       +
       			// log.Println("storing", like.Subject.Uri, "in database")

     

       160
        
       			err := queriesTx.InsertPost(ctx, db.InsertPostParams{

     

       161
        
       				Uri:      like.Subject.Uri,

     

       162
        
       				CreateTs: draftPost.Created,

     
···

       182
        
       		}

     

       183
        
       

     

       184
        
       		eventCount += 1

     

       185
       +
       		if eventCount % 1000 == 0 {

     

       186
        
       			if err := dbTx.Commit(); err != nil {

     

       187
       +
       				log.Printf("commit failed: %v\n", err)

     

       188
        
       			} else {

     

       189
        
       				txOpen = false

     

       190
        
       			}

     

       191
       +
       			// log.Println("db committed")

     

       192
        
       		}

     

       193
        
       	}

     

       194
        
       }