samples/Apache.IoTDB.Samples/Program.cs

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

using Microsoft.Extensions.Logging;
using NLog.Extensions.Logging;
using System.Threading.Tasks;
using System.CommandLine;
using System.Collections.Generic;
using System;

namespace Apache.IoTDB.Samples
{
    public static class Program
    {
        public static async Task Main(string[] args)
        {
            var singleOption = new Option<string>(
            "--single",
            () => "localhost",
            description: "Use single endpoint (e.g. --single localhost)");

        var multiOption = new Option<List<string>>(
            "--multi",
            description: "Use multiple endpoints (e.g. --multi localhost:6667 localhost:6668)")
        {
            AllowMultipleArgumentsPerToken = true
        };

        var rootCommand = new RootCommand
        {
            singleOption,
            multiOption
        };

        rootCommand.SetHandler(async (string single, List<string> multi) =>
        {
            var utilsTest = new UtilsTest();
            utilsTest.TestParseEndPoint();

            SessionPoolTest sessionPoolTest;

            if (!string.IsNullOrEmpty(single) && (multi == null || multi.Count == 0))
            {
                sessionPoolTest = new SessionPoolTest(single);
            }
            else if (multi != null && multi.Count != 0)
            {
                sessionPoolTest = new SessionPoolTest(multi);
            }
            else
            {
                Console.WriteLine("Please specify either --single or --multi endpoints.");
                return;
            }

            await sessionPoolTest.Test();


        }, singleOption, multiOption);

        await rootCommand.InvokeAsync(args);
        }

        public static void OpenDebugMode(this SessionPool session)
        {
            session.OpenDebugMode(builder =>
            {
                builder.AddConsole();
                builder.AddNLog();
            });
        }
    }
}