ChangeFeedInvoke.cs 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  1. using DocumentFormat.OpenXml.Drawing;
  2. using Microsoft.Azure.Cosmos;
  3. using Microsoft.Extensions.DependencyInjection;
  4. using System;
  5. using System.Collections.Generic;
  6. using System.Linq;
  7. using System.Reflection;
  8. using System.Text;
  9. using System.Text.Json;
  10. using System.Threading;
  11. using System.Threading.Tasks;
  12. using TEAMModelOS.SDK.Context.Configuration;
  13. using TEAMModelOS.SDK.Helper.Common.JsonHelper;
  14. using TEAMModelOS.SDK.Module.AzureCosmosDBV3;
  15. namespace TEAMModelOS.Service.Services.ChangeFeed
  16. {
  17. public class ChangeFeedInvoke : IChangeFeedInvoke
  18. {
  19. private readonly IAzureCosmosDBV3Repository azureCosmosDBV3Repository;
  20. public ChangeFeedInvoke(IAzureCosmosDBV3Repository _IAzureCosmosDBV3Repository) {
  21. azureCosmosDBV3Repository = _IAzureCosmosDBV3Repository;
  22. }
  23. public void MonitorChangeFeed(Dictionary<string , CosmosModelInfo> dict, IServiceCollection _services) {
  24. var serviceCollection = new Microsoft.Extensions.DependencyInjection.ServiceCollection();
  25. ServiceProvider serviceProvider = _services.BuildServiceProvider();
  26. List<ServiceDescriptor> services = new List<ServiceDescriptor>();
  27. Type type = typeof(IChangeFeedService<>);
  28. foreach (ServiceDescriptor service in _services) {
  29. if (type.Name.Equals(service.ServiceType.Name)) {
  30. services.Add(service);
  31. }
  32. }
  33. foreach (string CollectionName in dict.Keys)
  34. {
  35. if (CollectionName.Equals("AleaseContainer"))
  36. {
  37. continue;
  38. }
  39. if (dict[CollectionName].monitor) {
  40. ChangeFeedProcessor changeFeedProcessor = dict[CollectionName]
  41. .container
  42. .GetChangeFeedProcessorBuilder<object>(CollectionName, async (changes, token) =>
  43. await ProcessChanges(changes, dict[CollectionName].type , services, serviceProvider))
  44. .WithInstanceName(CollectionName)
  45. .WithLeaseContainer(dict["AleaseContainer"].container)
  46. .Build();
  47. changeFeedProcessor.StartAsync();
  48. }
  49. }
  50. }
  51. private async Task ProcessChanges(IReadOnlyCollection<object> changes, Type type,
  52. List<ServiceDescriptor> services, ServiceProvider serviceProvider)
  53. {
  54. foreach (ServiceDescriptor service in services) {
  55. var obj = serviceProvider.GetService(service.ServiceType);
  56. MethodInfo mi = service.ServiceType.GetMethod("Processor");
  57. Type t = typeof(IReadOnlyCollection<>);
  58. t = t.MakeGenericType(type);
  59. object bjt = JsonSerializer.Deserialize(changes.ToApiJson(), t);
  60. mi.Invoke(obj, new object[] { bjt });//调用方法
  61. }
  62. }
  63. }
  64. }