public async Task InitializeDatabase()
{
string endpoint = string.Format("{0}dbs", _docDbEndpoint);
string queryString = "SELECT * FROM dbs db WHERE (db.id = @id)";
var queryParams = new Dictionary <string, object>();
queryParams.Add("@id", _dbName);
DocDbRestQueryResult result = await QueryDocDbInternal(endpoint, queryString, queryParams, DocDbResourceType.Database, "");
IEnumerable databases = result.ResultSet as IEnumerable;
if (databases != null)
{
foreach (object database in databases)
{
if (database != null)
{
object id = ReflectionHelper.GetNamedPropertyValue(database, "id", true, false);
if ((id != null) && (id.ToString() == this._dbName))
{
object rid = ReflectionHelper.GetNamedPropertyValue(database, "_rid", true, false);
if (rid != null)
{
this._dbId = rid.ToString();
}
}
}
}
}
if (string.IsNullOrWhiteSpace(_dbId))
{
await CreateDatabase();
await CreateCollection();
}
}
public async Task InitializeCollection()
{
string endpoint = string.Format("{0}dbs/{1}/colls", _docDbEndpoint, _dbId);
string queryString = "SELECT * FROM colls c WHERE (c.id = @id)";
var queryParams = new Dictionary <string, object>();
queryParams.Add("@id", _collectionName);
DocDbRestQueryResult result = await QueryDocDbInternal(endpoint, queryString, queryParams, DocDbResourceType.Collection, _dbId);
IEnumerable collections = result.ResultSet as IEnumerable;
if (collections != null)
{
foreach (object col in collections)
{
object id = ReflectionHelper.GetNamedPropertyValue(col, "id", true, false);
if ((id != null) && (id.ToString() == this._collectionName))
{
object rid = ReflectionHelper.GetNamedPropertyValue(col, "_rid", true, false);
if (rid != null)
{
this._collectionId = rid.ToString();
return;
}
}
}
}
if (string.IsNullOrWhiteSpace(_collectionId))
{
await CreateCollection();
}
}
private async Task <DocDbRestQueryResult> QueryDocDbInternal(string endpoint, string queryString, Dictionary <string, Object> queryParams,
DocDbResourceType resourceType, string resourceId, int pageSize = -1, string continuationToken = null)
{
if (string.IsNullOrWhiteSpace(endpoint))
{
throw new ArgumentException("endpoint is null or whitespace");
}
if (string.IsNullOrWhiteSpace(queryString))
{
throw new ArgumentException("queryString is null or whitespace");
}
using (WebClient client = new WebClient())
{
client.Encoding = System.Text.Encoding.UTF8;
client.Headers.Add(CONTENT_TYPE_HEADER_KEY, APPLICATION_QUERY_JSON);
client.Headers.Add(ACCEPT_HEADER_KEY, APPLICATION_JSON);
client.Headers.Add(VERSION_HEADER_KEY, X_MS_VERSION);
// https://msdn.microsoft.com/en-us/library/azure/dn783368.aspx
// The date of the request, as specified in RFC 1123. The date format is expressed in
// Coordinated Universal Time (UTC), for example. Fri, 08 Apr 2015 03:52:31 GMT.
string formattedTimeString = DateTime.UtcNow.ToString("R", CultureInfo.InvariantCulture).ToLowerInvariant();
client.Headers.Add(DATE_HEADER_KEY, formattedTimeString);
client.Headers.Add(AUTHORIZATION_HEADER_KEY, GetAuthorizationToken(POST_VERB, DocDbResourceTypeHelper.GetResourceTypeString(resourceType), resourceId, formattedTimeString));
client.Headers.Add(IS_QUERY_HEADER_KEY, "true");
if (pageSize >= 0)
{
client.Headers.Add(MAX_ITEMS_HEADER_KEY, pageSize.ToString());
}
if (continuationToken != null && continuationToken.Length > 0)
{
client.Headers.Add(CONTINUATION_HEADER_KEY, continuationToken);
}
var body = new JObject();
body.Add("query", queryString);
if (queryParams != null && queryParams.Count > 0)
{
var paramsArray = new JArray();
foreach (string key in queryParams.Keys)
{
var param = new JObject();
param.Add("name", key);
param.Add("value", JToken.FromObject(queryParams[key]));
paramsArray.Add(param);
}
body.Add("parameters", paramsArray);
}
var result = new DocDbRestQueryResult();
string response = await AzureRetryHelper.OperationWithBasicRetryAsync <string>(async() => await client.UploadStringTaskAsync(endpoint, POST_VERB, body.ToString()));
JObject responseJobj = JObject.Parse(response);
JToken jsonResultSet = responseJobj.GetValue(DocDbResourceTypeHelper.GetResultSetKey(resourceType));
if (jsonResultSet != null)
{
result.ResultSet = (JArray)jsonResultSet;
}
WebHeaderCollection responseHeaders = client.ResponseHeaders;
string count = responseHeaders[ITEM_COUNT_RESPONSE_HEADER_KEY];
if (!string.IsNullOrEmpty(count))
{
result.TotalResults = int.Parse(count);
}
result.ContinuationToken = responseHeaders[CONTINUATION_HEADER_KEY];
return(result);
}
}
