Class: Aerospike::ExecuteTask

Inherits:
Task
  • Object
show all
Defined in:
lib/aerospike/task/execute_task.rb

Overview

ExecuteTask is used to poll for long running server execute job completion.

Instance Method Summary collapse

Methods inherited from Task

#completed?, #wait_till_completed

Constructor Details

#initialize(cluster, statement) ⇒ ExecuteTask

NewExecuteTask initializes task with fields needed to query server nodes.



22
23
24
25
26
27
28
29
 
# File 'lib/aerospike/task/execute_task.rb', line 22

def initialize(cluster, statement)
  super(cluster, false)

  @task_id = statement.task_id
  @scan = statement.is_scan?

  self
end

Instance Method Details

#all_nodes_done? ⇒ Boolean

queries all nodes for task completion status.



32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
 
# File 'lib/aerospike/task/execute_task.rb', line 32

def all_nodes_done?
  modul = @scan ? "scan" : "query"
  cmd1 = "query-show:trid=#{@task_id}"
  cmd2 = modul + "-show:trid=#{@task_id}"
  cmd3 = "jobs:module=" + modul + ";cmd=get-job;trid=#{@task_id}"

  nodes = @cluster.nodes
  done = false

  nodes.each do |node|
    command = cmd3
    if node.supports_feature?(Aerospike::Features::PARTITION_QUERY)
      command = cmd1
    elsif node.supports_feature?(Aerospike::Features::QUERY_SHOW)
      command = cmd2
    end

    conn = node.get_connection(0)
    responseMap, = Info.request(conn, command)
    node.put_connection(conn)

    response = responseMap[command]
    find = "job_id=#{@task_id}:"
    index = response.index(find)

    unless index
      # don't return on first check
      done = true
      next
    end

    b = index + find.length
    response = response[b, response.length]
    find = "job_status="
    index = response.index(find)

    next unless index

    b = index + find.length
    response = response[b, response.length]
    e = response.index(":")
    status = response[0, e]

    case status
    when "ABORTED"
      raise Aerospike::Exceptions::QueryTerminated
    when "IN PROGRESS"
      return false
    when "DONE"
      done = true
    end
  end

  done
end