Module: TaskJuggler::ProcessIntercom

Includes:
MessageHandler
Included in:
ProjectServer, ReportServer
Defined in:
lib/taskjuggler/daemon/ProcessIntercom.rb

Instance Method Summary collapse

Methods included from MessageHandler

#critical, #debug, #error, #fatal, #info, #warning

Instance Method Details

#checkKey(authKey, command) ⇒ Object 



127
128
129
130
131
132
133
134
135
136
137
 
# File 'lib/taskjuggler/daemon/ProcessIntercom.rb', line 127

def checkKey(authKey, command)
  if authKey == @authKey
    debug('', "Accepted authentication key for command '#{command}'")
  else
    warning('auth_key_rejected',
            "Rejected wrong authentication key #{authKey}" +
            "for command '#{command}'")
    return false
  end
  true
end

#connect(stdout, stderr, stdin, silent) ⇒ Object 



93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
 
# File 'lib/taskjuggler/daemon/ProcessIntercom.rb', line 93

def connect(stdout, stderr, stdin, silent)
  # Set the client lock.
  @clientConnection.lock
  debug('', 'Rerouting ProjectServer standard IO to client')
  # Make sure that all output to STDOUT and STDERR is sent to the client.
  # Input is read from the client STDIN.  We save a copy of the old file
  # handles so we can restore then later again.
  @stdout = $stdout
  @stderr = $stderr
  @stdin = $stdin
  $stdout = stdout if stdout
  $stderr = stderr if stdout
  $stdin = stdin if stdin
  debug('', 'IO is now routed to the client')
  Log.silent = silent
  true
end

#disconnect ⇒ Object 



111
112
113
114
115
116
117
118
119
120
121
 
# File 'lib/taskjuggler/daemon/ProcessIntercom.rb', line 111

def disconnect
  debug('', 'Restoring IO')
  Log.silent = true
  $stdout = @stdout if @stdout
  $stderr = @stderr if @stderr
  $stdin = @stdin if @stdin
  debug('', 'Standard IO has been restored')
  # Release the client lock
  @clientConnection.unlock
  true
end

#generateAuthKey ⇒ Object 



123
124
125
 
# File 'lib/taskjuggler/daemon/ProcessIntercom.rb', line 123

def generateAuthKey
  rand(1000000000).to_s
end

#initIntercom ⇒ Object 



71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
 
# File 'lib/taskjuggler/daemon/ProcessIntercom.rb', line 71

def initIntercom
  # This is the authentication key that clients will need to provide to
  # execute DRb methods.
  @authKey = generateAuthKey

  # This flag will be set to true by DRb method calls to terminate the
  # process.
  @terminate = false

  # This mutex is locked while a client is connected.
  @clientConnection = Mutex.new
  # This lock protects the @timerStart
  @timeLock = Monitor.new
  # The time stamp of the last client interaction.
  @timerStart = nil
end

#restartTimer ⇒ Object

This function must be called after each client interaction to restart the client connection timer.



141
142
143
144
145
146
 
# File 'lib/taskjuggler/daemon/ProcessIntercom.rb', line 141

def restartTimer
  @timeLock.synchronize do
    debug('', 'Reseting client connection timer')
    @timerStart = Time.new
  end
end

#startTerminator ⇒ Object

This method starts a new thread and waits for the @terminate variable to be true. If that happens, it waits for the @clientConnection lock or forces an exit after the timeout has been reached. It shuts down the DRb server.



162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
 
# File 'lib/taskjuggler/daemon/ProcessIntercom.rb', line 162

def startTerminator
  Thread.new do
    loop do
      if @terminate
        # We wait for the client to propery disconnect. In case this does
        # not happen, we'll wait for the timeout and exit anyway.
        restartTimer
        while @clientConnection.locked? && !timerExpired? do
          sleep 1
        end
        if timerExpired?
          warning('drb_timeout_shutdown',
                  'Shutting down DRb server due to timeout')
        else
          debug('', 'Shutting down the DRb server')
        end
        DRb.stop_service
        break
      else
        sleep 1
      end
    end
  end
end

#terminate ⇒ Object 



88
89
90
91
 
# File 'lib/taskjuggler/daemon/ProcessIntercom.rb', line 88

def terminate
  debug('', 'Terminating on external request')
  @terminate = true
end

#timerExpired? ⇒ Boolean

Check if the client interaction timer has already expired.

Returns:

  • (Boolean) 


149
150
151
152
153
154
155
156
 
# File 'lib/taskjuggler/daemon/ProcessIntercom.rb', line 149

def timerExpired?
  res = nil
  @timeLock.synchronize do
    # We should see client interaction every 2 minutes.
    res = (Time.new > @timerStart + 2 * 60)
  end
  res
end