using System;
using System.Diagnostics;
using System.Threading.Tasks;
using Grpc.Net.Client;
using MyNoSqlServer.Abstractions;
using MyNoSqlServer.Grpc;
using MyNoSqlServer.GrpcDataWriter;
using ProtoBuf.Grpc.Client;

namespace MyNoSqlServerTestGrpcClient
{
  class Program
  {
        static void Main(string[] args)
        {
            TestGrpcContracts().Wait();

            Console.WriteLine("Done");
        }

        private static async Task TestGrpcContracts()
        {
            GrpcClientFactory.AllowUnencryptedHttp2 = true;
            
            var transportChannel = GrpcChannel
                .ForAddress("http://127.0.0.1:5124")
                .CreateGrpcService<IMyNoSqlTransportGrpcService>();

            var grpcWriter = new MyNoSqlGrpcDataWriter(transportChannel);

            grpcWriter.RegisterSupportedEntity<TestEntity>("testtbl");
  
            var sw = new Stopwatch();
            sw.Start();
            
            var transaction = grpcWriter.BeginTransaction();

            transaction.InsertOrReplaceEntity(new TestEntity
            {
                PartitionKey = "PK",
                RowKey = "RK",
                Value = "Value"
            });

            await transaction.CommitAsync();

            sw.Stop();
            
            Console.WriteLine($"Record is inserted in {sw.Elapsed}");
            
            
            sw.Reset();
            var i = 0;
            sw.Start();
            await foreach (var _ in grpcWriter.GetRowsAsync<TestEntity>())
            {
                i++;
            }
            sw.Stop();
            
            Console.WriteLine($"Got {i} records in {sw.Elapsed}");
        }
    }

    public class TestEntity : MyNoSqlDbEntity
    {
        public string Value { get; set; }
    }
}