net/server.lua

   1
   2 local use_luaevent = require "core.configmanager".get("*", "core", "use_libevent");
   3
   4 if use_luaevent then
   5         use_luaevent = pcall(require, "luaevent.core");
   6         if not use_luaevent then
   7                 log("error", "libevent not found, falling back to select()");
   8         end
   9 end
  10
  11 local server;
  12
  13 if use_luaevent then
  14         server = require "net.server_event";
  15         -- util.timer requires "net.server", so instead of having
  16         -- Lua look for, and load us again (causing a loop) - set this here
  17         -- (usually it isn't set until we return, look down there...)
  18         package.loaded["net.server"] = server;
  19
  20         -- Backwards compatibility for timers, addtimer
  21         -- called a function roughly every second
  22         local add_task = require "util.timer".add_task;
  23         function server.addtimer(f)
  24                 return add_task(1, function (...) f(...); return 1; end);
  25         end
  26
  27         -- Overwrite signal.signal() because we need to ask libevent to
  28         -- handle them instead
  29         local ok, signal = pcall(require, "util.signal");
  30         if ok and signal then
  31                 local _signal_signal = signal.signal;
  32                 function signal.signal(signal_id, handler)
  33                         if type(signal_id) == "string" then
  34                                 signal_id = signal[signal_id:upper()];
  35                         end
  36                         if type(signal_id) ~= "number" then
  37                                 return false, "invalid-signal";
  38                         end
  39                         return server.hook_signal(signal_id, handler);
  40                 end
  41         end
  42 else
  43         server = require "net.server_select";
  44         package.loaded["net.server"] = server;
  45 end
  46
  47 -- require "net.server" shall now forever return this,
  48 -- ie. server_select or server_event as chosen above.
  49 return server;