worker/src/reporter/mod.rs

// Licensed to the Apache Software Foundation (ASF) under one or more
// contributor license agreements.  See the NOTICE file distributed with
// this work for additional information regarding copyright ownership.
// The ASF licenses this file to You 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.

mod reporter_grpc;
mod reporter_kafka;

pub use reporter_grpc::GrpcReporterConfiguration;
#[cfg(feature = "kafka-reporter")]
pub use reporter_kafka::KafkaReporterConfiguration;
use skywalking::reporter::{CollectItemConsume, CollectItemProduce};

pub enum ReporterConfiguration {
    Grpc(GrpcReporterConfiguration),

    #[cfg(feature = "kafka-reporter")]
    Kafka(KafkaReporterConfiguration),
}

pub async fn run_reporter(
    config: ReporterConfiguration, producer: impl CollectItemProduce,
    consumer: impl CollectItemConsume,
) -> anyhow::Result<()> {
    match config {
        ReporterConfiguration::Grpc(config) => {
            reporter_grpc::run_reporter(config, producer, consumer).await
        }
        #[cfg(feature = "kafka-reporter")]
        ReporterConfiguration::Kafka(config) => {
            reporter_kafka::run_reporter(config, producer, consumer).await
        }
    }
}