plugins/input/skywalkingv2/service_name_discovery_handle.go

// Copyright 2021 iLogtail Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http:www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package skywalkingv2

import (
	"context"

	"github.com/alibaba/ilogtail/plugins/input/skywalkingv2/skywalking/apm/network/language/agent"
)

type ServiceNameDiscoveryHandle struct {
	RegistryInformationCache
}

func (s ServiceNameDiscoveryHandle) Discovery(ctx context.Context, collection *agent.ServiceNameCollection) (*agent.ServiceNameMappingCollection, error) {
	defer panicRecover()
	var elements []*agent.ServiceNameMappingElement
	for _, element := range collection.Elements {
		if endpoint := s.RegistryInformationCache.registryEndpoint(element.GetApplicationId(), element.GetServiceName()); endpoint != nil {
			elements = append(elements, &agent.ServiceNameMappingElement{
				ServiceId: endpoint.id,
				Element: &agent.ServiceNameElement{
					ServiceName:   element.GetServiceName(),
					ApplicationId: element.GetApplicationId(),
				},
			})
		}
	}

	return &agent.ServiceNameMappingCollection{Elements: elements}, nil
}