unageanu-jiji 0.1.0

data/lib/jiji/service/hello_service.rb

@@ -0,0 +1,10 @@
+
+module JIJI
+  module Service
+	  class HelloService
+	    def hello
+	      "hello."
+	    end
+	  end
+  end
+end

data/lib/jiji/service/output_service.rb

@@ -0,0 +1,58 @@
+module JIJI
+  module Service
+    class OutputService
+
+      # 指定範囲の出力データを取得する。
+      def list_datas( process_id, names, scale, start_time, end_time )
+        p = process_manager.get( process_id )
+        list = {}
+        names.each {|n|
+	        buff = []
+          outputs = p.agent_manager.get(n[0]).output
+	        outputs.get(n[1]).each( scale,
+            Time.at(start_time), Time.at(end_time) ) {|data|
+	          buff << data
+	        }
+          list[n[0]] = {} if !list[n[0]]
+          list[n[0]][n[1]] = buff
+        }
+        return list
+      end
+
+      # プロセスの出力一覧を取得する
+      def list_outputs( process_id )
+        p = process_manager.get( process_id )
+        return p.agent_manager.inject({}) {|buff,item|
+          buff[item[0]] = item[1].output.inject({}) {|r,v|
+            r[v[0]] = v[1].options
+            r
+          }
+          buff
+        }
+      end
+
+      # アウトプットのプロパティを設定
+      def set_properties( process_id, name, properties )
+        p = process_manager.get( process_id )
+        outputs = p.agent_manager.get(name[0]).output
+        outputs.get(name[1]).set_properties( properties )
+        return :success
+      end
+
+      # プロセスのログを取得する
+      def get_log( process_id )
+        process_manager.get( process_id ) # プロセスの存在確認
+        file = "#{@process_dir}/#{process_id}/log.txt" # TODO 古いものも連結するか? サイズが大きくなってしまうので微妙。
+        if ( File.exist?( file ) ) 
+          return IO.read( file )
+        else
+          return ""
+        end
+      end
+      
+      attr :process_dir, true
+      attr :process_manager, true
+	  end
+
+  end
+end

data/lib/jiji/service/process_service.rb

@@ -0,0 +1,78 @@
+module JIJI
+  module Service
+    class ProcessService
+
+      # プロセス一覧を取得する。
+      def list_test( )
+        buff = []
+        @process_manager.each {|p|
+          buff << extract_properties( p )
+        }
+        return buff.sort_by {|item| item["create_date"] * -1 }
+      end
+
+      # プロセスを取得する。
+      def get( process_id )
+        p = @process_manager.get( process_id )
+        process_info = extract_properties( p )
+        process_info["agents"] = p["agents"] # getの場合はエージェントのプロパティも返す。
+        return process_info
+      end
+
+      # プロセスの設定を更新する。
+      def set( process_id, setting )
+        @process_manager.set( process_id, setting )
+        :success
+      end
+
+
+      # 新しいテストを作成&開始する
+      def new_test( name, memo, start_date, end_date, agents )
+        @process_manager.create_back_test(
+          name, memo, start_date, end_date, agents );
+      end
+
+      # テストの状態を取得する
+      def status( process_ids=[] )
+        process_ids.inject([]) {|buff,id|
+          p = @process_manager.get( id )
+          buff << extract_properties( p )
+        }
+      end
+
+      # テストを削除する
+      def delete_test( process_id )
+        @process_manager.delete_back_test( process_id )
+      end
+
+      # テストの実行をキャンセルする
+      def stop( process_id, agent_properties )
+        @process_manager.get( process_id ).stop
+        :success
+      end
+
+      # テストを再実行する
+      def restart( process_id, agent_properties )
+        @process_manager.restart_test(process_id, agent_properties)
+        :success
+      end
+
+      attr :process_manager, true
+    private
+      def extract_properties( process )
+          return {
+            "id"=>process["id"],
+            "name"=>process["name"],
+            "memo"=>process["memo"],
+            "start_date"=>process["start_date"],
+            "end_date"=>process["end_date"],
+            "create_date"=>process["create_date"],
+            "trade_enable"=>process["trade_enable"],
+            "state"=>process.state,
+            "progress"=>process.progress,
+          }
+      end
+	  end
+  
+  end
+end

data/lib/jiji/service/rate_service.rb

@@ -0,0 +1,33 @@
+module JIJI
+  module Service
+    class RateService
+
+      # 指定範囲のレートを取得する。
+      def list( pair, scale, start_time, end_time )
+        buff = []
+        @rate_dao.each( scale, pair.to_sym, Time.at(start_time), Time.at(end_time) ) {|data|
+          buff << [data[0].to_f, data[1].to_f, data[2].to_f,
+            data[3].to_f,data[16].to_i, data[17].to_i]
+        }
+        return buff
+      end
+
+      # 利用可能なレートの開始日時/終了日時を得る。
+      def range( pair )
+        dao = @rate_dao.dao( pair )
+        { :first=>dao.first_time(:raw).to_i,
+         :last=>dao.last_time(:raw).to_i }
+      end
+
+      # 指定した月で利用可能な日の一覧を得る。
+      def enable( pair, year, month )
+        dao = @rate_dao.dao( pair )
+        list = dao.list_data_files( :raw, "#{year}-#{sprintf("%02d", month)}" )
+        list.map {|d| d[-2,2] }
+      end
+
+      attr :rate_dao, true
+	  end
+
+  end
+end

data/lib/jiji/service/system_service.rb

@@ -0,0 +1,14 @@
+
+require 'jiji/error'
+
+module JIJI
+  module Service
+	  class SystemService
+      # サーバーを停止する。
+      def shutdown
+        @server.shutdown
+      end
+      attr :server, true
+    end
+  end
+end

data/lib/jiji/service/trade_result_service.rb

@@ -0,0 +1,32 @@
+module JIJI
+  module Service
+    class TradeResultService
+      
+      # 指定範囲のトレード結果を取得する。
+      def list( process_id, scale, start_time, end_time )
+        p = process_manager.get( process_id )
+        dao = p.agent_manager.trade_result_dao
+        result = dao.list_positions( scale, start_time ? Time.at(start_time) : nil, end_time ? Time.at(end_time) : nil )
+        # 現在進行中の建て玉はoperatorから取得する
+        op = p.agent_manager.operator
+        return result.map {|e|
+          op && op.positions.key?(e[0]) ? op.positions[e[0]].values : e[1] 
+        }
+      end
+      
+      # 指定範囲の損益を取得する。
+      def list_profit_or_loss( process_id, scale, start_time, end_time )
+        p = process_manager.get( process_id )
+        dao = p.agent_manager.trade_result_dao
+        buff = []
+        dao.each( scale, start_time ? Time.at(start_time) : nil, end_time ? Time.at(end_time) : nil ) {|data|
+          buff << data
+        }
+        return buff
+      end
+      
+      attr :process_manager, true
+	  end
+    
+  end
+end

data/lib/jiji/single_click_client.rb

@@ -0,0 +1,95 @@
+
+require "rubygems"
+require "clickclient"
+require 'jiji/util/block_to_session'
+require 'jiji/util/util'
+require 'jiji/error'
+
+# エラー発生時に実行時例外になる場合があるので修正。
+module ClickClient
+  def self.parse( content )
+    doc = REXML::Document.new( content )
+    unless ( doc.text( "./*/responseStatus" ) =~ /OK/  )
+      error = doc.text( "./*/message" )
+      raise "fail. #{error ? error : content }"
+    end
+    return doc
+  end
+end
+
+
+module JIJI
+
+  #
+  #==クリック証券へのアクセスを集約するためのサービス
+  #
+  class SingleClickClient
+
+
+    def initialize( account, conf, logger )
+      @conf = conf
+      @logger = logger
+      @account = account
+      @session = create_session
+    end
+
+    #リクエストを送付する。
+    #ブロックの第1引数としてセッションが渡される。
+    def request( &block )
+      begin
+        @session.request( &block )
+      rescue
+        # セッション切れの場合、再作成して再実行。
+        if $!.to_s =~ /Out Of Session\./
+          @logger.info "restart single click client."
+          @session.close
+          @session = create_session
+          @session.request( &block )
+        else
+          raise $!
+        end
+      end
+    end
+
+    #サービスを破棄する。
+    #不要になった場合、必ず実行すること。
+    def close
+      @session.close
+      @logger.info "close single click client."
+    end
+    attr :conf, true
+    attr :logger, true
+
+  private
+    def create_session
+      return Session.new { |wait|
+        logger.info "start single click client."
+        begin
+          JIJI::Util.log_if_error_and_throw( @logger ) {
+            client = ClickClient::Client.new
+            client.host_name = @conf.get([:securities,:host], "https://fx-demo.click-sec.com" )
+            @logger.info "connect host=#{client.host_name}, user=#{@account.user}"
+            client.fx_session(@account.user, @account.password){|fx|
+              wait.call( fx )
+            }
+          }
+        rescue Exception
+          @logger.info "connect failed."
+          e = JIJI::FatalError.new( JIJI::ERROR_NOT_CONNECTED, $!.to_s )
+          wait.call( ErrorSession.new( e ) )
+        end
+      }
+    end
+  end
+
+  # 接続エラー時に使用するセッション
+  #常にエラーをスローする。
+  class ErrorSession
+    def initialize(error)
+      @error = error
+    end
+    def method_missing(name, *args, &block)
+      raise @error
+    end
+  end
+end

data/lib/jiji/util/block_to_session.rb

@@ -0,0 +1,70 @@
+
+require 'thread'
+
+# ブロックを渡して処理を行なうAPIを
+# セッション風に使えるようにする
+class Session
+  def initialize
+    @alive = true
+    @alive_mutex = Mutex.new
+    @q = Queue.new
+    @t = Thread.fork {
+      yield proc {|*args|
+        while( @alive_mutex.synchronize { @alive } )
+          req = @q.pop
+          req.call( *args ) if req
+        end
+      }
+    }
+  end
+  # リクエストを送る
+  def request( &block )
+    return unless block_given?
+    req = Request.new(block)
+    @q.push req
+    req.wait
+  end
+  # セッションを破棄する
+  def close
+    @alive_mutex.synchronize { 
+      @alive = false
+    }
+    @q.push nil
+    @t.join
+  end
+  
+  # リクエスト
+  class Request
+    def initialize( block )
+      @mutex = Mutex.new
+      @cv = ConditionVariable.new
+      @finished = false
+      @value = nil
+      @error = nil
+      @proc = proc {|*args|
+        begin
+         @value = block.call(*args)
+        rescue Exception
+          @error = $! 
+        ensure
+         @mutex.synchronize{
+           @finished = true
+           @cv.signal
+          }
+        end
+      }
+    end
+    # リクエストの完了を待ち、結果を返す。
+    def wait
+      @mutex.synchronize{
+        @cv.wait( @mutex ) until @finished
+      }
+      raise @error if @error
+      @value
+    end
+    # リクエストを実行する。
+    def call(*args)
+      @proc.call(*args)
+    end
+  end
+end

data/lib/jiji/util/cache.rb

@@ -0,0 +1,79 @@
+require "fileutils"
+require 'jiji/util/file_lock'
+require 'kconv'
+require 'yaml'
+require 'jiji/util/fix_yaml_bug'
+
+module Cache
+
+  class YAMLCacheInterceptor
+    def initialize( range, cache_file_prefix, &block)
+      @name = "cache"
+      @range = range
+      @cache_file_prefix = cache_file_prefix
+      @block = block
+    end
+    def invoke( mi )
+      cache_file =  @cache_file_prefix + "_" +  mi.name.to_s
+      cache_file += "_" + @block.call(mi.name,*mi.arguments) if @block != nil
+      cache_file += ".yaml"
+
+      flock = FileLock.new(cache_file + ".lock")
+      current = Time.new
+
+      res = nil
+      flock.writelock(){
+        # キャッシュがあればキャッシュから読む
+        if ( File.exist?(cache_file) && current.to_i < (File.mtime(cache_file).to_i + @range) )
+          res = YAML.load_file cache_file
+        else
+          res =  mi.proceed
+          File.open( cache_file, "w" ) { |f|
+            f.write( YAML.dump(res) )
+          }
+          File.utime(current, current, cache_file)
+        end
+      }
+      return res
+    end
+    attr_reader :name
+  end
+
+  class YAMLCache
+    def initialize(delegate, range, cache_file_prefix, regex=".*", &block)
+      @delegate = delegate
+      @range = range
+      @cache_file_prefix = cache_file_prefix
+      @regex = regex
+      @block = block
+    end
+
+    def method_missing(name, *args )
+
+      return @delegate.send( name, *args ) if ( !name.to_s.match(@regex) )
+
+      cache_file =  @cache_file_prefix + "_" +  name.to_s
+      cache_file += "_" + @block.call(name,*args) if @block != nil
+
+      cache_file += ".yaml"
+
+      flock = FileLock.new(cache_file + ".lock")
+      current = Time.new
+
+      res = nil
+      flock.writelock(){
+        # キャッシュがあればキャッシュから読む
+        if ( File.exist?(cache_file) && current.to_i < (File.mtime(cache_file).to_i + @range) )
+          res = YAML.load_file cache_file
+        elsif
+          res =  @delegate.send( name, *args )
+          File.open( cache_file, "w" ) { |f|
+            f.write( YAML.dump(res) )
+          }
+          File.utime(current, current, cache_file)
+        end
+      }
+      return res
+    end
+  end
+end