File: redshift.rb

package info (click to toggle)
ruby-fog-aws 3.18.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 8,140 kB
  • sloc: ruby: 73,328; javascript: 14; makefile: 9; sh: 4
file content (131 lines) | stat: -rw-r--r-- 5,011 bytes parent folder | download | duplicates (5)
1
2
3
4
5
6
7
8
9
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
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
module Fog
  module AWS
    class Redshift < Fog::Service
      extend Fog::AWS::CredentialFetcher::ServiceMethods

      requires :aws_access_key_id, :aws_secret_access_key
      recognizes :region, :host, :path, :port, :scheme, :persistent, :use_iam_profile, :aws_session_token, :aws_credentials_expire_at, :instrumentor, :instrumentor_name

      request_path 'fog/aws/requests/redshift'

      request :describe_clusters
      request :describe_cluster_parameter_groups
      request :describe_cluster_parameters
      request :describe_cluster_security_groups
      request :describe_cluster_snapshots
      request :describe_cluster_subnet_groups
      request :describe_cluster_versions
      request :describe_default_cluster_parameters
      request :describe_events
      request :describe_orderable_cluster_options
      request :describe_reserved_node_offerings
      request :describe_reserved_nodes
      request :describe_resize
      request :create_cluster
      request :create_cluster_parameter_group
      request :create_cluster_security_group
      request :create_cluster_snapshot
      request :create_cluster_subnet_group
      request :modify_cluster
      request :modify_cluster_parameter_group
      request :modify_cluster_subnet_group
      request :delete_cluster
      request :delete_cluster_parameter_group
      request :delete_cluster_security_group
      request :delete_cluster_snapshot
      request :delete_cluster_subnet_group
      request :authorize_cluster_security_group_ingress
      request :authorize_snapshot_access
      request :copy_cluster_snapshot
      request :purchase_reserved_node_offering
      request :reboot_cluster
      request :reset_cluster_parameter_group
      request :restore_from_cluster_snapshot
      request :revoke_cluster_security_group_ingress
      request :revoke_snapshot_access

      class Mock
        def initialize(options={})
          Fog::Mock.not_implemented
        end
      end

      class Real
        include Fog::AWS::CredentialFetcher::ConnectionMethods
        # Initialize connection to Redshift
        #
        # ==== Notes
        # options parameter must include values for :aws_access_key_id and
        # :aws_secret_access_key in order to create a connection
        #
        # ==== Examples
        #   ses = SES.new(
        #    :aws_access_key_id => your_aws_access_key_id,
        #    :aws_secret_access_key => your_aws_secret_access_key
        #   )
        #
        # ==== Parameters
        # * options<~Hash> - config arguments for connection.  Defaults to {}.
        #   * region<~String> - optional region to use. For instance, 'us-east-1' and etc.
        #
        # ==== Returns
        # * Redshift object with connection to AWS.

        def initialize(options={})
          @use_iam_profile = options[:use_iam_profile]
          @region = options[:region] || 'us-east-1'
          setup_credentials(options)

          @instrumentor       = options[:instrumentor]
          @instrumentor_name  = options[:instrumentor_name] || 'fog.aws.redshift'
          @connection_options     = options[:connection_options] || {}
          @host = options[:host] || "redshift.#{@region}.amazonaws.com"
          @version = '2012-12-01'
          @path       = options[:path]        || '/'
          @persistent = options[:persistent]  || false
          @port       = options[:port]        || 443
          @scheme     = options[:scheme]      || 'https'

          @connection = Fog::XML::Connection.new("#{@scheme}://#{@host}:#{@port}#{@path}", @persistent, @connection_options)
        end

        private
        def setup_credentials(options)
          @aws_access_key_id      = options[:aws_access_key_id]
          @aws_secret_access_key  = options[:aws_secret_access_key]
          @aws_session_token      = options[:aws_session_token]
          @aws_credentials_expire_at = options[:aws_credentials_expire_at]

          @signer = Fog::AWS::SignatureV4.new( @aws_access_key_id, @aws_secret_access_key,@region,'redshift')
        end

        def request(params, &block)
          refresh_credentials_if_expired

          parser = params.delete(:parser)
          date   = Fog::Time.now
          params[:headers]['Date'] = date.to_date_header
          params[:headers]['x-amz-date'] = date.to_iso8601_basic

          params[:headers]['Host'] = @host
          params[:headers]['x-amz-redshift-version'] = @version
          params[:headers]['x-amz-security-token'] = @aws_session_token if @aws_session_token
          params[:headers]['Authorization'] = @signer.sign params, date
          params[:parser] = parser

          if @instrumentor
            @instrumentor.instrument("#{@instrumentor_name}.request", params) do
              _request(params, &block)
            end
          else
            _request(params, &block)
          end
        end

        def _request(params, &block)
          @connection.request(params, &block)
        end
      end
    end
  end
end