[messages] fix some issues

Author: capcom6

internal/sms-gateway/modules/messages/repository.go

@@ -136,7 +136,7 @@ func (r *Repository) Insert(message *Message) error {
 		return nil
 	}
 
-	if mysql.IsPrimaryKeyViolation(err) {
+	if errors.Is(err, gorm.ErrDuplicatedKey) || mysql.IsPrimaryKeyViolation(err) {
 		return ErrMessageAlreadyExists
 	}
 
@@ -159,7 +159,10 @@ func (r *Repository) UpdateState(message *Message) error {
 		}
 
 		for _, v := range message.Recipients {
-			if err := tx.Model(&v).Where("message_id = ?", message.ID).Select("State", "Error").Updates(&v).Error; err != nil {
+			if err := tx.Model(&MessageRecipient{}).
+				Where("message_id = ? AND phone_number = ?", message.ID, v.PhoneNumber).
+				Select("state", "error").
+				Updates(map[string]any{"state": v.State, "error": v.Error}).Error; err != nil {
 				return err
 			}
 		}

pkg/mysql/errors.go

@@ -1,10 +1,15 @@
 package mysql
 
-import "github.com/go-sql-driver/mysql"
+import (
+	"errors"
+
+	"github.com/go-sql-driver/mysql"
+)
 
 func IsPrimaryKeyViolation(err error) bool {
-	if mysqlErr, ok := err.(*mysql.MySQLError); ok {
-		return mysqlErr.Number == 1062
+	var me *mysql.MySQLError
+	if errors.As(err, &me) {
+		return me.Number == 1062
 	}
 	return false
 }