feat: 植物识别

service/plant/wiki_class.go

@@ -21,8 +21,10 @@ func (s *WikiClassService) AddClass(req plantReq.CreateWikiClass) error {
 	if !errors.Is(global.DB.Where("name = ?", req.Name).First(&plant.Class{}).Error, gorm.ErrRecordNotFound) {
 		return errors.New("存在重复分类名称，请修改名称")
 	}
-	if errors.Is(global.DB.Where("id = ?", req.OssId).First(&system.Oss{}).Error, gorm.ErrRecordNotFound) {
-		return errors.New("不存在此图片")
+	if req.OssId != "" {
+		if errors.Is(global.DB.Where("id = ?", req.OssId).First(&system.Oss{}).Error, gorm.ErrRecordNotFound) {
+			return errors.New("不存在此图片")
+		}
 	}
 	return global.DB.Create(&plant.Class{
 		Name:  req.Name,
@@ -42,8 +44,8 @@ func (s *WikiClassService) UpdateClass(req plantReq.UpdateWikiClass) error {
 func (s *WikiClassService) ClassPage(req common.PageInfo) (list interface{}, total int64, err error) {
 	limit := req.PageSize
 	offset := req.PageSize * (req.Current - 1)
-	db := global.DB.Model(&plant.Class{}).Preload("Oss")
-	var classes []*plant.Class
+	db := global.DB.Model(&plant.Class{})
+	var classes []plant.Class
 	err = db.Count(&total).Error
 	if err != nil {
 		return
@@ -65,7 +67,7 @@ func (s *WikiClassService) DeleteClass(req common.IdsReq) error {
 // ClassList 列表
 func (s *WikiClassService) ClassList() (list interface{}, err error) {
 	var classes []plant.Class
-	err = global.DB.Order("created_at desc").Find(&classes).Error
+	err = global.DB.Preload("Oss").Order("created_at desc").Find(&classes).Error
 	if err != nil {
 		return
 	}