finx mongodb

Signed-off-by: Login Victor <batazor111@gmail.com>
shortlink-org · Aug 7, 2020 · 684393b · 684393b
1 parent 8360d2d
commit 684393b
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 2 deletions.
diff --git a/internal/store/mongo/mongo.go b/internal/store/mongo/mongo.go
@@ -244,7 +244,12 @@ func (m *MongoLinkList) singleWrite(ctx context.Context, source *link.Link) (*li
 
 	_, err = collection.InsertOne(ctx, &data)
 	if err != nil {
-		return nil, &link.NotFoundError{Link: data, Err: fmt.Errorf("Failed marsharing link: %s", data.Url)}
+		switch err.(mongo.WriteException).WriteErrors[0].Code {
+		case 11000:
+			return nil, &link.NotUniqError{Link: data, Err: fmt.Errorf("Duplicate URL: %s", data.Url)}
+		default:
+			return nil, &link.NotFoundError{Link: data, Err: fmt.Errorf("Failed marsharing link: %s", data.Url)}
+		}
 	}
 
 	return data, nil
@@ -270,7 +275,12 @@ func (m *MongoLinkList) batchWrite(ctx context.Context, sources []*link.Link) ([
 
 	_, err := collection.InsertMany(ctx, docs)
 	if err != nil {
-		return nil, &link.NotFoundError{Link: sources[0], Err: fmt.Errorf("Failed marsharing link: %s", sources[0].Url)}
+		switch err.(mongo.WriteException).WriteErrors[0].Code {
+		case 11000:
+			return nil, &link.NotUniqError{Link: sources[0], Err: fmt.Errorf("Duplicate URL: %s", sources[0].Url)}
+		default:
+			return nil, &link.NotFoundError{Link: sources[0], Err: fmt.Errorf("Failed marsharing link: %s", sources[0].Url)}
+		}
 	}
 
 	return sources, nil

diff --git a/pkg/link/errors.go b/pkg/link/errors.go
@@ -13,3 +13,12 @@ type NotFoundError struct { // nolint unused
 func (e *NotFoundError) Error() string {
 	return fmt.Sprintf("Not found link: %s", e.Link.Url)
 }
+
+type NotUniqError struct { // nolint unused
+	Link *Link
+	Err  error
+}
+
+func (e *NotUniqError) Error() string {
+	return fmt.Sprintf("Not uniq link: %s", e.Link.Url)
+}