migrate from github

internal/repository/notification_repo/mark_read.go

@@ -0,0 +1,69 @@
+package notification_repo
+
+import (
+	"github.com/XingfenD/yoresee_doc/internal/model"
+	"gorm.io/gorm"
+)
+
+type NotificationMarkReadOperation struct {
+	repo        *NotificationRepository
+	receiverID  int64
+	externalIDs []string
+	tx          *gorm.DB
+}
+
+func (r *NotificationRepository) MarkRead(receiverID int64, externalIDs []string) *NotificationMarkReadOperation {
+	return &NotificationMarkReadOperation{
+		repo:        r,
+		receiverID:  receiverID,
+		externalIDs: externalIDs,
+	}
+}
+
+func (op *NotificationMarkReadOperation) WithTx(tx *gorm.DB) *NotificationMarkReadOperation {
+	op.tx = tx
+	return op
+}
+
+func (op *NotificationMarkReadOperation) Exec() error {
+	if len(op.externalIDs) == 0 {
+		return nil
+	}
+	db := op.repo.db
+	if op.tx != nil {
+		db = op.tx
+	}
+	return db.Model(&model.Notification{}).
+		Where("receiver_id = ? AND external_id IN ?", op.receiverID, op.externalIDs).
+		Updates(map[string]any{"status": "read"}).
+		Error
+}
+
+type NotificationMarkAllReadOperation struct {
+	repo       *NotificationRepository
+	receiverID int64
+	tx         *gorm.DB
+}
+
+func (r *NotificationRepository) MarkAllRead(receiverID int64) *NotificationMarkAllReadOperation {
+	return &NotificationMarkAllReadOperation{
+		repo:       r,
+		receiverID: receiverID,
+	}
+}
+
+func (op *NotificationMarkAllReadOperation) WithTx(tx *gorm.DB) *NotificationMarkAllReadOperation {
+	op.tx = tx
+	return op
+}
+
+func (op *NotificationMarkAllReadOperation) Exec() error {
+	db := op.repo.db
+	if op.tx != nil {
+		db = op.tx
+	}
+	return db.Model(&model.Notification{}).
+		Where("receiver_id = ? AND status != ?", op.receiverID, "read").
+		Updates(map[string]any{"status": "read"}).
+		Error
+}