dendrite/internal/cosmosdbutil/connection.go

package cosmosdbutil

import (
	"strings"

	"github.com/matrix-org/dendrite/internal/cosmosdbapi"
	"github.com/matrix-org/dendrite/setup/config"
)

const accountEndpointName = "AccountEndpoint"
const accountKeyName = "AccountKey"
const databaseName = "DatabaseName"
const containerName = "ContainerName"

func getConnectionString(d *config.DataSource) config.DataSource {
	var connString string
	connString = string(*d)
	return config.DataSource(strings.Replace(connString, "cosmosdb:", "", 1))
}

func getConnectionProperties(connectionString string) map[string]string {
	connectionItemsRaw := strings.Split(connectionString, ";")
	connectionItems := map[string]string{}
	for _, item := range connectionItemsRaw {
		if len(item) > 0 {
			itemSplit := strings.SplitN(item, "=", 2)
			connectionItems[itemSplit[0]] = itemSplit[1]
		}
	}
	return connectionItems
}

func GetCosmosConnection(d *config.DataSource) cosmosdbapi.CosmosConnection {
	connString := getConnectionString(d)
	connMap := getConnectionProperties(string(connString))
	accountEndpoint := connMap[accountEndpointName]
	accountKey := connMap[accountKeyName]
	return cosmosdbapi.GetCosmosConnection(accountEndpoint, accountKey)
}

func GetCosmosConfig(d *config.DataSource) cosmosdbapi.CosmosConfig {
	connString := getConnectionString(d)
	connMap := getConnectionProperties(string(connString))
	database := connMap[databaseName]
	container := connMap[containerName]
	return cosmosdbapi.CosmosConfig{
		DatabaseName:  database,
		ContainerName: container,
	}
}