commit a0840455a41fa5bffb8af182dedcfdb353608f61 · bretton.dev/coves

internal/core/users/interfaces.go

···

       8
       8
        
       	GetByDID(ctx context.Context, did string) (*User, error)

     

       9
       9
        
       	GetByHandle(ctx context.Context, handle string) (*User, error)

     

       10
       10
        
       	UpdateHandle(ctx context.Context, did, newHandle string) (*User, error)

     

       11
       11
       +
       	GetByDIDs(ctx context.Context, dids []string) (map[string]*User, error)

     

       11
       12
        
       }

     

       12
       13
        
       

     

       13
       14
        
       // UserService defines the interface for user business logic

+38

internal/db/postgres/user_repo.go

···

       6
       6
        
       	"database/sql"

     

       7
       7
        
       	"fmt"

     

       8
       8
        
       	"strings"

     

       9
       9
       +
       

     

       10
       10
       +
       	"github.com/lib/pq"

     

       9
       11
        
       )

     

       10
       12
        
       

     

       11
       13
        
       type postgresUserRepo struct {

     
···

       103
       105
        
       

     

       104
       106
        
       	return user, nil

     

       105
       107
        
       }

     

       108
       108
       +
       

     

       109
       109
       +
       // GetByDIDs retrieves multiple users by their DIDs in a single query

     

       110
       110
       +
       // Returns a map of DID -> User for efficient lookups

     

       111
       111
       +
       // Missing users are not included in the result map (no error for missing users)

     

       112
       112
       +
       func (r *postgresUserRepo) GetByDIDs(ctx context.Context, dids []string) (map[string]*users.User, error) {

     

       113
       113
       +
       	if len(dids) == 0 {

     

       114
       114
       +
       		return make(map[string]*users.User), nil

     

       115
       115
       +
       	}

     

       116
       116
       +
       

     

       117
       117
       +
       	// Build parameterized query with IN clause

     

       118
       118
       +
       	// Use ANY($1) for PostgreSQL array support with pq.Array() for type conversion

     

       119
       119
       +
       	query := `SELECT did, handle, pds_url, created_at, updated_at FROM users WHERE did = ANY($1)`

     

       120
       120
       +
       

     

       121
       121
       +
       	rows, err := r.db.QueryContext(ctx, query, pq.Array(dids))

     

       122
       122
       +
       	if err != nil {

     

       123
       123
       +
       		return nil, fmt.Errorf("failed to query users by DIDs: %w", err)

     

       124
       124
       +
       	}

     

       125
       125
       +
       	defer rows.Close()

     

       126
       126
       +
       

     

       127
       127
       +
       	// Build map of results

     

       128
       128
       +
       	result := make(map[string]*users.User, len(dids))

     

       129
       129
       +
       	for rows.Next() {

     

       130
       130
       +
       		user := &users.User{}

     

       131
       131
       +
       		err := rows.Scan(&user.DID, &user.Handle, &user.PDSURL, &user.CreatedAt, &user.UpdatedAt)

     

       132
       132
       +
       		if err != nil {

     

       133
       133
       +
       			return nil, fmt.Errorf("failed to scan user row: %w", err)

     

       134
       134
       +
       		}

     

       135
       135
       +
       		result[user.DID] = user

     

       136
       136
       +
       	}

     

       137
       137
       +
       

     

       138
       138
       +
       	if err = rows.Err(); err != nil {

     

       139
       139
       +
       		return nil, fmt.Errorf("error iterating user rows: %w", err)

     

       140
       140
       +
       	}

     

       141
       141
       +
       

     

       142
       142
       +
       	return result, nil

     

       143
       143
       +
       }