back round 2 view raw

appview: db: follow: add GetFollowers and GetFollowing functions to fetch Follows #483

merged

opened by

ptr.pet 4 months ago targeting master from ptr.pet/core: followers-following-list

Signed-off-by: dusk y.bera003.06@protonmail.com

options

unified split

Changed files

+68 -41

appview

follow.go

timeline.go

+67 -40

appview/db/follow.go

···

       1
       1
        
       package db

     

       2
       2
        
       

     

       3
       3
        
       import (

     

       4
       4
       +
       	"fmt"

     

       4
       5
        
       	"log"

     

       6
       6
       +
       	"strings"

     

       5
       7
        
       	"time"

     

       6
       8
        
       )

     

       7
       9
        
       

     
···

       56
       58
        
       func GetFollowerFollowingCount(e Execer, did string) (int, int, error) {

     

       57
       59
        
       	followers, following := 0, 0

     

       58
       60
        
       	err := e.QueryRow(

     

       59
       59
       -
       		`SELECT 

     

       61
       61
       +
       		`SELECT

     

       60
       62
        
       		COUNT(CASE WHEN subject_did = ? THEN 1 END) AS followers,

     

       61
       63
        
       		COUNT(CASE WHEN user_did = ? THEN 1 END) AS following

     

       62
       64
        
       		FROM follows;`, did, did).Scan(&followers, &following)

     
···

       66
       68
        
       	return followers, following, nil

     

       67
       69
        
       }

     

       68
       70
        
       

     

       71
       71
       +
       func GetFollows(e Execer, limit int, filters ...filter) ([]Follow, error) {

     

       72
       72
       +
       	var follows []Follow

     

       73
       73
       +
       

     

       74
       74
       +
       	var conditions []string

     

       75
       75
       +
       	var args []any

     

       76
       76
       +
       	for _, filter := range filters {

     

       77
       77
       +
       		conditions = append(conditions, filter.Condition())

     

       78
       78
       +
       		args = append(args, filter.Arg()...)

     

       79
       79
       +
       	}

     

       80
       80
       +
       

     

       81
       81
       +
       	whereClause := ""

     

       82
       82
       +
       	if conditions != nil {

     

       83
       83
       +
       		whereClause = " where " + strings.Join(conditions, " and ")

     

       84
       84
       +
       	}

     

       85
       85
       +
       	limitClause := ""

     

       86
       86
       +
       	if limit > 0 {

     

       87
       87
       +
       		limitClause = " limit ?"

     

       88
       88
       +
       		args = append(args, limit)

     

       89
       89
       +
       	}

     

       90
       90
       +
       

     

       91
       91
       +
       	query := fmt.Sprintf(

     

       92
       92
       +
       		`select user_did, subject_did, followed_at, rkey

     

       93
       93
       +
       		from follows

     

       94
       94
       +
       		%s

     

       95
       95
       +
       		order by followed_at desc

     

       96
       96
       +
       		%s

     

       97
       97
       +
       	`, whereClause, limitClause)

     

       98
       98
       +
       

     

       99
       99
       +
       	rows, err := e.Query(query, args...)

     

       100
       100
       +
       	if err != nil {

     

       101
       101
       +
       		return nil, err

     

       102
       102
       +
       	}

     

       103
       103
       +
       	for rows.Next() {

     

       104
       104
       +
       		var follow Follow

     

       105
       105
       +
       		var followedAt string

     

       106
       106
       +
       		err := rows.Scan(

     

       107
       107
       +
       			&follow.UserDid,

     

       108
       108
       +
       			&follow.SubjectDid,

     

       109
       109
       +
       			&followedAt,

     

       110
       110
       +
       			&follow.Rkey,

     

       111
       111
       +
       		)

     

       112
       112
       +
       		if err != nil {

     

       113
       113
       +
       			return nil, err

     

       114
       114
       +
       		}

     

       115
       115
       +
       		followedAtTime, err := time.Parse(time.RFC3339, followedAt)

     

       116
       116
       +
       		if err != nil {

     

       117
       117
       +
       			log.Println("unable to determine followed at time")

     

       118
       118
       +
       			follow.FollowedAt = time.Now()

     

       119
       119
       +
       		} else {

     

       120
       120
       +
       			follow.FollowedAt = followedAtTime

     

       121
       121
       +
       		}

     

       122
       122
       +
       		follows = append(follows, follow)

     

       123
       123
       +
       	}

     

       124
       124
       +
       	return follows, nil

     

       125
       125
       +
       }

     

       126
       126
       +
       

     

       127
       127
       +
       func GetFollowers(e Execer, did string) ([]Follow, error) {

     

       128
       128
       +
       	return GetFollows(e, 0, FilterEq("subject_did", did))

     

       129
       129
       +
       }

     

       130
       130
       +
       

     

       131
       131
       +
       func GetFollowing(e Execer, did string) ([]Follow, error) {

     

       132
       132
       +
       	return GetFollows(e, 0, FilterEq("user_did", did))

     

       133
       133
       +
       }

     

       134
       134
       +
       

     

       69
       135
        
       type FollowStatus int

     

       70
       136
        
       

     

       71
       137
        
       const (

     
···

       96
       162
        
       		return IsFollowing

     

       97
       163
        
       	}

     

       98
       164
        
       }

     

       99
       99
       -
       

     

       100
       100
       -
       func GetAllFollows(e Execer, limit int) ([]Follow, error) {

     

       101
       101
       -
       	var follows []Follow

     

       102
       102
       -
       

     

       103
       103
       -
       	rows, err := e.Query(`

     

       104
       104
       -
       		select user_did, subject_did, followed_at, rkey

     

       105
       105
       -
       		from follows

     

       106
       106
       -
       		order by followed_at desc

     

       107
       107
       -
       		limit ?`, limit,

     

       108
       108
       -
       	)

     

       109
       109
       -
       	if err != nil {

     

       110
       110
       -
       		return nil, err

     

       111
       111
       -
       	}

     

       112
       112
       -
       	defer rows.Close()

     

       113
       113
       -
       

     

       114
       114
       -
       	for rows.Next() {

     

       115
       115
       -
       		var follow Follow

     

       116
       116
       -
       		var followedAt string

     

       117
       117
       -
       		if err := rows.Scan(&follow.UserDid, &follow.SubjectDid, &followedAt, &follow.Rkey); err != nil {

     

       118
       118
       -
       			return nil, err

     

       119
       119
       -
       		}

     

       120
       120
       -
       

     

       121
       121
       -
       		followedAtTime, err := time.Parse(time.RFC3339, followedAt)

     

       122
       122
       -
       		if err != nil {

     

       123
       123
       -
       			log.Println("unable to determine followed at time")

     

       124
       124
       -
       			follow.FollowedAt = time.Now()

     

       125
       125
       -
       		} else {

     

       126
       126
       -
       			follow.FollowedAt = followedAtTime

     

       127
       127
       -
       		}

     

       128
       128
       -
       

     

       129
       129
       -
       		follows = append(follows, follow)

     

       130
       130
       -
       	}

     

       131
       131
       -
       

     

       132
       132
       -
       	if err := rows.Err(); err != nil {

     

       133
       133
       -
       		return nil, err

     

       134
       134
       -
       	}

     

       135
       135
       -
       

     

       136
       136
       -
       	return follows, nil

     

       137
       137
       -
       }

+1 -1

appview/db/timeline.go

···

       137
       137
        
       }

     

       138
       138
        
       

     

       139
       139
        
       func getTimelineFollows(e Execer) ([]TimelineEvent, error) {

     

       140
       140
       -
       	follows, err := GetAllFollows(e, Limit)

     

       140
       140
       +
       	follows, err := GetFollows(e, Limit)

     

       141
       141
        
       	if err != nil {

     

       142
       142
        
       		return nil, err

     

       143
       143
        
       	}