update short id

internal/controller/collection_controller.go

@@ -7,6 +7,7 @@ import (
 	"github.com/answerdev/answer/internal/schema"
 	"github.com/answerdev/answer/internal/service"
 	"github.com/answerdev/answer/pkg/converter"
+	"github.com/answerdev/answer/pkg/uid"
 	"github.com/gin-gonic/gin"
 	"github.com/jinzhu/copier"
 	"github.com/segmentfault/pacman/errors"
@@ -37,6 +38,7 @@ func (cc *CollectionController) CollectionSwitch(ctx *gin.Context) {
 	if handler.BindAndCheck(ctx, req) {
 		return
 	}
+	req.ObjectID = uid.DeShortID(req.ObjectID)
 
 	dto := &schema.CollectionSwitchDTO{}
 	_ = copier.Copy(dto, req)

internal/controller/follow_controller.go

@@ -5,6 +5,7 @@ import (
 	"github.com/answerdev/answer/internal/base/middleware"
 	"github.com/answerdev/answer/internal/schema"
 	"github.com/answerdev/answer/internal/service/follow"
+	"github.com/answerdev/answer/pkg/uid"
 	"github.com/gin-gonic/gin"
 	"github.com/jinzhu/copier"
 )
@@ -34,7 +35,7 @@ func (fc *FollowController) Follow(ctx *gin.Context) {
 	if handler.BindAndCheck(ctx, req) {
 		return
 	}
-
+	req.ObjectID = uid.DeShortID(req.ObjectID)
 	dto := &schema.FollowDTO{}
 	_ = copier.Copy(dto, req)
 	dto.UserID = middleware.GetLoginUserIDFromContext(ctx)

internal/controller/report_controller.go

@@ -8,6 +8,7 @@ import (
 	"github.com/answerdev/answer/internal/service/permission"
 	"github.com/answerdev/answer/internal/service/rank"
 	"github.com/answerdev/answer/internal/service/report"
+	"github.com/answerdev/answer/pkg/uid"
 	"github.com/gin-gonic/gin"
 	"github.com/segmentfault/pacman/errors"
 )
@@ -39,7 +40,7 @@ func (rc *ReportController) AddReport(ctx *gin.Context) {
 	if handler.BindAndCheck(ctx, req) {
 		return
 	}
-
+	req.ObjectID = uid.DeShortID(req.ObjectID)
 	req.UserID = middleware.GetLoginUserIDFromContext(ctx)
 	can, err := rc.rankService.CheckOperationPermission(ctx, req.UserID, permission.ReportAdd, "")
 	if err != nil {

internal/repo/collection/collection_repo.go

@@ -30,6 +30,7 @@ func NewCollectionRepo(data *data.Data, uniqueIDRepo unique.UniqueIDRepo) collec
 
 // AddCollection add collection
 func (cr *collectionRepo) AddCollection(ctx context.Context, collection *entity.Collection) (err error) {
+	needAdd := false
 	_, err = cr.data.DB.Transaction(func(session *xorm.Session) (result any, err error) {
 		var has bool
 		dbcollection := &entity.Collection{}
@@ -41,16 +42,23 @@ func (cr *collectionRepo) AddCollection(ctx context.Context, collection *entity.
 		if has {
 			return
 		}
+		needAdd = true
+		return
+	})
+	if err != nil {
+		return
+	}
+	if needAdd {
 		id, err := cr.uniqueIDRepo.GenUniqueIDStr(ctx, collection.TableName())
 		if err == nil {
 			collection.ID = id
-			_, err = session.Insert(collection)
+			_, err = cr.data.DB.Insert(collection)
 			if err != nil {
-				return result, errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
+				return errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
 			}
 		}
-		return
-	})
+	}
+
 	return err
 }