shogi_server.rb

   1 ## $Id$
   2
   3 ## Copyright (C) 2004 NABEYA Kenichi (aka nanami@2ch)
   4 ## Copyright (C) 2007-2008 Daigo Moriwaki (daigo at debian dot org)
   5 ##
   6 ## This program is free software; you can redistribute it and/or modify
   7 ## it under the terms of the GNU General Public License as published by
   8 ## the Free Software Foundation; either version 2 of the License, or
   9 ## (at your option) any later version.
  10 ##
  11 ## This program is distributed in the hope that it will be useful,
  12 ## but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 ## GNU General Public License for more details.
  15 ##
  16 ## You should have received a copy of the GNU General Public License
  17 ## along with this program; if not, write to the Free Software
  18 ## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19
  20 require 'kconv'
  21 require 'getoptlong'
  22 require 'thread'
  23 require 'timeout'
  24 require 'socket'
  25 require 'yaml'
  26 require 'yaml/store'
  27 require 'digest/md5'
  28 require 'webrick'
  29 require 'fileutils'
  30 require 'logger'
  31
  32 require 'shogi_server/board'
  33 require 'shogi_server/game'
  34 require 'shogi_server/league'
  35 require 'shogi_server/login'
  36 require 'shogi_server/move'
  37 require 'shogi_server/piece'
  38 require 'shogi_server/player'
  39 require 'shogi_server/timeout_queue'
  40 require 'shogi_server/usi'
  41 require 'shogi_server/util'
  42 require 'shogi_server/command'
  43 require 'shogi_server/buoy'
  44
  45 module ShogiServer # for a namespace
  46
  47 Max_Identifier_Length = 32
  48 Default_Timeout = 60            # for single socket operation
  49 Default_Game_Name = "default-1500-0"
  50 One_Time = 10
  51 Least_Time_Per_Move = 1
  52 Login_Time = 300                # time for LOGIN
  53 Release  = "$Id$"
  54 Revision = (r = /Revision: (\d+)/.match("$Revision$") ? r[1] : 0)
  55
  56 RELOAD_FILES = ["shogi_server/league/floodgate.rb",
  57                 "shogi_server/league/persistent.rb",
  58                 "shogi_server/pairing.rb"]
  59 BASE_DIR = File.expand_path(File.dirname(__FILE__))
  60
  61 def reload
  62   RELOAD_FILES.each do |f|
  63     load File.join(BASE_DIR, f)
  64   end
  65 end
  66 module_function :reload
  67
  68 class Logger < ::Logger
  69   @@mkdir_mutex = Mutex.new
  70
  71   def initialize(logdev, shift_age = 0, shift_size = 1048576)
  72     super
  73     class << @logdev
  74       def shift_log_period(now)
  75         age_file = age_file_name(now)
  76         move_age_file_in_the_way(age_file)
  77         mkdir_for(age_file)
  78
  79         @dev.close
  80         rename_file(@filename, age_file)
  81         @dev = create_logfile(@filename)
  82         return true
  83       end
  84
  85       def age_file_name(time)
  86         postfix = previous_period_end(time).strftime("%Y%m%d")  # YYYYMMDD
  87         age_file = File.join(
  88                      File.dirname(@filename),
  89                      postfix[0..3], # YYYY
  90                      postfix[4..5], # MM
  91                      postfix[6..7], # DD
  92                      File.basename(@filename))
  93         return age_file
  94       end
  95
  96       def age_file_exists?(age_file)
  97         return FileTest.exist?(age_file)
  98       end
  99
 100       def rename_file(old_file, new_file)
 101         File.rename(old_file, new_file)
 102       end
 103
 104       def move_age_file_in_the_way(age_file)
 105         return unless age_file_exists?(age_file)
 106
 107         now = Time.now
 108         new_file = "%s.%s%06d"  % [age_file, now.strftime("%Y%m%d%H%M%S"), now.usec]
 109         @dev.write("[WARN] An existing '#{age_file}' is beeing moved to '#{new_file}'\n")
 110         rename_file(age_file, new_file)
 111       end
 112
 113       def mkdir_for(path)
 114         @@mkdir_mutex.synchronize do
 115           unless FileTest.directory?(File.dirname(path))
 116             begin
 117               FileUtils.mkdir_p File.dirname(path)
 118             rescue
 119               @dev.write("[ERROR] Could not create a directory: %s\n" % [File.dirname(path)])
 120             end
 121           end
 122         end # mutex
 123       end
 124
 125     end # @logdev
 126   end
 127
 128 end # class Logger
 129
 130 class Formatter < ::Logger::Formatter
 131   def initialize
 132     super
 133     @datetime_format = "%Y-%m-%dT%H:%M:%S"
 134   end
 135
 136   def call(severity, time, progname, msg)
 137     %!%s [%s] %s\n! % [format_datetime(time), severity, msg2str(msg)]
 138   end
 139 end
 140
 141 end # module ShogiServer