Function DeleteLibrary

api/model/library.go:356–417

DeleteLibrary 删除知识库

(eid int64, id int64)

// DeleteLibrary 删除知识库
func DeleteLibrary(eid int64, id int64) error {
	// 检查知识库是否存在
	library, err := GetLibraryByID(eid, id)
	if err != nil {
		return err
	}

	// 检查知识库下是否有文件
	files, err := GetFilesByLibraryID(eid, library.ID)
	if err != nil {
		return err
	}
	if len(files) > 0 {
		// return fmt.Errorf("cannot delete library with files")
		for _, file := range files {
			if err := DeleteFile(eid, file.ID); err != nil {
				// 忽略"record not found"错误，因为文件可能已被其他进程删除
				if !errors.Is(err, gorm.ErrRecordNotFound) {
					return err
				}
			}
		}
	}

	// 开启事务
	tx := DB.Begin()
	defer func() {
		if r := recover(); r != nil {
			tx.Rollback()
		}
	}()

	var entityIDs []int64
	if err := tx.Model(&EntityChunkRelation{}).
		Distinct("entity_id").
		Where("eid = ? AND library_id = ?", eid, id).
		Pluck("entity_id", &entityIDs).Error; err != nil {
		tx.Rollback()
		return err
	}
	if err := tx.Where("eid = ? AND library_id = ?", eid, id).Delete(&EntityChunkRelation{}).Error; err != nil {
		tx.Rollback()
		return err
	}

	// 删除知识库
	if err := tx.Where("eid = ? AND id = ?", eid, id).Delete(&Library{}).Error; err != nil {
		tx.Rollback()
		return err
	}

	if err := DeleteOrphanEntitiesByIDsWithDB(tx, eid, entityIDs); err != nil {
		tx.Rollback()
		return err
	}

	if err := tx.Commit().Error; err != nil {
		return err