1	#Region "Microsoft.VisualBasic::b3da10841d6f9a62ea415710a98e602d, Microsoft.VisualBasic.Core\ApplicationServices\ServerModule.vb"
2
3	' Author:
4	'
5	'       asuka (amethyst.asuka@gcmodeller.org)
6	'       xie (genetics@smrucc.org)
7	'       xieguigang (xie.guigang@live.com)
8	'
9	' Copyright (c) 2018 GPL3 Licensed
10	'
11	'
12	' GNU GENERAL PUBLIC LICENSE (GPL3)
13	'
14	'
15	' This program is free software: you can redistribute it and/or modify
16	' it under the terms of the GNU General Public License as published by
17	' the Free Software Foundation, either version 3 of the License, or
18	' (at your option) any later version.
19	'
20	' This program is distributed in the hope that it will be useful,
21	' but WITHOUT ANY WARRANTY; without even the implied warranty of
22	' MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
23	' GNU General Public License for more details.
24	'
25	' You should have received a copy of the GNU General Public License
26	' along with this program. If not, see <http://www.gnu.org/licenses/>.
27
28
29
30	' /********************************************************************************/
31
32	' Summaries:
33
34	'     Class ServerModule
35	'
36	'         Constructor: (+1 Overloads) Sub New
37	'
38	'         Function: Run
39	'
40	'         Sub: (+2 Overloads) Dispose
41	'
42	'     Class ProtocolInvoker
43	'
44	'         Properties: Protocol, TcpRequest, TextEncoding
45	'
46	'         Constructor: (+1 Overloads) Sub New
47	'         Function: (+3 Overloads) SendMessage
48	'
49	'
50	' /********************************************************************************/
51
52	#End Region
53
54	Imports System.Runtime.CompilerServices
55	Imports System.Text
56	Imports Microsoft.VisualBasic.Language.Default
57	Imports Microsoft.VisualBasic.Net.Protocols
58	Imports Microsoft.VisualBasic.Net.Protocols.Reflection
59	Imports Microsoft.VisualBasic.Net.Tcp
60	Imports Microsoft.VisualBasic.Serialization.JSON
61	Imports Microsoft.VisualBasic.Text
62
63	Namespace ApplicationServices
64
65	''' <summary>
66	''' The Tcp socket server abstract
67	''' </summary>
68	Public MustInherit Class ServerModule : Implements IDisposable
69
70	''' <summary>
71	''' Tcp socket
72	''' </summary>
73	Protected socket As TcpServicesSocket
74
75	''' <summary>
76	''' Create a new server module based on a tcp server socket.
77	''' </summary>
78	''' <param name="port">The listen port of the tcp socket.</param>
79	Sub New(port As Integer)
80	socket = New TcpServicesSocket(port, AddressOf LogException) With {
81	.Responsehandler = ProtocolHandler()
82	}
83	End Sub
84
85	Protected MustOverride Sub LogException(ex As Exception)
86	''' <summary>
87	''' Generally, using a <see cref="Protocol"/> attribute using reflection way is recommended.
88	''' </summary>
89	''' <returns></returns>
90	Protected MustOverride Function ProtocolHandler() As ProtocolHandler
91
92	Public Overridable Function Run() As Integer
93	Return socket.Run
94	End Function
95
96	#Region "IDisposable Support"
97	Private disposedValue As Boolean ' To detect redundant calls
98
99	' IDisposable
100	Protected Overridable Sub Dispose(disposing As Boolean)
101	If Not disposedValue Then
102	If disposing Then
103	' TODO: dispose managed state (managed objects).
104	Call socket.Dispose()
105	End If
106
107	' TODO: free unmanaged resources (unmanaged objects) and override Finalize() below.
108	' TODO: set large fields to null.
109	End If
110	disposedValue = True
111	End Sub
112
113	' TODO: override Finalize() only if Dispose(disposing As Boolean) above has code to free unmanaged resources.
114	'Protected Overrides Sub Finalize()
115	'    ' Do not change this code.  Put cleanup code in Dispose(disposing As Boolean) above.
116	'    Dispose(False)
117	'    MyBase.Finalize()
118	'End Sub
119
120	' This code added by Visual Basic to correctly implement the disposable pattern.
121	Public Sub Dispose() Implements IDisposable.Dispose
122	' Do not change this code.  Put cleanup code in Dispose(disposing As Boolean) above.
123	Dispose(True)
124	' TODO: uncomment the following line if Finalize() is overridden above.
125	' GC.SuppressFinalize(Me)
126	End Sub
127	#End Region
128	End Class
129
130	Public Class ProtocolInvoker(Of T As {IComparable, IFormattable, IConvertible})
131
132	Public ReadOnly Property Protocol As New Protocol(GetType(T))
133	Public ReadOnly Property TcpRequest As TcpRequest
134	Public ReadOnly Property TextEncoding As DefaultValue(Of Encoding)
135
136	<MethodImpl(MethodImplOptions.AggressiveInlining)>
137	Sub New(hostName$, remotePort%, Optional encoding As Encodings = Encodings.UTF8WithoutBOM)
138	TcpRequest = New TcpRequest(hostName, remotePort)
139	TextEncoding = encoding.CodePage
140	End Sub
141
142	<MethodImpl(MethodImplOptions.AggressiveInlining)>
143	Public Function SendMessage(protocol As T, message$, Optional textEncoding As Encoding = Nothing) As RequestStream
144	Return SendMessage(protocol, (textEncoding Or Me.TextEncoding).GetBytes(message))
145	End Function
146
147	Public Function SendMessage(protocol As T, data As Byte()) As RequestStream
148	Dim category& = Me.Protocol.EntryPoint
149	Dim protocolL& = CLng(CObj(protocol))
150	Dim message As New RequestStream(category, protocolL, data)
151
152	Return TcpRequest.SendMessage(message)
153	End Function
154
155	<MethodImpl(MethodImplOptions.AggressiveInlining)>
156	Public Function SendMessage(Of V As {New, Class})(protocol As T, data As V) As RequestStream
157	Return SendMessage(protocol, data.GetJson)
158	End Function
159	End Class
160	End Namespace