EDMI: Add Testing to EDMIBenchmark, Add ListFiles to EDMIService

Service.EDMIService/EDMIService.vb

@@ -30,12 +30,24 @@ Public Class EDMIService
     Public Sub New()
         Dim oOperationContext As OperationContext = OperationContext.Current
         Dim oInstanceContext As InstanceContext = oOperationContext.InstanceContext
-        Dim oUsername = oOperationContext.ServiceSecurityContext.WindowsIdentity.Name
+        Dim oUsername = StripDomainFromUsername(oOperationContext.ServiceSecurityContext.WindowsIdentity.Name)
 
         _username = oUsername
         _logger = LogConfig.GetLogger()
+
+        _logger.Debug("New Request by User [{0}]", _username)
     End Sub
 
+    Public Function StripDomainFromUsername(UserName As String)
+        If UserName.Contains("\") Then
+            Return UserName.Split("\")(1)
+        ElseIf UserName.Contains("@") Then
+            Return UserName.Split("@")(0)
+        Else
+            Return UserName
+        End If
+    End Function
+
 #Region "Auth"
     Private Function TestUserAuth() As Boolean
         Try
@@ -391,6 +403,21 @@ Public Class EDMIService
         End Try
     End Function
 
+    Public Function ListFilesForUser() As Messages.DocumentListResponse Implements IEDMIService.ListFilesForUser
+        Try
+            Dim oSQL = $"SELECT * FROM VWIDB_DOC_DATA WHERE ADDED_WHO = UPPER('{_username}')"
+            Dim oDatatable As DataTable = MSSQL.GetDatatable(oSQL)
+            oDatatable.TableName = "DocumentList"
+
+            Return New Messages.DocumentListResponse() With {
+                .Datatable = oDatatable
+            }
+        Catch ex As Exception
+            _logger.Error(ex)
+            Throw New FaultException(ex.Message)
+        End Try
+    End Function
+
 #End Region
 
 #Region "Index"

Service.EDMIService/IEDMIService.vb

@@ -53,6 +53,9 @@ Interface IEDMIService
 
     <OperationContract>
     Function GetFileByObjectId(Data As Messages.DocumentStreamRequest) As Messages.DocumentStreamResponse
+
+    <OperationContract>
+    Function ListFilesForUser() As Messages.DocumentListResponse
 #End Region
 
 #Region "Index"

Service.EDMIService/Messages.vb

@@ -1,7 +1,27 @@
 Imports System.IO
+Imports System.Runtime.Serialization
 Imports System.ServiceModel
 
 Namespace Messages
+    <Serializable>
+    <DataContract>
+    <KnownType(GetType(DBNull))>
+    Public MustInherit Class BaseResponse
+        <DataMember>
+        Public Property OK As Boolean
+        <DataMember>
+        Public Property ErrorMessage As String
+
+        Public Sub New()
+            OK = True
+        End Sub
+
+        Public Sub New(ErrorMessage As String)
+            OK = False
+            Me.ErrorMessage = ErrorMessage
+        End Sub
+    End Class
+
     <MessageContract>
     Public Class DocumentStreamRequest
         <MessageBodyMember>
@@ -16,6 +36,13 @@ Namespace Messages
         <MessageBodyMember(Order:=1)>
         Public FileContents As Stream
     End Class
+
+    Public Class DocumentListResponse
+        Inherits BaseResponse
+
+        Public Datatable As DataTable
+    End Class
+
 End Namespace