src/main/scala/org/apache/nlpcraft/model/opencensus/NCStackdriverTraceExporter.java

/*
 * 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.
 */

package org.apache.nlpcraft.model.opencensus;

import io.opencensus.exporter.trace.stackdriver.*;
import org.apache.nlpcraft.common.*;
import org.apache.nlpcraft.common.config.*;
import org.apache.nlpcraft.model.*;
import java.io.*;

/**
 * Probe lifecycle component that manages OpenCensus <a target=_ href="https://cloud.google.com/stackdriver/">Stackdriver</a> trace exporter. See
 * <a target=_ href="https://opencensus.io/exporters/supported-exporters/java/stackdriver-trace/">https://opencensus.io/exporters/supported-exporters/java/stackdriver-trace/</a>
 * for more details. This is a probe lifecycle component - see {@link NCLifecycle} for usage information.
 * <p>
 * Note that this exporter uses "always" sampling by default that is only suitable
 * for demo and development purposes. For production use you need to override {@link NCBaseTraceExporter#setSampling()}
 * method to provide a more efficient sampling strategy.
 * <p>
 * See <a target=_ href="https://nlpcraft.apache.org/server-and-probe.html">documentation</a> on how to configure probe life cycle
 * components, default values, etc.
 */
public class NCStackdriverTraceExporter extends NCBaseTraceExporter {
    private static class Config extends NCConfigurableJava {
        private static final String pre = "nlpcraft.probe.opencensus.stackdriver";

        private final String gpi = getString(pre + ".googleProjectId");
    }

    private static final Config cfg = new Config();

    /**
     * Creates new Stackdriver OpenCensus trace export lifecycle component.
     */
    public NCStackdriverTraceExporter() {
        super("Stackdriver");
    }

    @Override
    public void onInit() {
        try {
            StackdriverTraceExporter.createAndRegister(StackdriverTraceConfiguration.builder().setProjectId(cfg.gpi).build());
        }
        catch (IOException e) {
            throw new NCException(
                String.format("Stackdriver OpenCensus trace exporter cannot be registered for project: %s", cfg.gpi), e
            );
        }

        setSampling();

        log.info("Stackdriver OpenCensus trace exporter started for project: {}", cfg.gpi);
    }
}