Record associations to the database (#263)

* Record associations to the database

Signed-off-by: Victor Chang <[email protected]>

Author: mocsharp

src/Api/DicomAssociationInfo.cs

@@ -0,0 +1,49 @@
+/*
+ * Copyright 2022 MONAI Consortium
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+using System;
+
+namespace Monai.Deploy.InformaticsGateway.Api
+{
+    public class DicomAssociationInfo : MongoDBEntityBase
+    {
+        public DateTime DateTimeDisconnected { get; set; }
+        public string CorrelationId { get; set; }
+        public int FileCount { get; private set; }
+        public string CallingAeTitle { get; set; }
+        public string CalledAeTitle { get; set; }
+        public string RemoteHost { get; set; }
+        public int RemotePort { get; set; }
+        public string Errors { get; set; }
+        public TimeSpan Duration { get; private set; }
+
+        public DicomAssociationInfo()
+        {
+            FileCount = 0;
+        }
+
+        public void FileReceived()
+        {
+            FileCount++;
+        }
+
+        public void Disconnect()
+        {
+            DateTimeDisconnected = DateTime.UtcNow;
+            Duration = DateTimeDisconnected.Subtract(DateTimeCreated);
+        }
+    }
+}

src/Database/Api/Repositories/IDicomAssociationInfoRepository.cs

@@ -0,0 +1,28 @@
+/*
+ * Copyright 2022 MONAI Consortium
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+using Monai.Deploy.InformaticsGateway.Api;
+
+namespace Monai.Deploy.InformaticsGateway.Database.Api.Repositories
+{
+    public interface IDicomAssociationInfoRepository
+    {
+        Task<List<DicomAssociationInfo>> ToListAsync(CancellationToken cancellationToken = default);
+
+        Task<DicomAssociationInfo> AddAsync(DicomAssociationInfo item, CancellationToken cancellationToken = default);
+
+    }
+}

src/Database/DatabaseManager.cs

@@ -54,6 +54,7 @@ public static IServiceCollection ConfigureDatabase(this IServiceCollection servi
                     services.AddScoped(typeof(ISourceApplicationEntityRepository), typeof(EntityFramework.Repositories.SourceApplicationEntityRepository));
                     services.AddScoped(typeof(IStorageMetadataRepository), typeof(EntityFramework.Repositories.StorageMetadataWrapperRepository));
                     services.AddScoped(typeof(IPayloadRepository), typeof(EntityFramework.Repositories.PayloadRepository));
+                    services.AddScoped(typeof(IDicomAssociationInfoRepository), typeof(EntityFramework.Repositories.DicomAssociationInfoRepository));
                     return services;
                 case DbType_MongoDb:
                     services.AddSingleton<IMongoClient, MongoClient>(s => new MongoClient(connectionStringConfigurationSection[SR.DatabaseConnectionStringKey]));
@@ -65,6 +66,7 @@ public static IServiceCollection ConfigureDatabase(this IServiceCollection servi
                     services.AddScoped(typeof(ISourceApplicationEntityRepository), typeof(MongoDB.Repositories.SourceApplicationEntityRepository));
                     services.AddScoped(typeof(IStorageMetadataRepository), typeof(MongoDB.Repositories.StorageMetadataWrapperRepository));
                     services.AddScoped(typeof(IPayloadRepository), typeof(MongoDB.Repositories.PayloadRepository));
+                    services.AddScoped(typeof(IDicomAssociationInfoRepository), typeof(MongoDB.Repositories.DicomAssociationInfoRepository));
 
                     return services;
                 default:

src/Database/EntityFramework/Configuration/DicomAssociationInfoConfiguration.cs

@@ -0,0 +1,39 @@
+/*
+ * Copyright 2021-2022 MONAI Consortium
+ * Copyright 2021 NVIDIA Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Metadata.Builders;
+using Monai.Deploy.InformaticsGateway.Api;
+
+namespace Monai.Deploy.InformaticsGateway.Database.EntityFramework.Configuration
+{
+    internal class DicomAssociationInfoConfiguration : IEntityTypeConfiguration<DicomAssociationInfo>
+    {
+        public void Configure(EntityTypeBuilder<DicomAssociationInfo> builder)
+        {
+            builder.HasKey(j => j.Id);
+            builder.Property(j => j.CalledAeTitle).IsRequired();
+            builder.Property(j => j.CalledAeTitle).IsRequired();
+            builder.Property(j => j.DateTimeCreated).IsRequired();
+            builder.Property(j => j.DateTimeDisconnected).IsRequired();
+            builder.Property(j => j.CorrelationId).IsRequired();
+            builder.Property(j => j.FileCount).IsRequired();
+            builder.Property(j => j.RemoteHost).IsRequired();
+            builder.Property(j => j.RemotePort).IsRequired();
+        }
+    }
+}

src/Database/EntityFramework/InformaticsGatewayContext.cs

@@ -40,6 +40,7 @@ public InformaticsGatewayContext(DbContextOptions<InformaticsGatewayContext> opt
         public virtual DbSet<InferenceRequest> InferenceRequests { get; set; }
         public virtual DbSet<Payload> Payloads { get; set; }
         public virtual DbSet<StorageMetadataWrapper> StorageMetadataWrapperEntities { get; set; }
+        public virtual DbSet<DicomAssociationInfo> DicomAssociationHistories { get; set; }
 
         protected override void OnModelCreating(ModelBuilder modelBuilder)
         {
@@ -51,6 +52,7 @@ protected override void OnModelCreating(ModelBuilder modelBuilder)
             modelBuilder.ApplyConfiguration(new InferenceRequestConfiguration());
             modelBuilder.ApplyConfiguration(new PayloadConfiguration());
             modelBuilder.ApplyConfiguration(new StorageMetadataWrapperEntityConfiguration());
+            modelBuilder.ApplyConfiguration(new DicomAssociationInfoConfiguration());
         }
 
         protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)