nifi_sdk_ruby 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: be48d97b56f690c502c1b8723c4867461ec0798c
-  data.tar.gz: 6a77e0569fb024980fb3fca6c00eb53f04060266
+  metadata.gz: 01032e0711255ad4a4a189e1298ff17694276da4
+  data.tar.gz: f579d3be582ce5a2d4d24b33b9930b0ad7cc0042
 SHA512:
-  metadata.gz: a9edf3dbce0917300ed3fbb1e44e7274656d1623ecca087d7f55a917c49676c5f7757f1e24ee05b10a37a0ab3b76c41f989ee71d6299219e1d76114107bd8793
-  data.tar.gz: 1fd9b45e1e7dea96ece0f062cacd638bf46cab9a319ec46c14e214cd3a9eec92c67881491f4533966e02c65ffd1d891acf20e7a357b3477eba65879c1a2e7f2e
+  metadata.gz: c640f80758ad0f6fca9beb8b8387b1bd93edfed2fa517f5651548af9e5e34086b54351a873377ea1ff248f5675ba898adcc0a79cc9af9fc36d110c0d2694dc94
+  data.tar.gz: 5e157bddbda327b4bfd59402bfba7cdd8485e627a53a115f94d55bc0282c2269910efd93ee12b0508d2b295a935b20a47f093d49e0a2dcf3051e12c4319e03a8

data/.gitignore

@@ -7,6 +7,7 @@
 /pkg/*
 /spec/reports/
 /tmp/
+.idea/*
 
 # rspec failure tracking
 .rspec_status

data/examples/basic.rb

@@ -1,6 +1,6 @@
 require 'pp'
 
-require '../lib/nifi_sdk_ruby'
+require_relative '../lib/nifi_sdk_ruby'
 
 nifi_client = Nifi.new()
 nifi_client.set_debug true
@@ -40,6 +40,6 @@ puts "\n"
 puts nifi_client.delete_process_group new_pg['id']
 puts "\n"
 
-# Upload a template to Root Process Group 
+# Upload a template to Root Process Group
 puts 'Upload template to Root PG'
 puts nifi_client.upload_template(:path => 'IN.hmStaff.taskStatus.xml')

data/lib/nifi_sdk_ruby/version.rb

@@ -1,3 +1,3 @@
 module NifiSdkRuby
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

data/lib/nifi_sdk_ruby.rb

@@ -50,7 +50,7 @@ class Nifi
     end
 
     if !(!!debug == debug)
-      abort 'debug msut be a boolean'
+      abort 'debug must be a boolean'
     end
 
     @@debug = debug
@@ -66,7 +66,7 @@ class Nifi
     end
 
     if !(!!async == async)
-      abort 'async msut be a boolean'
+      abort 'async must be a boolean'
     end
 
     @@async = async
@@ -149,6 +149,45 @@ class Nifi
     end
   end
 
+  def get_process(id = nil)
+    if id.nil?
+      abort 'id is mandatory.'
+    end
+
+    url = @@base_url + '/processors/' + id
+    self.class.http_client(url)
+  end
+
+  def start_process(*args)
+    args = args.reduce Hash.new, :merge
+    if args[:id].nil? or args[:version].nil?
+      abort 'id and version params are mandatory'
+    end
+
+    id = args[:id]
+    version = args[:version]
+
+    params = '{"revision":{"version":'+ version +'},"id":"'+ id +'","component":{"id":"'+ id +
+        '","state":"RUNNING"},"status":{"runStatus":"Running"}}'
+    base_url = @@base_url + '/processors/' + id
+    self.class.http_client(base_url, 'PUT', params)
+  end
+
+  def stop_process(*args)
+    args = args.reduce Hash.new, :merge
+    if args[:id].nil? or args[:version].nil?
+      abort 'id and version params are mandatory'
+    end
+
+    id = args[:id]
+    version = args[:version]
+
+    params = '{"revision":{"version":'+ version +'},"id":"'+ id +'","component":{"id":"'+ id +
+        '","state":"STOPPED"},"status":{"runStatus": "Stopped"}}'
+    base_url = @@base_url + '/processors/' + id
+    self.class.http_client(base_url, 'PUT', params)
+  end
+
   def process_group_by_name?(name = nil)
 
     if name.nil?
@@ -164,6 +203,29 @@ class Nifi
     pg.count == 1 ? true : false
   end
 
+  def create_template_instance(*args)
+    args = args.reduce Hash.new, :merge
+
+    if args[:id].nil? and args[:name].nil?
+      abort 'either specify id of the template or it\'s name '
+    end
+
+    if args[:name]
+      abort "Could not find template called #{args[:name]}" unless template_by_name?(args[:name])
+      id = get_template_by_name(args[:name])[0][0]
+    else
+      abort "Could not find template with id #{args[:id]}" unless template_by_id?(args[:id])
+      id = args[:id]
+    end
+
+    originX = args[:originX] ? args[:originX] : '0.0'
+    originY = args[:originY] ? args[:originY] : '0.0'
+    process_group = args[:process_group_id] ? args[:process_group_id] : 'root'
+    params = '{"templateId": "'+ id +'", "originX": '+ originX +', "originY": '+ originY +'}'
+    base_url = @@base_url + "/process-groups/#{process_group}/template-instance"
+    self.class.http_client(base_url, 'POSTRAW', params)
+  end
+
   def upload_template(*args)
 
     args = args.reduce Hash.new, :merge
@@ -249,6 +311,21 @@ class Nifi
     pg.count == 1 ? true : false
   end
 
+  def template_by_id?(id = nil)
+
+    if id.nil?
+      abort 'id is mandatory.'
+    end
+
+    res = self.class.exists
+
+    pg = res.select do |r|
+      r['identifier'] == "/templates/#{id}"
+    end
+
+    pg.count == 1 ? true : false
+  end
+
   private
 
   def self.exists
@@ -285,6 +362,7 @@ class Nifi
         c.multipart_form_post = true
         c.post(params)
       when 'PUT'
+        c.headers['Content-Type'] = 'application/json'
         c.put(params)
       when 'DELETE'
         c.delete
@@ -305,3 +383,4 @@ class Nifi
     end
   end
 end
+

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: nifi_sdk_ruby
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Israel Calvete
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-04-26 00:00:00.000000000 Z
+date: 2017-05-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httparty
@@ -142,7 +142,6 @@ files:
 - lib/nifi_sdk_ruby.rb
 - lib/nifi_sdk_ruby/version.rb
 - nifi_sdk_ruby.gemspec
-- pkg/.gitkeep
 - release.md
 homepage: http://rubygems.org/gems/nifi-sdk-ruby
 licenses: