HttpHandler.cs 9.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256
  1. using BeetleX;
  2. using BeetleX.Buffers;
  3. using BeetleX.EventArgs;
  4. using SpanJson;
  5. using System;
  6. using System.Collections.Generic;
  7. using System.Text;
  8. using System.Threading.Tasks;
  9. namespace PlatformBenchmarks
  10. {
  11. public partial class HttpHandler : ServerHandlerBase
  12. {
  13. private static AsciiString _line = new AsciiString("\r\n");
  14. private static AsciiString _2line = new AsciiString("\r\n\r\n");
  15. private static AsciiString _httpsuccess = new AsciiString("HTTP/1.1 200 OK\r\n");
  16. private static readonly AsciiString _headerServer = "Server: Beetlex\r\n";
  17. private static readonly AsciiString _headerContentLength = "Content-Length: ";
  18. private static readonly AsciiString _headerContentLengthZero = "Content-Length: 0\r\n";
  19. private static readonly AsciiString _headerContentTypeText = "Content-Type: text/plain\r\n";
  20. private static readonly AsciiString _headerContentTypeHtml = "Content-Type: text/html; charset=UTF-8\r\n";
  21. private static readonly AsciiString _headerContentTypeJson = "Content-Type: application/json\r\n";
  22. private static readonly AsciiString _path_Json = "/json";
  23. private static readonly AsciiString _path_Db = "/db";
  24. private static readonly AsciiString _path_Queries = "/queries";
  25. private static readonly AsciiString _path_Plaintext = "/plaintext";
  26. private static readonly AsciiString _path_Updates = "/updates";
  27. private static readonly AsciiString _path_Fortunes = "/fortunes";
  28. private static readonly AsciiString _result_plaintext = "Hello, World!";
  29. private static byte _Space = 32;
  30. private static byte _question = 63;
  31. private NextQueueGroup NextQueueGroup;
  32. public HttpHandler()
  33. {
  34. int threads = System.Math.Min(Environment.ProcessorCount, 16);
  35. NextQueueGroup = new NextQueueGroup(threads);
  36. }
  37. public void Default(ReadOnlySpan<byte> url, PipeStream stream, HttpToken token, ISession session)
  38. {
  39. stream.Write("<b> beetlex server</b><hr/>");
  40. stream.Write($"{Encoding.ASCII.GetString(url)} not found!");
  41. OnCompleted(stream, session, token);
  42. }
  43. public override void Connected(IServer server, ConnectedEventArgs e)
  44. {
  45. base.Connected(server, e);
  46. e.Session.Socket.NoDelay = true;
  47. var token = new HttpToken();
  48. token.Db = new RawDb(new ConcurrentRandom(), Npgsql.NpgsqlFactory.Instance);
  49. token.NextQueue = NextQueueGroup.Next();
  50. e.Session.Tag = token;
  51. }
  52. private int AnalysisUrl(ReadOnlySpan<byte> url)
  53. {
  54. for (int i = 0; i < url.Length; i++)
  55. {
  56. if (url[i] == _question)
  57. return i;
  58. }
  59. return -1;
  60. }
  61. class RequestWork : IEventWork
  62. {
  63. public void Dispose()
  64. {
  65. }
  66. public PipeStream Stream { get; set; }
  67. public ISession Session { get; set; }
  68. public HttpToken Token { get; set; }
  69. public HttpHandler Handler { get; set; }
  70. public Task Execute()
  71. {
  72. Handler.OnProcess(Stream, Token, Session);
  73. return Task.CompletedTask;
  74. }
  75. }
  76. private void OnProcess(PipeStream pipeStream,HttpToken token,ISession sessino)
  77. {
  78. var line = _line.AsSpan();
  79. int len = (int)pipeStream.FirstBuffer.Length;
  80. var receiveData = pipeStream.FirstBuffer.Memory.Span;
  81. ReadOnlySpan<byte> http= line;
  82. ReadOnlySpan<byte> method= line;
  83. ReadOnlySpan<byte> url= line;
  84. int offset2 = 0;
  85. int count = 0;
  86. for(int i=0;i<len;i++)
  87. {
  88. if(receiveData[i]==line[0])
  89. {
  90. http = receiveData.Slice(offset2, i - offset2);
  91. break;
  92. }
  93. else
  94. {
  95. if (receiveData[i] == _Space)
  96. {
  97. if (count != 0)
  98. {
  99. url = receiveData.Slice(offset2, i - offset2);
  100. offset2 = i + 1;
  101. }
  102. else
  103. {
  104. method = receiveData.Slice(offset2, i - offset2);
  105. offset2 = i + 1;
  106. count++;
  107. }
  108. }
  109. }
  110. }
  111. OnStartLine(http, method, url, sessino, token, pipeStream);
  112. }
  113. public override void SessionReceive(IServer server, SessionReceiveEventArgs e)
  114. {
  115. base.SessionReceive(server, e);
  116. PipeStream pipeStream = e.Session.Stream.ToPipeStream();
  117. HttpToken token = (HttpToken)e.Session.Tag;
  118. if (Program.Debug)
  119. {
  120. RequestWork work = new RequestWork();
  121. work.Handler = this;
  122. work.Session = e.Session;
  123. work.Stream = pipeStream;
  124. work.Token = token;
  125. token.NextQueue.Enqueue(work);
  126. }
  127. else
  128. {
  129. OnProcess(pipeStream, token, e.Session);
  130. }
  131. }
  132. public virtual void OnStartLine(ReadOnlySpan<byte> http, ReadOnlySpan<byte> method, ReadOnlySpan<byte> url, ISession session, HttpToken token, PipeStream stream)
  133. {
  134. int queryIndex = AnalysisUrl(url);
  135. ReadOnlySpan<byte> baseUrl = default;
  136. ReadOnlySpan<byte> queryString = default;
  137. if (queryIndex > 0)
  138. {
  139. baseUrl = url.Slice(0, queryIndex);
  140. queryString = url.Slice(queryIndex + 1, url.Length - queryIndex - 1);
  141. }
  142. else
  143. {
  144. baseUrl = url;
  145. }
  146. OnWriteHeader(stream, token);
  147. if (baseUrl.Length == _path_Plaintext.Length && baseUrl.StartsWith(_path_Plaintext))
  148. {
  149. stream.Write(_headerContentTypeText.Data, 0, _headerContentTypeText.Length);
  150. OnWriteContentLength(stream, token);
  151. Plaintext(url, stream, token, session);
  152. }
  153. else if (baseUrl.Length == _path_Json.Length && baseUrl.StartsWith(_path_Json))
  154. {
  155. stream.Write(_headerContentTypeJson.Data, 0, _headerContentTypeJson.Length);
  156. OnWriteContentLength(stream, token);
  157. Json(url, stream, token, session);
  158. }
  159. else if (baseUrl.Length == _path_Db.Length && baseUrl.StartsWith(_path_Db))
  160. {
  161. stream.Write(_headerContentTypeJson.Data, 0, _headerContentTypeJson.Length);
  162. OnWriteContentLength(stream, token);
  163. db(stream, token, session);
  164. }
  165. else if (baseUrl.Length == _path_Queries.Length && baseUrl.StartsWith(_path_Queries))
  166. {
  167. stream.Write(_headerContentTypeJson.Data, 0, _headerContentTypeJson.Length);
  168. OnWriteContentLength(stream, token);
  169. queries(Encoding.ASCII.GetString(queryString), stream, token, session);
  170. }
  171. else if (baseUrl.Length == _path_Updates.Length && baseUrl.StartsWith(_path_Updates))
  172. {
  173. stream.Write(_headerContentTypeJson.Data, 0, _headerContentTypeJson.Length);
  174. OnWriteContentLength(stream, token);
  175. updates(Encoding.ASCII.GetString(queryString), stream, token, session);
  176. }
  177. else if (baseUrl.Length == _path_Fortunes.Length && baseUrl.StartsWith(_path_Fortunes))
  178. {
  179. stream.Write(_headerContentTypeHtml.Data, 0, _headerContentTypeHtml.Length);
  180. OnWriteContentLength(stream, token);
  181. fortunes(stream, token, session);
  182. }
  183. else
  184. {
  185. stream.Write(_headerContentTypeHtml.Data, 0, _headerContentTypeHtml.Length);
  186. OnWriteContentLength(stream, token);
  187. Default(url, stream, token, session);
  188. }
  189. }
  190. private void OnWriteHeader(PipeStream stream, HttpToken token)
  191. {
  192. stream.Write(_httpsuccess.Data, 0, _httpsuccess.Length);
  193. stream.Write(_headerServer.Data, 0, _headerServer.Length);
  194. ArraySegment<byte> date = GMTDate.Default.DATE;
  195. stream.Write(date.Array, date.Offset, date.Count);
  196. }
  197. private void OnWriteContentLength(PipeStream stream, HttpToken token)
  198. {
  199. stream.Write(_headerContentLength.Data, 0, _headerContentLength.Length);
  200. token.ContentLength = stream.Allocate(10);
  201. stream.Write(_2line, 0, 4);
  202. token.ContentPostion = stream.CacheLength;
  203. }
  204. private void OnCompleted(PipeStream stream, ISession session, HttpToken token)
  205. {
  206. stream.ReadFree((int)stream.Length);
  207. token.FullLength((stream.CacheLength - token.ContentPostion).ToString());
  208. session.Stream.Flush();
  209. }
  210. }
  211. }