microsoft-visualbasic-runtime/ApplicationServices/ServerModule.vb

Imports System.Runtime.CompilerServices
Imports System.Text
Imports Microsoft.VisualBasic.Language.Default
Imports Microsoft.VisualBasic.Net.Protocols
Imports Microsoft.VisualBasic.Net.Protocols.Reflection
Imports Microsoft.VisualBasic.Net.Tcp
Imports Microsoft.VisualBasic.Serialization.JSON
Imports Microsoft.VisualBasic.Text

Namespace ApplicationServices

    ''' <summary>
    ''' The Tcp socket server abstract
    ''' </summary>
    Public MustInherit Class ServerModule : Implements IDisposable

        ''' <summary>
        ''' Tcp socket
        ''' </summary>
        Protected socket As TcpServicesSocket

        ''' <summary>
        ''' Create a new server module based on a tcp server socket.
        ''' </summary>
        ''' <param name="port">The listen port of the tcp socket.</param>
        Sub New(port As Integer)
            socket = New TcpServicesSocket(port, AddressOf LogException) With {
                .Responsehandler = ProtocolHandler()
            }
        End Sub

        Protected MustOverride Sub LogException(ex As Exception)
        ''' <summary>
        ''' Generally, using a <see cref="Protocol"/> attribute using reflection way is recommended.
        ''' </summary>
        ''' <returns></returns>
        Protected MustOverride Function ProtocolHandler() As ProtocolHandler

        Public Overridable Function Run() As Integer
            Return socket.Run
        End Function

#Region "IDisposable Support"
        Private disposedValue As Boolean ' To detect redundant calls

        ' IDisposable
        Protected Overridable Sub Dispose(disposing As Boolean)
            If Not disposedValue Then
                If disposing Then
                    ' TODO: dispose managed state (managed objects).
                    Call socket.Dispose()
                End If

                ' TODO: free unmanaged resources (unmanaged objects) and override Finalize() below.
                ' TODO: set large fields to null.
            End If
            disposedValue = True
        End Sub

        ' TODO: override Finalize() only if Dispose(disposing As Boolean) above has code to free unmanaged resources.
        'Protected Overrides Sub Finalize()
        '    ' Do not change this code.  Put cleanup code in Dispose(disposing As Boolean) above.
        '    Dispose(False)
        '    MyBase.Finalize()
        'End Sub

        ' This code added by Visual Basic to correctly implement the disposable pattern.
        Public Sub Dispose() Implements IDisposable.Dispose
            ' Do not change this code.  Put cleanup code in Dispose(disposing As Boolean) above.
            Dispose(True)
            ' TODO: uncomment the following line if Finalize() is overridden above.
            ' GC.SuppressFinalize(Me)
        End Sub
#End Region
    End Class

    Public Class ProtocolInvoker(Of T As {IComparable, IFormattable, IConvertible})

        Public ReadOnly Property Protocol As New Protocol(GetType(T))
        Public ReadOnly Property TcpRequest As TcpRequest
        Public ReadOnly Property TextEncoding As DefaultValue(Of Encoding)

        <MethodImpl(MethodImplOptions.AggressiveInlining)>
        Sub New(hostName$, remotePort%, Optional encoding As Encodings = Encodings.UTF8WithoutBOM)
            TcpRequest = New TcpRequest(hostName, remotePort)
            TextEncoding = encoding.CodePage
        End Sub

        <MethodImpl(MethodImplOptions.AggressiveInlining)>
        Public Function SendMessage(protocol As T, message$, Optional textEncoding As Encoding = Nothing) As RequestStream
            Return SendMessage(protocol, (textEncoding Or Me.TextEncoding).GetBytes(message))
        End Function

        Public Function SendMessage(protocol As T, data As Byte()) As RequestStream
            Dim category& = Me.Protocol.EntryPoint
            Dim protocolL& = CLng(CObj(protocol))
            Dim message As New RequestStream(category, protocolL, data)

            Return TcpRequest.SendMessage(message)
        End Function

        <MethodImpl(MethodImplOptions.AggressiveInlining)>
        Public Function SendMessage(Of V As {New, Class})(protocol As T, data As V) As RequestStream
            Return SendMessage(protocol, data.GetJson)
        End Function
    End Class
End Namespace