internal/dao/register.go

@@ -7,7 +7,6 @@ import (
 	"exhibition-register/pkg/app"
 	"exhibition-register/pkg/msg"
 	"fmt"
-
 	"go.uber.org/zap"
 	"gorm.io/gorm"
 )
@@ -80,12 +79,12 @@ func RecordList(in *exhibition.RecordListReq) (data []*model.RegisterRecord, tot
 	if total == 0 {
 		return
 	}
-	sortBy := "create_at %s,id  %s"
+	sortBy := "created_at desc"
 	if in.SortKey == 1 {
-		sortBy = fmt.Sprintf(sortBy, "ASC", "ASC")
+		sortBy = fmt.Sprintf("created_at%s%s", " ", in.Sort)
 	}
 	if in.SortKey == 2 {
-		sortBy = fmt.Sprintf(sortBy, "DESC", "DESC")
+		sortBy = fmt.Sprintf("updated_at%s%s", " ", in.Sort)
 	}
 
 	if err = dbLink.Order(sortBy).Limit(int(in.PageSize)).Offset(int(offset)).Find(&data).Error; err != nil {