9c39c8cbd7
Go workspace (go.work) with 5 microservices + shared library: - gateway (8080), auth-service (8081), purchaser-service (8082) - textile-service (8083), washing-service (8084) - shared: proto definitions, common packages Infrastructure: docker-compose for local dev, K8s manifests for K3s cluster deployment (mysql/redis/etcd + traefik ingress). Frontend (iloom-flatten) added as git submodule. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
55 lines
1.4 KiB
Go
55 lines
1.4 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/muyuqingfeng/iloom/auth-service/internal/model"
|
|
)
|
|
|
|
type NotificationRepo struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func NewNotificationRepo(db *sql.DB) *NotificationRepo {
|
|
return &NotificationRepo{db: db}
|
|
}
|
|
|
|
func (r *NotificationRepo) ListByRecipient(ctx context.Context, userID string) ([]model.Notification, error) {
|
|
query := `SELECT id, recipient_id, sender_id, sender_company_id, title, content, type, plan_id, is_read, read_at, created_at
|
|
FROM ilm_notification WHERE recipient_id = ? ORDER BY created_at DESC`
|
|
rows, err := r.db.QueryContext(ctx, query, userID)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("list notifications: %w", err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
var notifications []model.Notification
|
|
for rows.Next() {
|
|
var n model.Notification
|
|
if err := rows.Scan(
|
|
&n.ID, &n.RecipientID, &n.SenderID, &n.SenderCompanyID,
|
|
&n.Title, &n.Content, &n.Type, &n.PlanID,
|
|
&n.IsRead, &n.ReadAt, &n.CreatedAt,
|
|
); err != nil {
|
|
return nil, fmt.Errorf("scan notification: %w", err)
|
|
}
|
|
notifications = append(notifications, n)
|
|
}
|
|
return notifications, nil
|
|
}
|
|
|
|
func (r *NotificationRepo) MarkRead(ctx context.Context, id, userID string) error {
|
|
now := time.Now()
|
|
_, err := r.db.ExecContext(ctx,
|
|
`UPDATE ilm_notification SET is_read = true, read_at = ? WHERE id = ? AND recipient_id = ?`,
|
|
now, id, userID,
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("mark notification read: %w", err)
|
|
}
|
|
return nil
|
|
}
|