PigNode view

PigNode.rb

class PigNode < Mustache
  attr_accessor :properties
  def initialize properties
    @properties = properties
  end
  
  def operation
    properties['operation']
  end

  def additional_info
    if properties['macro']
      return " (MACRO: #{properties['macro']})"      
    end
    if properties['limit']
      return " (#{properties['limit']})"
    end
    if properties['join'] &&
        properties['join']['strategy'] &&
        properties['join']['type']
      strategy = properties['join']['strategy']
      type     = properties['join']['type']
      return " (#{type}, #{strategy})"
    end    
  end

  def step_type_color
    case properties['step_type']
    when 'mapper' then
      return "#3299BB"
    when 'reducer' then
      return "#FF9900"
    when 'tez' then
      return "#F5D04C"
    else
      return "#BF0A0D"
    end    
  end
  
  def alias
    properties['alias']
  end
  
  def expression
    properties['expression']
  end

  def storage_location
    properties['storage_location']
  end

  def storage_function
    properties['storage_function']
  end

  def join
    j = properties['join']
    if j && j['by']
      relations = j['by'].map{|rel| rel['alias']}
      fields    = []
      
      j['by'].first['fields'].each_with_index do |field, idx|
        fields[idx] = []
        j['by'].each do |rel|
          fields[idx] << rel['fields'][idx]
        end        
      end
      
      result = {
        :relations => relations,
        :by => fields.map{|field_list| {:fields => field_list}}         
      }
      return result
    end    
  end

  def schema
    properties['schema']
  end
end