-
Notifications
You must be signed in to change notification settings - Fork 494
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
27b9350
commit f8739d9
Showing
5 changed files
with
300 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
4 changes: 4 additions & 0 deletions
4
Microsoft.Azure.Cosmos.Samples/Usage/CustomDiagnosticAndEventListener/AppSettings.json
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,4 @@ | ||
| { | ||
| "CosmosDBEndPointUrl": "https://localhost:8081", | ||
| "CosmosDBAuthorizationKey": "Super secret key" | ||
| } |
167 changes: 167 additions & 0 deletions
167
...Cosmos.Samples/Usage/CustomDiagnosticAndEventListener/CustomDiagnosticAndEventListener.cs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,167 @@ | ||
| namespace Sample.Listeners | ||
| { | ||
| using System.Diagnostics.Tracing; | ||
| using System.Diagnostics; | ||
| using System.Collections.Concurrent; | ||
|
|
||
| internal class CustomDiagnosticAndEventListener : | ||
| EventListener, // Override Event Listener to capture Event source events | ||
| IObserver<KeyValuePair<string, object>>, // Override IObserver to capture Activity events | ||
| IObserver<DiagnosticListener>, | ||
| IDisposable | ||
| { | ||
| private readonly string diagnosticSourceName; | ||
| private readonly string eventSourceName; | ||
|
|
||
| private ConcurrentBag<IDisposable>? Subscriptions = new(); | ||
| private ConcurrentBag<Activity> Activities { get; } = new(); | ||
|
|
||
| public CustomDiagnosticAndEventListener(string diagnosticSourceName, string eventSourceName) | ||
| { | ||
| this.diagnosticSourceName = diagnosticSourceName; | ||
| this.eventSourceName = eventSourceName; | ||
|
|
||
| DiagnosticListener.AllListeners.Subscribe(this); | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// IObserver Override | ||
| /// </summary> | ||
| public void OnCompleted() { | ||
| throw new NotImplementedException(); | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// IObserver Override | ||
| /// </summary> | ||
| public void OnError(Exception error) { | ||
| throw new NotImplementedException(); | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// IObserver Override | ||
| /// </summary> | ||
| public void OnNext(KeyValuePair<string, object> value) | ||
| { | ||
| lock (this.Activities) | ||
| { | ||
| // Check for disposal | ||
| if (this.Subscriptions == null) return; | ||
|
|
||
| string startSuffix = ".Start"; | ||
| string stopSuffix = ".Stop"; | ||
| string exceptionSuffix = ".Exception"; | ||
|
|
||
| if (Activity.Current == null) | ||
| { | ||
| return; | ||
| } | ||
|
|
||
| if (value.Key.EndsWith(startSuffix)) | ||
| { | ||
| this.Activities.Add(Activity.Current); | ||
| } | ||
| else if (value.Key.EndsWith(stopSuffix) || value.Key.EndsWith(exceptionSuffix)) | ||
| { | ||
| foreach (Activity activity in this.Activities) | ||
| { | ||
| if (activity.Id == Activity.Current.Id) | ||
| { | ||
| Console.WriteLine(); | ||
| Console.WriteLine("---------------------------------------------------------------------------------------------------------"); | ||
| Console.WriteLine($" Activity Name: {activity.DisplayName}"); | ||
| Console.WriteLine($" Activity Operation Name: {activity.OperationName}"); | ||
| foreach (KeyValuePair<string, string?> actualTag in activity.Tags) | ||
| { | ||
| Console.WriteLine($" {actualTag.Key} ==> {actualTag.Value}"); | ||
| } | ||
| Console.WriteLine("---------------------------------------------------------------------------------------------------------"); | ||
| return; | ||
| } | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// IObserver Override | ||
| /// </summary> | ||
| public void OnNext(DiagnosticListener value) | ||
| { | ||
| if (value.Name == this.diagnosticSourceName && this.Subscriptions != null) | ||
| { | ||
| Console.WriteLine(); | ||
| Console.WriteLine($"CustomDiagnosticAndEventListener : OnNext : {value.Name}"); | ||
| lock (this.Activities) | ||
| { | ||
| this.Subscriptions?.Add(value.Subscribe(this)); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// EventListener Override | ||
| /// </summary> | ||
| protected override void OnEventSourceCreated(EventSource eventSource) | ||
| { | ||
| if (eventSource != null && eventSource.Name.Equals(this.eventSourceName)) | ||
| { | ||
| Console.WriteLine(); | ||
| Console.WriteLine($"CustomDiagnosticAndEventListener : OnEventSourceCreated : {eventSource.Name}"); | ||
| this.EnableEvents(eventSource, EventLevel.Informational); // Enable information level events | ||
| } | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// EventListener Override | ||
| /// </summary> | ||
| protected override void OnEventWritten(EventWrittenEventArgs eventData) | ||
| { | ||
| Console.WriteLine(); | ||
| Console.WriteLine("---------------------------------------------------------------------------------------------------------"); | ||
| Console.WriteLine($" Event Name: {eventData.EventName}"); | ||
| Console.WriteLine($" Event Level: {eventData.Level}"); | ||
| if(eventData.Payload != null) | ||
| { | ||
| int counter = 0; | ||
| foreach (object? payload in eventData.Payload) | ||
| { | ||
| Console.WriteLine($" Event Payload {counter++}: {payload}"); | ||
| } | ||
| } | ||
| else | ||
| { | ||
| Console.WriteLine($" Event Payload: NULL"); | ||
| } | ||
| Console.WriteLine("---------------------------------------------------------------------------------------------------------"); | ||
| } | ||
|
|
||
| public override void Dispose() | ||
| { | ||
| Console.WriteLine("CustomDiagnosticAndEventListener : Dispose"); | ||
| base.Dispose(); | ||
|
|
||
| if (this.Subscriptions == null) | ||
| { | ||
| return; | ||
| } | ||
|
|
||
| ConcurrentBag<IDisposable> subscriptions; | ||
| lock (this.Activities) | ||
| { | ||
| subscriptions = this.Subscriptions; | ||
| this.Subscriptions = null; | ||
| } | ||
|
|
||
| foreach (IDisposable subscription in subscriptions) | ||
| { | ||
| subscription.Dispose(); // Dispose of DiagnosticListener subscription | ||
| } | ||
|
|
||
| foreach (Activity activity in this.Activities) | ||
| { | ||
| activity.Dispose(); // Dispose of Activity | ||
| } | ||
| } | ||
| } | ||
| } |
31 changes: 31 additions & 0 deletions
31
...os.Samples/Usage/CustomDiagnosticAndEventListener/CustomDiagnosticAndEventListener.csproj
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,31 @@ | ||
| <Project Sdk="Microsoft.NET.Sdk"> | ||
|
|
||
| <PropertyGroup> | ||
| <OutputType>Exe</OutputType> | ||
| <TargetFramework>net6.0</TargetFramework> | ||
| <ImplicitUsings>enable</ImplicitUsings> | ||
| <Nullable>enable</Nullable> | ||
| </PropertyGroup> | ||
|
|
||
| <ItemGroup> | ||
| <Content Include="..\..\..\..\..\Users\sourabhjain\.nuget\packages\microsoft.azure.cosmos\3.33.0-preview\contentFiles\any\netstandard2.0\ThirdPartyNotice.txt" Link="ThirdPartyNotice.txt"> | ||
| <Private>False</Private> | ||
| <NuGetPackageVersion>3.33.0-preview</NuGetPackageVersion> | ||
| <NuGetItemType>Content</NuGetItemType> | ||
| <NuGetPackageId>Microsoft.Azure.Cosmos</NuGetPackageId> | ||
| <Pack>false</Pack> | ||
| </Content> | ||
| </ItemGroup> | ||
|
|
||
| <ItemGroup> | ||
| <PackageReference Include="Microsoft.Azure.Cosmos" Version="3.35.2-preview" /> | ||
| <PackageReference Include="Microsoft.Extensions.Configuration" Version="3.1.0" /> | ||
| <PackageReference Include="Microsoft.Extensions.Configuration.Json" Version="3.1.0" /> | ||
| </ItemGroup> | ||
|
|
||
| <ItemGroup> | ||
| <None Include="AppSettings.json" Link="AppSettings.json"> | ||
| <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory> | ||
| </None> | ||
| </ItemGroup> | ||
| </Project> |
91 changes: 91 additions & 0 deletions
91
Microsoft.Azure.Cosmos.Samples/Usage/CustomDiagnosticAndEventListener/Program.cs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,91 @@ | ||
| namespace Cosmos.Samples.ApplicationInsights | ||
| { | ||
| using System.Net; | ||
| using Microsoft.Azure.Cosmos; | ||
| using Microsoft.Extensions.Configuration; | ||
| using Newtonsoft.Json; | ||
| using Sample.Listeners; | ||
|
|
||
| internal class Program | ||
| { | ||
| private static readonly string databaseName = "samples"; | ||
| private static readonly string containerName = "custom-listener-sample"; | ||
|
|
||
| static async Task Main() | ||
| { | ||
| IConfigurationRoot configuration = new ConfigurationBuilder() | ||
| .AddJsonFile("AppSettings.json") | ||
| .Build(); | ||
|
|
||
| string endpoint = configuration["CosmosDBEndPointUrl"]; | ||
| if (string.IsNullOrEmpty(endpoint)) | ||
| { | ||
| throw new ArgumentNullException("Please specify a valid CosmosDBEndPointUrl in the appSettings.json"); | ||
| } | ||
|
|
||
| string authKey = configuration["CosmosDBAuthorizationKey"]; | ||
| if (string.IsNullOrEmpty(authKey) || string.Equals(authKey, "Super secret key")) | ||
| { | ||
| throw new ArgumentException("Please specify a valid CosmosDBAuthorizationKey in the appSettings.json"); | ||
| } | ||
|
|
||
| using CustomDiagnosticAndEventListener listener | ||
| = new CustomDiagnosticAndEventListener( | ||
| diagnosticSourceName: "Azure.Cosmos.Operation", | ||
| eventSourceName: "Azure-Cosmos-Operation-Request-Diagnostics"); | ||
|
|
||
| CosmosClientOptions options = new CosmosClientOptions() | ||
| { | ||
| IsDistributedTracingEnabled = true // Defaults to true, set to false to disable | ||
| }; | ||
| using (CosmosClient client = new CosmosClient(endpoint, authKey, options)) | ||
| { | ||
| Console.WriteLine($"Getting container reference for {containerName}."); | ||
|
|
||
| ContainerProperties properties = new ContainerProperties(containerName, partitionKeyPath: "/id"); | ||
|
|
||
| await client.CreateDatabaseIfNotExistsAsync(databaseName); | ||
| Container container = await client.GetDatabase(databaseName).CreateContainerIfNotExistsAsync(properties); | ||
|
|
||
| await Program.RunCrudDemo(container); | ||
| } | ||
| } | ||
|
|
||
| public static async Task RunCrudDemo(Container container) | ||
| { | ||
| // Any operations will automatically generate telemetry | ||
|
|
||
| for (int i = 1; i <= 5; i++) | ||
| { | ||
| await container.CreateItemAsync(new Item { Id = $"{i}", Status = "new" }, new PartitionKey($"{i}")); | ||
| Console.WriteLine($"Created document with id: {i}"); | ||
| } | ||
|
|
||
| for (int i = 1; i <= 5; i++) | ||
| { | ||
| await container.ReadItemAsync<Item>($"{i}", new PartitionKey($"{i}")); | ||
| Console.WriteLine($"Read document with id: {i}"); | ||
| } | ||
|
|
||
| for (int i = 1; i <= 5; i++) | ||
| { | ||
| await container.ReplaceItemAsync(new Item { Id = $"{i}", Status = "updated" }, $"{i}", new PartitionKey($"{i}")); | ||
| Console.WriteLine($"Updated document with id: {i}"); | ||
| } | ||
|
|
||
| for (int i = 1; i <= 5; i++) | ||
| { | ||
| await container.DeleteItemAsync<Item>($"{i}", new PartitionKey($"{i}")); | ||
| Console.WriteLine($"Deleted document with id: {i}"); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| internal class Item | ||
| { | ||
| [JsonProperty("id")] | ||
| public string Id { get; set; } | ||
|
|
||
| public string Status { get; set; } | ||
| } | ||
| } |