SimpleTopicExample.java

package tech.ydb.examples.topic;

import tech.ydb.auth.iam.CloudAuthHelper;
import tech.ydb.core.grpc.GrpcTransport;


/**
 * @author Nikolay Perfilov
 */
public abstract class SimpleTopicExample {
    protected static final String TOPIC_NAME = System.getenv("YDB_TOPIC_NAME");
    protected static final String CONSUMER_NAME = System.getenv("YDB_CONSUMER_NAME");

    protected void doMain(String[] args) {
        if (args.length > 1) {
            System.err.println("Too many arguments");
            return;
        }
        String connString;
        if (args.length == 1) {
            connString = args[0];
        } else {
            System.err.println("Pass <connection-string> as an argument. " +
                    "Example: some.host.name.com:2135?database=/Root\n");
            return;
        }

        System.out.println("connection-string: " + connString + "\n");

        try (GrpcTransport transport = GrpcTransport.forConnectionString(connString)
                .withAuthProvider(CloudAuthHelper.getAuthProviderFromEnviron())
                .build()) {
            run(transport);
        } catch (Throwable t) {
            t.printStackTrace();
        }
    }

    protected abstract void run(GrpcTransport transport);
}