Class: SageMaker

Inherits:
Mapper show all
Defined in:
lib/aws_recon/collectors/sagemaker.rb

Overview

Collect SageMaker Resources

Constant Summary

Constants inherited from Mapper

Mapper::SINGLE_REGION_SERVICES

Instance Method Summary collapse

Methods inherited from Mapper

#initialize

Constructor Details

This class inherits a constructor from Mapper

Instance Method Details

#collect ⇒ Object

Returns an array of resources.



10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/aws_recon/collectors/sagemaker.rb', line 10

def collect
  resources = []

  #
  # list_notebook_instances
  #
  @client.list_notebook_instances.each_with_index do |response, page|
    log(response.context.operation_name, page)

    response.notebook_instances.each do |instance|
      struct = OpenStruct.new(@client.describe_notebook_instance({
                                                                   notebook_instance_name: instance.notebook_instance_name
                                                                 }).to_h)
      struct.type = 'notebook_instance'
      struct.arn = instance.notebook_instance_arn

      resources.push(struct.to_h)
    end
  end

  #
  # list_endpoints
  #
  @client.list_endpoints.each_with_index do |response, page|
    log(response.context.operation_name, page)

    response.endpoints.each do |instance|
      struct = OpenStruct.new(@client.describe_endpoint({
                                                          endpoint_name: instance.endpoint_name
                                                        }).to_h)
      struct.type = 'endpoint'
      struct.arn = instance.endpoint_arn

      resources.push(struct.to_h)
    end
  end

  resources
end