Method UpsertEdge

modules/generator/processor/servicegraphs/store/store.go:101–141

UpsertEdge fetches an Edge from the store and updates it using the given callback. If the Edge doesn't exist yet, it creates a new one with the default TTL. If the Edge is complete after applying the callback, it's completed and removed.

(key string, side Side, update Callback)

// doesn't exist yet, it creates a new one with the default TTL.
// If the Edge is complete after applying the callback, it's completed and removed.
func (s *store) UpsertEdge(key string, side Side, update Callback) (isNew bool, err error) {
	s.mtx.Lock()
	defer s.mtx.Unlock()

	if storedEdge, ok := s.m[key]; ok {
		edge := storedEdge.Value.(*Edge)
		update(edge)

		if edge.isComplete() {
			s.onComplete(edge)
			s.deleteEdge(storedEdge)
		}

		return false, nil
	}

	if s.hasDroppedCounterpart(key, side) {
		return true, ErrDroppedSpanSide
	}

	edge := s.grabEdge(key)
	update(edge)

	if edge.isComplete() {
		s.onComplete(edge)
		s.returnEdge(edge)
		return true, nil
	}

	// Check we can add new edges
	if s.l.Len() >= s.maxItems {
		// todo: try to evict expired items
		s.returnEdge(edge)
		return false, ErrTooManyItems
	}

	ele := s.l.PushBack(edge)
	s.m[key] = ele

	return true, nil
}

// Expire evicts all expired items in the store.
func (s *store) Expire() {