protected void WriteAdminKey(AdminKeyResult adminKeyResult)
{
if (adminKeyResult != null)
{
WriteObject(PSSearchAdminKey.Create(adminKeyResult));
}
}
public void CanGetAdminKeys()
{
Run(() =>
{
SearchManagementClient searchMgmt = GetSearchManagementClient();
// List admin keys
AdminKeyResult adminKeyResult =
searchMgmt.AdminKeys.Get(Data.ResourceGroupName, Data.SearchServiceName);
Assert.NotNull(adminKeyResult);
Assert.NotNull(adminKeyResult.PrimaryKey);
Assert.NotNull(adminKeyResult.SecondaryKey);
Assert.NotEmpty(adminKeyResult.PrimaryKey);
Assert.NotEmpty(adminKeyResult.SecondaryKey);
});
}
public override void Initialize(MockContext context)
{
base.Initialize(context);
MockContext = context;
SearchManagementClient client = context.GetServiceClient <SearchManagementClient>();
SearchServiceName = EnsureSearchService(client);
AdminKeyResult adminKeyResult = client.AdminKeys.List(ResourceGroupName, SearchServiceName);
Assert.NotNull(adminKeyResult);
PrimaryApiKey = adminKeyResult.PrimaryKey;
ListQueryKeysResult queryKeyResult = client.QueryKeys.List(ResourceGroupName, SearchServiceName);
Assert.NotNull(queryKeyResult);
Assert.Equal(1, queryKeyResult.Value.Count);
QueryApiKey = queryKeyResult.Value[0].Key;
}
public override void Initialize(MockContext context)
{
base.Initialize(context);
MockContext = context;
SearchManagementClient client = context.GetServiceClient <SearchManagementClient>();
SearchServiceName = EnsureSearchService(client);
AdminKeyResult adminKeyResult = client.AdminKeys.Get(ResourceGroupName, SearchServiceName);
Assert.NotNull(adminKeyResult);
PrimaryApiKey = adminKeyResult.PrimaryKey;
var queryKeys = client.QueryKeys.ListBySearchService(ResourceGroupName, SearchServiceName);
Assert.NotNull(queryKeys);
Assert.Single(queryKeys);
QueryApiKey = queryKeys.First().Key;
}
public void CanRegenerateAdminKeys()
{
Run(() =>
{
SearchManagementClient searchMgmt = GetSearchManagementClient();
AdminKeyResult originalAdminKeys =
searchMgmt.AdminKeys.Get(Data.ResourceGroupName, Data.SearchServiceName);
AdminKeyResult keysWithNewPrimary =
searchMgmt.AdminKeys.Regenerate(Data.ResourceGroupName, Data.SearchServiceName, AdminKeyKind.Primary);
Assert.False(string.IsNullOrEmpty(keysWithNewPrimary.PrimaryKey));
Assert.NotEqual(originalAdminKeys.PrimaryKey, keysWithNewPrimary.PrimaryKey);
Assert.Equal(originalAdminKeys.SecondaryKey, keysWithNewPrimary.SecondaryKey);
AdminKeyResult keysWithNewSecondary =
searchMgmt.AdminKeys.Regenerate(Data.ResourceGroupName, Data.SearchServiceName, AdminKeyKind.Secondary);
Assert.False(string.IsNullOrEmpty(keysWithNewSecondary.SecondaryKey));
Assert.Equal(keysWithNewPrimary.PrimaryKey, keysWithNewSecondary.PrimaryKey);
Assert.NotEqual(keysWithNewPrimary.SecondaryKey, keysWithNewSecondary.SecondaryKey);
});
}
public static PSSearchAdminKey Create(AdminKeyResult adminKeyResult)
{
return(new PSSearchAdminKey(adminKeyResult));
}
public PSSearchAdminKey(AdminKeyResult adminKeyResult)
{
Primary = adminKeyResult.PrimaryKey;
Secondary = adminKeyResult.SecondaryKey;
}
public static async Task Main(string[] args)
{
IConfigurationBuilder builder = new ConfigurationBuilder().AddJsonFile("appsettings.json");
IConfigurationRoot configuration = builder.Build();
// Updated this information in the appsettings.json in this project.
// See the instructions at the link above for the service principal details and the subscription id.
var tenantId = configuration["TenantId"];
var clientId = configuration["ClientId"];
var clientSecret = configuration["ClientSecret"];
var subscriptionId = configuration["SubscriptionId"];
if (new[] { tenantId, clientId, clientSecret, subscriptionId }.Any(i => i.StartsWith("[")))
{
Console.WriteLine("Please provide values for tenantId, clientId, secret and subscriptionId.");
}
else
{
// Build the service credentials using the service principal.
var creds = await ApplicationTokenProvider.LoginSilentAsync(tenantId, clientId, clientSecret);
// Create the various management clients using the credentials instantiated above.
var subscriptionClient = new SubscriptionClient(creds);
var resourceClient = new ResourceManagementClient(creds);
resourceClient.SubscriptionId = subscriptionId;
var searchClient = new SearchManagementClient(creds);
searchClient.SubscriptionId = subscriptionId;
// Get some general subscription information, not Azure Search specific
var subscription = await subscriptionClient.Subscriptions.GetAsync(subscriptionId);
DisplaySubscriptionDetails(subscription);
// Register the Azure Search resource provider with the subscription. In the Azure Resource Manager model, you need
// to register a resource provider in a subscription before you can use it.
// You only need to do this once per subscription/per resource provider.
// More details on registering a resource provider here: https://docs.microsoft.com/rest/api/resources/Providers/Register
var provider = resourceClient.Providers.Register("Microsoft.Search");
DisplayProviderDetails(provider);
// List all search services in the subscription by resource group.
// More details on listing resources here: https://docs.microsoft.com/rest/api/resources/resources/list
var groups = await resourceClient.ResourceGroups.ListAsync();
Console.WriteLine("----------------------------------------------------");
Console.WriteLine("List all search services in the subscription by resource group");
Console.WriteLine("----------------------------------------------------");
foreach (var group in groups)
{
var searchServices = await searchClient.Services.ListByResourceGroupAsync(group.Name);
if (searchServices.Count() > 0)
{
Console.WriteLine("resourceGroup: {0}", group.Name);
}
{
foreach (var service in searchServices)
{
Console.WriteLine(" service name: {0}, sku: {1}, location: {2}", service.Name, service.Sku.Name, service.Location);
}
}
}
Console.WriteLine();
// Create a new search service called "sampleservice#" (where # is a random number, to make it less likely to have collisions)
var random = new Random();
string newServiceName = "sampleservice" + random.Next(0, 1000000).ToString();
string newGroupName = "samplegroup" + random.Next(0, 1000000).ToString();
await resourceClient.ResourceGroups.CreateOrUpdateAsync(newGroupName, new ResourceGroup { Location = "West US" });
// PLEASE NOTE: By default, the code below will create a Standard (S1) search service which is a paid service tier.
// You can change this to the free service tier, but if you already have a free service in this subscription the operation will fail.
var newService = await searchClient.Services.CreateOrUpdateAsync(newGroupName, newServiceName,
new SearchService()
{
Location = "West US",
Sku = new Sku()
{
Name = SkuName.Standard
}, // use "standard" for standard services
PartitionCount = 1,
ReplicaCount = 1
});
// Wait for service provisioning to complete
while (newService.ProvisioningState == ProvisioningState.Provisioning)
{
Thread.Sleep(TimeSpan.FromSeconds(10));
// Retrieve service definition by ResourceGroup name and Service Name
newService = await searchClient.Services.GetAsync(newGroupName, newService.Name);
}
// Retrieve service admin API keys
AdminKeyResult adminKeys = searchClient.AdminKeys.Get(newGroupName, newService.Name);
Console.WriteLine("----------------------------------------------------");
Console.WriteLine("Service admin API keys");
Console.WriteLine("----------------------------------------------------");
Console.WriteLine("Primary admin API key: {0}", adminKeys.PrimaryKey);
Console.WriteLine("Secondary admin API key: {0}", adminKeys.SecondaryKey);
Console.WriteLine();
// Regenerate admin API keys
// (use /primary to regenerate the primary admin API key)
AdminKeyResult newPrimary = searchClient.AdminKeys.Regenerate(newGroupName, newService.Name, AdminKeyKind.Primary);
AdminKeyResult newSecondary = searchClient.AdminKeys.Regenerate(newGroupName, newService.Name, AdminKeyKind.Secondary);
Console.WriteLine("----------------------------------------------------");
Console.WriteLine("Regenerate admin API keys");
Console.WriteLine("----------------------------------------------------");
Console.WriteLine("New primary admin API key: {0}", newPrimary.PrimaryKey);
Console.WriteLine("New secondary admin API key: {0}", newSecondary.SecondaryKey);
Console.WriteLine();
// Create a new query API key
QueryKey newQueryKey = searchClient.QueryKeys.Create(newGroupName, newService.Name, "new query key");
Console.WriteLine("----------------------------------------------------");
Console.WriteLine("Create a new query API key");
Console.WriteLine("----------------------------------------------------");
Console.WriteLine("New query API key: {0}", newQueryKey.Key);
Console.WriteLine();
// Retrieve query API key
searchClient.QueryKeys.Create(newGroupName, newService.Name, "new query key2");
QueryKey getQueryKey = (await searchClient.QueryKeys.ListBySearchServiceAsync(newGroupName, newService.Name)).Where(s => s.Name == "new query key2").First();
Console.WriteLine("----------------------------------------------------");
Console.WriteLine("Retrieve query API key");
Console.WriteLine("----------------------------------------------------");
Console.WriteLine("Retrieved query API key: {0}", getQueryKey.Key);
Console.WriteLine();
// Delete a query API key by name
await searchClient.QueryKeys.DeleteAsync(newGroupName, newService.Name, "new query key2");
Console.WriteLine("----------------------------------------------------");
Console.WriteLine("Deleted query API key by name");
Console.WriteLine("----------------------------------------------------");
Console.WriteLine();
// NOTE: This operation will fail for the free service tier.
if (newService.Sku.Name != SkuName.Free)
{
// Scale up service to 2 replicas. This operation will take several minutes to complete.
newService = searchClient.Services.Update(newGroupName, newService.Name, new SearchService()
{
ReplicaCount = 2
});
// Wait for provisioning to complete
while (newService.ProvisioningState == ProvisioningState.Provisioning)
{
await Task.Delay(TimeSpan.FromSeconds(10));
// Retrieve service definition by ResourceGroup name and Service Name
newService = await searchClient.Services.GetAsync(newGroupName, newService.Name);
}
Console.WriteLine("----------------------------------------------------");
Console.WriteLine("Scale up service to 2 replicas");
Console.WriteLine("----------------------------------------------------");
Console.WriteLine("Partition Count: {0}", newService.PartitionCount);
Console.WriteLine("Replica Count: {0}", newService.ReplicaCount);
Console.WriteLine();
// Scale back down to 1 replica x 1 partition
newService = searchClient.Services.Update(newGroupName, newService.Name, new SearchService()
{
ReplicaCount = 1, PartitionCount = 1
});
// Wait for provisioning to complete
while (newService.ProvisioningState == ProvisioningState.Provisioning)
{
await Task.Delay(TimeSpan.FromSeconds(10));
// Retrieve service definition by ResourceGroup name and Service Name
newService = await searchClient.Services.GetAsync(newGroupName, newService.Name);
}
Console.WriteLine("----------------------------------------------------");
Console.WriteLine("Scale back down to 1 replica x 1 partition");
Console.WriteLine("----------------------------------------------------");
Console.WriteLine("Partition Count: {0}", newService.PartitionCount);
Console.WriteLine("Replica Count: {0}", newService.ReplicaCount);
Console.WriteLine();
}
// Delete search service
await searchClient.Services.DeleteAsync(newGroupName, newService.Name);
if (searchClient.Services.ListByResourceGroup(newGroupName).Count() == 0)
{
Console.WriteLine("----------------------------------------------------");
Console.WriteLine("Search service successfully deleted");
Console.WriteLine("----------------------------------------------------");
Console.WriteLine();
}
}
}
