mlserver 1.0.1 → 1.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 565870637b0850519fc89c56d3aa2dbf8a400d26577440d81b1db195ce4eae78
-  data.tar.gz: 8632c4624dcd90c02a76ba2437a92ca74d3fa4d93c709f3e75b0332f0f5b668b
+  metadata.gz: 86fe66bb2ece6aa1aeb92f25f69b88d266f8743bf9b931aa40bfdd1a22945fba
+  data.tar.gz: f1a46e419ab1b5474006064c28593a14f1f741bc6bdfd95fa68463bb45507346
 SHA512:
-  metadata.gz: beb25988af8aeda7895d8fbc794c1cf9f4c3d6b9efc32ad9886bf09534eff26c525d0d2a5108d34d0d265029524d3c9a4615fc1d36df7ad2420dc0e450ea9d89
-  data.tar.gz: a07ce050c8f7efd559218c5ed1ca9f852aa582cb0186ff35f7ac644e3dedc0849fda84d0cabbf4dbff9bf6a4b4d035f257b7a509eee7e7128b4fa35eac4f3e54
+  metadata.gz: 9e0813a7844e61e75e1fb5bc95271bed3c1f7af903055aa16e8293d0c588669e67c172f05810d3dd7b74d1ca49351a771ef34877b88c2eddb78066aca3166bb9
+  data.tar.gz: 39c3f002aa07a6fd5e36e69748b1a0432d3d6bf7e9d438efd0ff977660d40c4bb6ce4322e8014d05dcd2df01d8bea99390412ec4358edfb1c2e1e522426c94ef

data/bin/mlserver

@@ -11,6 +11,24 @@ s = {}
 x = ArgsParser::Args.new(options: o, switches: s)
 
 conf_file = x.data[0]
-require_relative conf_file
+
+if !conf_file
+  conf_file = "mlserver.conf.rb"
+end
+
+if File.directory?(conf_file)
+  conf_file = "#{conf_file}/mlserver.conf.rb"
+end
+
+if !File.exist?(conf_file)
+  STDERR.puts "Error: Conf file does not exist"
+  exit 1
+end
+
+full_conf_file_path = File.expand_path(conf_file, Dir.pwd)
+
+Dir.chdir(File.dirname full_conf_file_path)
+
+require full_conf_file_path
 
 MLserver::Server.start()

data/lib/MLserver/logger.rb

@@ -1,12 +1,13 @@
 module MLserver
   class Logger
-    @@log_levels = [:info, :warn, :error]
+    @@log_levels = [:info, :warn, :error, :traffic_in, :traffic_out]
 
-    def initialize(out:, err:, log_colors: {}, outputs: {error: :err})
+    def initialize(out:, err:, log_colors: {}, outputs: {error: :err}, levels_with_timestamp: [:traffic_in, :traffic_out])
       @out = out
       @err = err
       @log_colors = log_colors
       @outputs = outputs
+      @levels_with_timestamp = levels_with_timestamp
     end
 
     def log(message, level = :info)
@@ -16,6 +17,9 @@ module MLserver
       if @log_colors[level]
         message = message.color @log_colors[level]
       end
+
+      message = "#{Time.now} | #{message}" if @levels_with_timestamp.include?(level)
+
       out.puts message
     end
 
@@ -38,13 +42,15 @@ module MLserver
     def log_traffic(ip, direction, data)
       symbol = (direction == :incoming ? "=>" : "<=")
 
-      log("#{format_ip_address ip} #{symbol} #{data}")
+      log("#{format_ip_address ip} #{symbol} #{data}", (direction == :incoming ? :traffic_in : :traffic_out))
     end
   end
 
   dlcolors = {
     warn: :yellow,
-    error: :red
+    error: :red,
+    traffic_in: :green,
+    traffic_out: :blue
   }
 
   DefaultLogger = Logger.new(out: STDOUT, err: STDERR, log_colors: dlcolors)

data/lib/MLserver.rb

@@ -1,6 +1,6 @@
 module MLserver
   def self.version
-    "1.0.1"
+    "1.0.2"
   end
 end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mlserver
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.0.2
 platform: ruby
 authors:
 - Matthias Lee
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-05-29 00:00:00.000000000 Z
+date: 2023-06-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: argparse