feat: initial commit

internal/store/follower.go

@@ -0,0 +1,46 @@
+package store
+
+import (
+	"context"
+	"database/sql"
+
+	"github.com/lib/pq"
+)
+
+type Follower struct { 
+	ID         int64  `json:"id"`
+	FolloweeID int64  `json:"followee_id"`
+	CreatedAt  string `json:"created_at"`
+}
+
+type FollowerStore struct {
+	db *sql.DB
+}
+
+func (s *FollowerStore) Follow(ctx context.Context, followeeID, followerID int64) error {
+	query := `INSERT INTO followers (user_id, follower_id) VALUES ($1, $2)`
+	ctx, cancel := context.WithTimeout(ctx, QueryTimeout)
+	defer cancel()
+
+	_, err := s.db.ExecContext(ctx, query, followerID, followeeID)
+	if err != nil {
+		if pqErr, ok := err.(*pq.Error); ok && pqErr.Code == "23505" {
+			return ErrConflict
+		}
+	}
+
+	return nil
+}
+
+func (s *FollowerStore) Unfollow(ctx context.Context, followeeID, followerID int64) error {
+	query := `DELETE FROM followers WHERE user_id = $1 AND follower_id = $2`
+	ctx, cancel := context.WithTimeout(ctx, QueryTimeout)
+	defer cancel()
+
+	_, err := s.db.ExecContext(ctx, query, followerID, followeeID)
+	if pqErr, ok := err.(*pq.Error); ok && pqErr.Code == "23505" {
+		return ErrNotFound
+	}
+
+	return nil
+}