Class: GoodData::LCM2::CreateSegmentMasters

Inherits:
BaseAction show all
Defined in:
lib/gooddata/lcm/actions/create_segment_masters.rb

Constant Summary collapse

DESCRIPTION = 
'Create Master Projects for Segments'
PARAMS = 
define_params(self) do
  description 'Client Used for Connecting to GD'
  param :gdc_gd_client, instance_of(Type::GdClientType), required: true

  description 'Organization Name'
  param :organization, instance_of(Type::StringType), required: true

  description 'ADS Client'
  param :ads_client, instance_of(Type::AdsClientType), required: true

  # description 'Queries Used'
  # param :query, instance_of(Type::ReleaseQueryType), required: false

  description 'Segments to manage'
  param :segments, array_of(instance_of(Type::SegmentType)), required: true

  description 'Tokens'
  param :tokens, instance_of(Type::TokensType), required: true
end
DEFAULT_TABLE_NAME = 
'LCM_RELEASE'

Constants included from Dsl::Dsl

Dsl::Dsl::DEFAULT_OPTS, Dsl::Dsl::TYPES

Class Method Summary collapse

Methods inherited from BaseAction

check_params

Methods included from Dsl::Dsl

#define_params, #define_type, #process

Class Method Details

.call(params) ⇒ Object 



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
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
 
# File 'lib/gooddata/lcm/actions/create_segment_masters.rb', line 37

def call(params)
  # Check if all required parameters were passed
  BaseAction.check_params(PARAMS, params)

  results = []

  client = params.gdc_gd_client
  development_client = params.development_client

  domain_name = params.organization || params.domain
  domain = client.domain(domain_name) || fail("Invalid domain name specified - #{domain_name}")
  domain_segments = domain.segments

  # TODO: Support for 'per segment' provisioning
  segments = params.segments

  # Output param with info about which projects should be synchronized
  synchronize_projects = []

  segments.map do |segment_in| # rubocop:disable Metrics/BlockLength
    segment_id = segment_in.segment_id
    development_pid = segment_in.development_pid
    driver = segment_in.driver.downcase
    token = params.tokens[driver.to_sym] || fail("Token for driver '#{driver}' was not specified")

    # Create master project Postgres
    version = get_project_version(params, segment_id) + 1

    master_name = segment_in.master_name.gsub('#{version}', version.to_s)

    # Get project instance based on PID. Fail if invalid one was specified.
    # TODO: Use development client for getting project
    development_client.projects(development_pid) || fail("Invalid Development PID specified - #{development_pid}")
    segment = domain_segments.find do |ds|
      ds.segment_id == segment_id
    end

    # Create new master project
    params.gdc_logger.info "Creating master project - name: '#{master_name}' development_project: '#{development_pid}', segment: '#{segment_id}', driver: '#{driver}'"
    project = client.create_project(title: master_name, auth_token: token, driver: driver == 'vertica' ? 'vertica' : 'Pg')

    # Do we have hash with synchronization info for current development project?
    synchronize_info = synchronize_projects.find do |info|
      info[:from] == development_pid
    end

    # If not, create new one
    unless synchronize_info
      synchronize_info = {
        segment: segment_id,
        from: development_pid,
        to: []
      }
      synchronize_projects << synchronize_info
    end

    # Add target project (new master for segment) into synchronization info
    synchronize_info[:to] << { pid: project.pid }

    # Does segment exists? If not, create new one and set initial master
    if segment
      segment_in[:is_new] = false
    else
      params.gdc_logger.info "Creating segment #{segment_id}, master #{project.pid}"
      segment = domain.create_segment(segment_id: segment_id, master_project: project)
      segment.synchronize_clients
      segment_in[:is_new] = true
    end

    master_project = nil

    begin
      master_project = segment.master_project
    rescue => e
      GoodData.logger.warn "Unable to get segment master, reason: #{e.message}"
    end

    if master_project.nil? || master_project.deleted?
      segment.master_project = project
      segment.save
      segment_in[:is_new] = true
    end

    segment_in[:master_pid] = project.pid
    segment_in[:version] = version
    segment_in[:timestamp] = Time.now.utc.iso8601

    # Show new project
    params.gdc_logger.info JSON.pretty_generate(project.json)

    # Add new segment master project with additional info into output results
    results << {
      segment_id: segment_id,
      name: master_name,
      development_pid: development_pid,
      master_pid: project.pid,
      driver: driver,
      status: 'created'
    }

    project
  end

  # Return results
  {
    results: results,
    params: {
      synchronize: synchronize_projects
    }
  }
end

.get_project_version(params, segment_id) ⇒ Object 



149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
 
# File 'lib/gooddata/lcm/actions/create_segment_masters.rb', line 149

def get_project_version(params, segment_id)
  replacements = {
    table_name: params.release_table_name || DEFAULT_TABLE_NAME,
    segment_id: segment_id
  }

  path = File.expand_path('../../data/select_from_lcm_release.sql.erb', __FILE__)
  query = GoodData::Helpers::ErbHelper.template_file(path, replacements)

  res = params.ads_client.execute_select(query)

  return 0 if res.empty?

  res[0][:version].to_i
end