commit 276ced5c69574ce36d69f6c7abe039aecb818572 · edavis.dev/bsky-feeds

+36 -8

cmd/mostliked/main.go

···

       1
       1
        
       package main

     

       2
       2
        
       

     

       3
       3
        
       import (

     

       4
       4
       +
       	"database/sql"

     

       4
       5
        
       	"log"

     

       6
       6
       +
       	"os"

     

       7
       7
       +
       	"os/signal"

     

       8
       8
       +
       	"syscall"

     

       5
       9
        
       

     

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

     

       7
       11
        
       	"github.com/gorilla/websocket"

     

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

     

       8
       13
        
       )

     

       9
       14
        
       

     

       10
       15
        
       func main() {

     
···

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

     

       14
       19
        
       	}

     

       15
       20
        
       	defer conn.Close()

     

       21
       21
       +
       

     

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

     

       23
       23
       +
       	if err != nil {

     

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

     

       25
       25
       +
       	}

     

       26
       26
       +
       	defer dbCnx.Close()

     

       16
       27
        
       

     

       17
       28
        
       	jetstreamEvents := make(chan []byte)

     

       18
       18
       -
       	go mostliked.Handler(jetstreamEvents)

     

       29
       29
       +
       	go mostliked.Handler(jetstreamEvents, dbCnx)

     

       30
       30
       +
       

     

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

     

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

     

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

     

       34
       34
       +
       

     

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

     

       36
       36
       +
       		<-signalChan

     

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

     

       38
       38
       +
       		close(jetstreamEvents)

     

       39
       39
       +
       		dbCnx.Close()

     

       40
       40
       +
       		conn.Close()

     

       41
       41
       +
       		close(cleanupDone)

     

       42
       42
       +
       	}(conn, dbCnx, jetstreamEvents)

     

       19
       43
        
       

     

       20
       44
        
       	log.Println("starting up")

     

       21
       21
       -
       	for {

     

       22
       22
       -
       		_, message, err := conn.ReadMessage()

     

       23
       23
       -
       		if err != nil {

     

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

     

       25
       25
       -
       			break

     

       45
       45
       +
       	go func(conn *websocket.Conn) {

     

       46
       46
       +
       		for {

     

       47
       47
       +
       			_, message, err := conn.ReadMessage()

     

       48
       48
       +
       			if err != nil {

     

       49
       49
       +
       				log.Println("error reading from websocket:", err)

     

       50
       50
       +
       				break

     

       51
       51
       +
       			}

     

       52
       52
       +
       			jetstreamEvents <- message

     

       26
       53
        
       		}

     

       27
       27
       -
       		jetstreamEvents <- message

     

       28
       28
       -
       	}

     

       54
       54
       +
       	}(conn)

     

       55
       55
       +
       

     

       56
       56
       +
       	<-cleanupDone

     

       29
       57
        
       }

+1 -7

pkg/mostliked/handler.go

···

       61
       61
        
       	return ""

     

       62
       62
        
       }

     

       63
       63
        
       

     

       64
       64
       -
       func Handler(events <-chan []byte) {

     

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

     

       65
       65
        
       	ctx := context.Background()

     

       66
       66
        
       

     

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

     

       68
       68
       -
       	if err != nil {

     

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

     

       70
       70
       -
       	}

     

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

     

       72
       68
        
       		log.Fatal("couldn't create tables")

     

       73
       69
        
       	}

     

       74
       74
       -
       	defer dbCnx.Close()

     

       75
       75
       -
       

     

       76
       70
        
       	queries := db.New(dbCnx)

     

       77
       71
        
       

     

       78
       72
        
       	drafts := ccache.New(ccache.Configure[DraftPost]().MaxSize(50_000).GetsPerPromote(1))