public override Task OnConnectedAsync() { Producer producerToRegister; string clientConnectionId = Context.ConnectionId; var headers = Context.GetHttpContext().Request.Headers; // authorization tokens // TODO: Implement token validation string tenantToken = headers["x-andyx-tenant-authoriziation"]; string componentToken = headers["x-andyx-component-authoriziation"]; string tenant = headers["x-andyx-tenant"].ToString(); string product = headers["x-andyx-product"].ToString(); string component = headers["x-andyx-component"].ToString(); string topic = headers["x-andyx-topic"].ToString(); bool isPersistent = Boolean.Parse(headers["x-andyx-topic-is-persistent"]); string producerName = headers["x-andyx-producer"].ToString(); logger.LogInformation($"Producer '{producerName}' at {tenant}/{product}/{component}/{topic} requested connection"); //check if the producer is already connected var connectedTenant = tenantRepository.GetTenant(tenant); if (connectedTenant == null) { logger.LogInformation($"Producer '{producerName}' failed to connect, tenant '{tenant}' does not exists"); return(OnDisconnectedAsync(new Exception($"There is no tenant registered with this name '{tenant}'"))); } // check tenant token validation bool isTenantTokenValidated = tenantRepository.ValidateTenantToken(tenant, tenantToken); if (isTenantTokenValidated != true) { logger.LogInformation($"Producer '{producerName}' failed to connect, access is forbidden. Not authorized"); return(OnDisconnectedAsync(new Exception($"Producer '{producerName}' failed to connect, access is forbidden"))); } var connectedProduct = tenantRepository.GetProduct(tenant, product); if (connectedProduct == null) { if (connectedTenant.Settings.AllowProductCreation != true) { logger.LogInformation($"Producer '{producerName}' failed to connect, tenant '{tenant}' does not allow to create new product"); return(OnDisconnectedAsync(new Exception($"There is no product registered with this name '{product}'. Tenant '{tenant}' does not allow to create new product"))); } var productDetails = tenantFactory.CreateProduct(product); tenantRepository.AddProduct(tenant, product, productDetails); storageHubService.CreateProductAsync(tenant, productDetails); } else { storageHubService.UpdateProductAsync(tenant, connectedProduct); } var connectedComponent = tenantRepository.GetComponent(tenant, product, component); if (connectedComponent == null) { var componentDetails = tenantFactory.CreateComponent(component); tenantRepository.AddComponent(tenant, product, component, componentDetails); storageHubService.CreateComponentAsync(tenant, product, componentDetails); } else { // check component token validation bool isComponentTokenValidated = tenantRepository.ValidateComponentToken(tenant, product, component, componentToken, producerName, false); if (isComponentTokenValidated != true) { logger.LogInformation($"Producer '{producerName}' failed to connect, access is forbidden. Not authorized, check component token"); return(OnDisconnectedAsync(new Exception($"Producer '{producerName}' failed to connect, access is forbidden, check component token"))); } storageHubService.UpdateComponentAsync(tenant, product, connectedComponent); } var connectedTopic = tenantRepository.GetTopic(tenant, product, component, topic); if (connectedTopic == null) { // Check if the component allows to create a new topic. connectedComponent = tenantRepository.GetComponent(tenant, product, component); if (connectedComponent.Settings.AllowTopicCreation != true) { logger.LogInformation($"Component '{component}' does not allow to create a new topic {topic} at '{tenant}/{product}/{component}'. To allow creating update property AllowTopicCreation at component."); return(OnDisconnectedAsync(new Exception($"Component '{component}' does not allow to create a new topic {topic} at '{tenant}/{product}/{component}'. To allow creating update property AllowTopicCreation at component."))); } var topicDetails = tenantFactory.CreateTopic(topic, isPersistent); tenantRepository.AddTopic(tenant, product, component, topic, topicDetails); storageHubService.CreateTopicAsync(tenant, product, component, topicDetails); } else { storageHubService.UpdateTopicAsync(tenant, product, component, connectedTopic); } if (producerHubRepository.GetProducerByProducerName(tenant, product, component, topic, producerName).Equals(default(KeyValuePair <string, Producer>)) != true) { logger.LogWarning($"Producer '{producerName}' at {tenant}/{product}/{component}/{topic} is already connected"); return(OnDisconnectedAsync(new Exception($"There is a producer with name '{producerName}' at {tenant}/{product}/{component}/{topic} connected to this node"))); } producerToRegister = producerFactory.CreateProducer(tenant, product, component, topic, producerName); producerHubRepository.AddProducer(clientConnectionId, producerToRegister); storageHubService.ConnectProducerAsync(producerToRegister); Clients.Caller.ProducerConnected(new Model.Producers.Events.ProducerConnectedDetails() { Id = producerToRegister.Id, Tenant = tenant, Product = product, Component = component, Topic = topic, ProducerName = producerName }); logger.LogInformation($"Producer '{producerName}' at {tenant}/{product}/{component}/{topic} is connected"); return(base.OnConnectedAsync()); }
public override Task OnConnectedAsync() { Consumer consumerToRegister; string clientConnectionId = Context.ConnectionId; var headers = Context.GetHttpContext().Request.Headers; // authorization tokens // TODO: Implement token validation string tenantToken = headers["x-andyx-tenant-authoriziation"]; string componentToken = headers["x-andyx-component-authoriziation"]; string tenant = headers["x-andyx-tenant"].ToString(); string product = headers["x-andyx-product"].ToString(); string component = headers["x-andyx-component"].ToString(); string topic = headers["x-andyx-topic"].ToString(); bool isPersistent = bool.Parse(headers["x-andyx-topic-is-persistent"]); string consumerName = headers["x-andyx-consumer"].ToString(); SubscriptionType subscriptionType = (SubscriptionType)Enum.Parse(typeof(SubscriptionType), headers["x-andyx-consumer-type"].ToString()); InitialPosition initialPosition = (InitialPosition)Enum.Parse(typeof(InitialPosition), headers["x-andyx-consumer-initial-position"].ToString()); logger.LogInformation($"Consumer '{consumerName}' and subscription type '{subscriptionType}' at {tenant}/{product}/{component}/{topic} requested connection"); // check if the consumer is already connected var connectedTenant = tenantRepository.GetTenant(tenant); if (connectedTenant == null) { logger.LogInformation($"Consumer '{consumerName}' failed to connect, tenant '{tenant}' does not exists"); return(OnDisconnectedAsync(new Exception($"There is no tenant registered with this name '{tenant}'"))); } // check tenant token validation bool isTenantTokenValidated = tenantRepository.ValidateTenantToken(tenant, tenantToken); if (isTenantTokenValidated != true) { logger.LogInformation($"Consumer '{consumerName}' failed to connect, access is forbidden. Not authorized"); return(OnDisconnectedAsync(new Exception($"Consumer '{consumerName}' failed to connect, access is forbidden"))); } var connectedProduct = tenantRepository.GetProduct(tenant, product); if (connectedProduct == null) { if (connectedTenant.Settings.AllowProductCreation != true) { logger.LogInformation($"Consumer '{consumerName}' failed to connect, tenant '{tenant}' does not allow to create new product"); return(OnDisconnectedAsync(new Exception($"There is no product registered with this name '{product}'. Tenant '{tenant}' does not allow to create new product"))); } var productDetails = tenantFactory.CreateProduct(product); tenantRepository.AddProduct(tenant, product, productDetails); storageHubService.CreateProductAsync(tenant, productDetails); } else { storageHubService.UpdateProductAsync(tenant, connectedProduct); } var connectedComponent = tenantRepository.GetComponent(tenant, product, component); if (connectedComponent == null) { var componentDetails = tenantFactory.CreateComponent(component); tenantRepository.AddComponent(tenant, product, component, componentDetails); storageHubService.CreateComponentAsync(tenant, product, componentDetails); } else { // check component token validation bool isComponentTokenValidated = tenantRepository.ValidateComponentToken(tenant, product, component, componentToken, consumerName, true); if (isComponentTokenValidated != true) { logger.LogInformation($"Consumer '{consumerName}' failed to connect, access is forbidden. Not authorized, check component token"); return(OnDisconnectedAsync(new Exception($"Consumer '{consumerName}' failed to connect, access is forbidden, check component token"))); } storageHubService.UpdateComponentAsync(tenant, product, connectedComponent); } var connectedTopic = tenantRepository.GetTopic(tenant, product, component, topic); if (connectedTopic == null) { connectedComponent = tenantRepository.GetComponent(tenant, product, component); if (connectedComponent.Settings.AllowTopicCreation != true) { logger.LogInformation($"Component '{component}' does not allow to create a new topic {topic} at '{tenant}/{product}/{component}'. To allow creating update property AllowTopicCreation at component."); return(OnDisconnectedAsync(new Exception($"Component '{component}' does not allow to create a new topic {topic} at '{tenant}/{product}/{component}'. To allow creating update property AllowTopicCreation at component."))); } var topicDetails = tenantFactory.CreateTopic(topic, isPersistent); tenantRepository.AddTopic(tenant, product, component, topic, topicDetails); storageHubService.CreateTopicAsync(tenant, product, component, topicDetails); } else { storageHubService.UpdateTopicAsync(tenant, product, component, connectedTopic); } string consumerIdOnRepo = $"{tenant}{product}{component}{topic}|{consumerName}"; var consumerConencted = consumerHubRepository.GetConsumerById(consumerIdOnRepo); if (consumerConencted != null) { if (subscriptionType == SubscriptionType.Exclusive) { logger.LogWarning($"Consumer '{consumerName}' and subscription type '{subscriptionType}' at {tenant}/{product}/{component}/{topic} is already connected"); return(OnDisconnectedAsync(new Exception($"There is a consumer with name '{consumerName}' and with type 'EXCLUSIVE' is connected to this node"))); } if (subscriptionType == SubscriptionType.Failover) { if (consumerConencted.Connections.Count >= 2) { logger.LogWarning($"Consumer '{consumerName}' and subscription type '{subscriptionType}' at {tenant}/{product}/{component}/{topic} is already connected with 2 instances"); return(OnDisconnectedAsync(new Exception($"There are two consumers with name '{consumerName}' and with type 'Failover' are connected to this node"))); } } } consumerToRegister = consumerFactory.CreateConsumer(tenant, product, component, topic, consumerName, subscriptionType, initialPosition); consumerHubRepository.AddConsumer(consumerIdOnRepo, consumerToRegister); consumerHubRepository.AddConsumerConnection(consumerIdOnRepo, clientConnectionId); storageHubService.ConnectConsumerAsync(consumerToRegister); Clients.Caller.ConsumerConnected(new Model.Consumers.Events.ConsumerConnectedDetails() { Id = consumerToRegister.Id, Tenant = tenant, Product = product, Component = component, Topic = topic, ConsumerName = consumerName, SubscriptionType = subscriptionType, InitialPosition = initialPosition }); // if consumer is not persistent, do not store the message, just allow streaming if (isPersistent == true) { // Sent not acknoledged messages to this consumer (for exclusive and for the first shared/failover consumer connected) if (subscriptionType == SubscriptionType.Exclusive) { storageHubService.RequestUnacknowledgedMessagesConsumer(consumerToRegister); } if (subscriptionType == SubscriptionType.Shared || subscriptionType == SubscriptionType.Failover) { if (consumerConencted == null) { storageHubService.RequestUnacknowledgedMessagesConsumer(consumerToRegister); } } } logger.LogInformation($"Consumer '{consumerName}' and subscription type '{subscriptionType}' at {tenant}/{product}/{component}/{topic} is connected"); return(base.OnConnectedAsync()); }
public bool AddTopic(string tenantName, string productName, string componentName, string topicName) { return(_tenantRepository .AddTopic(tenantName, productName, componentName, topicName, _tenantFactory.CreateTopic(topicName))); }