public CosmosDbLivySessionPool(ILivyClient client, CosmosCollectionSettings cosmosCollectionSettings, LivySessionConfiguration livySessionConfiguration, int maxSessions)
{
_client = client;
_cosmosCollectionSettings = cosmosCollectionSettings;
_livySessionConfiguration = livySessionConfiguration;
_maxSessions = maxSessions;
}
static async Task <CosmosDbLivySession> CreateSession(ILivyClient client, CosmosCollectionSettings cosmosCollectionSettings, LivySessionConfiguration livySessionConfiguration)
{
var session = new CosmosDbLivySession(client, cosmosCollectionSettings, livySessionConfiguration);
await session.WaitForSessionAsync().ConfigureAwait(false);
return(session);
}
static Task <ILivySession> CreateSessionAsync(ILivyClient client, LivySessionConfiguration livyConfig)
{
var livySessionConfiguration = livyConfig.Clone();
livySessionConfiguration.Name += " " + Interlocked.Increment(ref _sessionCount);
return(client.CreateSessionAsync(livySessionConfiguration));
}
public CosmosDbLivyObjectPooledSession(ILivyClient client, CosmosCollectionSettings settings, LivySessionConfiguration config, int max) :
base(() => CreateSession(client, settings, config), max)
{
}
static async Task CloseAllSessions(ILivyClient client) => await(await client.GetSessionsAsync()).Select(session => session.CloseAsync());
public CosmosDbLivySession(ILivyClient client, CosmosCollectionSettings settings, LivySessionConfiguration livyConfig)
{
_settings = settings;
_livyConfig = livyConfig;
_session = new Lazy <Task <ILivySession> >(() => CreateSessionAsync(client, livyConfig));
}
