Skip to content

Commit

Permalink
Merge pull request #258 from nomiero/2.3
Browse files Browse the repository at this point in the history
Add support for using multiple database accounts in the same spark session
  • Loading branch information
nomiero committed Sep 28, 2018
2 parents a03585a + bab9bc8 commit 25dc659
Showing 1 changed file with 5 additions and 5 deletions.
Expand Up @@ -41,19 +41,19 @@ object CosmosDBConnection {
// For verification purpose
var lastConnectionPolicy: ConnectionPolicy = _
var lastConsistencyLevel: Option[ConsistencyLevel] = _
var client: DocumentClient = _
var clients: scala.collection.mutable.Map[String, DocumentClient] = scala.collection.mutable.Map[String,DocumentClient]()

def getClient(connectionMode: ConnectionMode, clientConfiguration: ClientConfiguration): DocumentClient = synchronized {
if (client == null) {
client = new DocumentClient(
if (!clients.contains(clientConfiguration.host)) {
clients(clientConfiguration.host) = new DocumentClient(
clientConfiguration.host,
clientConfiguration.key,
clientConfiguration.connectionPolicy,
clientConfiguration.consistencyLevel)
CosmosDBConnection.lastConsistencyLevel = Some(clientConfiguration.consistencyLevel)
CosmosDBConnection.lastConsistencyLevel = Some(clientConfiguration.consistencyLevel)
}

client
clients.get(clientConfiguration.host).get
}
}

Expand Down

0 comments on commit 25dc659

Please sign in to comment.