Function DeleteOrphanEntitiesByIDsWithDB

api/model/entity.go:197–228

(db *gorm.DB, eid int64, entityIDs []int64)

}

func DeleteOrphanEntitiesByIDsWithDB(db *gorm.DB, eid int64, entityIDs []int64) error {
	if db == nil {
		return errors.New("db is nil")
	}
	if eid <= 0 {
		return errors.New("eid is empty")
	}
	if len(entityIDs) == 0 {
		return nil
	}

	uniq := make(map[int64]struct{}, len(entityIDs))
	cleaned := make([]int64, 0, len(entityIDs))
	for _, id := range entityIDs {
		if id <= 0 {
			continue
		}
		if _, ok := uniq[id]; ok {
			continue
		}
		uniq[id] = struct{}{}
		cleaned = append(cleaned, id)
	}
	if len(cleaned) == 0 {
		return nil
	}

	return db.Model(&Entity{}).
		Where("eid = ? AND id IN ?", eid, cleaned).
		Where("NOT EXISTS (SELECT 1 FROM entity_chunk_relations WHERE entity_chunk_relations.eid = ? AND entity_chunk_relations.entity_id = entities.id)", eid).
		Delete(&Entity{}).Error
}

func GetEntityVectorCollectionName(eid int64) string {
	return fmt.Sprintf("entity_eid_%d", eid)