masamune 0.12.3 → 0.13.0

data/lib/masamune/template.rb

@@ -27,6 +27,7 @@ module Masamune
   class Template
     def initialize(paths = [])
       @paths = Array.wrap(paths)
+      @paths << File.join(File.dirname(__FILE__), 'transform')
     end
 
     def render(template, parameters = {})
@@ -48,7 +49,6 @@ module Masamune
 
     class << self
       def render_to_file(template, parameters = {})
-        raise IOError, "File not found: #{template}" unless File.exists?(template)
         Tempfile.new('masamune').tap do |file|
           file.write(render_to_string(template, parameters))
           file.close
@@ -56,7 +56,6 @@ module Masamune
       end
 
       def render_to_string(template, parameters = {})
-        raise IOError, "File not found: #{template}" unless File.exists?(template)
         instance = Template.new(File.dirname(template))
         combine instance.render(template, parameters)
       end

data/lib/masamune/transform/define_foreign_key.psql.erb

@@ -0,0 +1,39 @@
+--  The MIT License (MIT)
+--
+--  Copyright (c) 2014-2015, VMware, Inc. All Rights Reserved.
+--
+--  Permission is hereby granted, free of charge, to any person obtaining a copy
+--  of this software and associated documentation files (the "Software"), to deal
+--  in the Software without restriction, including without limitation the rights
+--  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+--  copies of the Software, and to permit persons to whom the Software is
+--  furnished to do so, subject to the following conditions:
+--
+--  The above copyright notice and this permission notice shall be included in
+--  all copies or substantial portions of the Software.
+--
+--  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+--  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+--  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+--  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+--  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+--  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+--  THE SOFTWARE.
+
+<%
+  skip_check_exist ||= false
+  skip_check_valid ||= false
+%>
+
+<%- target.foreign_key_constraints.each do |id, column_names, reference_name, reference_column_names| -%>
+<%- foreign_key_name = "#{target.name}_#{id}_fkey" -%>
+<%- unless target.temporary? || skip_check_exist -%>
+DO $$ BEGIN
+IF NOT EXISTS (SELECT 1 FROM pg_constraint c WHERE c.conname = '<%= foreign_key_name %>') THEN
+<%- end -%>
+ALTER TABLE <%= target.name %> ADD CONSTRAINT <%= foreign_key_name %> FOREIGN KEY (<%= column_names.join(', ') %>) REFERENCES <%= reference_name %>(<%= reference_column_names.join(', ') %>)<%= ' NOT VALID DEFERRABLE INITIALLY DEFERRED' if skip_check_valid %>;
+<%- unless target.temporary? || skip_check_exist -%>
+END IF; END $$;
+
+<%- end -%>
+<%- end -%>

data/lib/masamune/transform/define_index.psql.erb

@@ -20,14 +20,18 @@
 --  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 --  THE SOFTWARE.
 
+<%
+  skip_check_exist ||= false
+%>
+
 <%- target.index_columns.each do |column_names, unique, id| -%>
 <%- index_name = "#{target.name}_#{id}_index" -%>
-<%- unless target.temporary? -%>
+<%- unless target.temporary? || skip_check_exist -%>
 DO $$ BEGIN
 IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = '<%= index_name %>') THEN
 <%- end -%>
-CREATE <%= unique ? 'UNIQUE INDEX' : 'INDEX' %> <%= index_name %> ON <%= target.name %> (<%= column_names.join(', ') %>);
-<%- unless target.temporary? -%>
+CREATE <%= unique ? 'UNIQUE INDEX' : 'INDEX' %> <%= index_name %> ON <%= target.name %> (<%= column_names.to_a.join(', ') %>);
+<%- unless target.temporary? || skip_check_exist -%>
 END IF; END $$;
 
 <%- end -%>

data/lib/masamune/transform/define_schema.rb

@@ -28,18 +28,18 @@ module Masamune::Transform
 
     extend ActiveSupport::Concern
 
-    def define_schema(catalog, store_id)
+    def define_schema(catalog, store_id, options = {})
       context = catalog[store_id]
       operators = []
 
       operators += context.extra(:pre)
 
       context.dimensions.each do |_, dimension|
-        operators << define_table(dimension)
+        operators << define_table(dimension, [], options)
       end
 
       context.facts.each do |_, fact|
-        operators << define_table(fact)
+        operators << define_table(fact, [], options)
       end
 
       operators += context.extra(:post)

data/lib/masamune/transform/define_table.psql.erb

@@ -20,10 +20,15 @@
 --  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 --  THE SOFTWARE.
 
-<% files ||= [] %>
+<%
+  files ||= []
+  with_index = locals.fetch(:with_index, true)
+  with_foreign_key = locals.fetch(:with_foreign_key, true)
+  with_unique_constraint = locals.fetch(:with_unique_constraint, true)
+%>
 
 <%- target.children.each do |child| -%>
-<%= render 'define_table.psql.erb', target: child %>
+<%= render 'define_table.psql.erb', target: child, **locals.except(:target, :files)  %>
 <%- end -%>
 
 <%- target.enum_columns.each do |_, column| -%>
@@ -52,6 +57,17 @@ CREATE TABLE IF NOT EXISTS <%= target.name %>
   <%- end -%>
 );
 
+<%- unless target.temporary? || target.primary_keys.empty? -%>
+DO $$ BEGIN
+IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = '<%= target.name %>_pkey') THEN
+ALTER TABLE <%= target.name %> ADD PRIMARY KEY (<%= target.primary_keys.map(&:name).join(', ') %>);
+END IF; END $$;
+<%- end -%>
+
+<%- if with_foreign_key && !target.delay_constraints? -%>
+<%= render 'define_foreign_key.psql.erb', target: target %>
+<%- end -%>
+
 <%- target.sequence_columns.each do |_, column| -%>
 DO $$ BEGIN
 IF NOT EXISTS (SELECT 1 WHERE sequence_owner('<%= column.sequence_id %>') = '<%= column.qualified_name %>') THEN
@@ -69,8 +85,13 @@ END IF; END $$;
 COPY <%= target.name %> FROM '<%= file %>' WITH (<%= copy_options.join(", ") %>);
 <%- end -%>
 
+<%- if with_unique_constraint && !target.delay_constraints? -%>
 <%= render 'define_unique.psql.erb', target: target %>
+<%- end -%>
+
+<%- if with_index && !target.delay_index? -%>
 <%= render 'define_index.psql.erb', target: target %>
+<%- end -%>
 
 <% target.insert_rows.each do |row| %>
 INSERT INTO <%= target.name %> (<%= row.insert_columns.join(', ') %>)
@@ -86,7 +107,7 @@ ANALYZE <%= target.name %>;
 <%- row.natural_keys.each do |column| -%>
 CREATE OR REPLACE FUNCTION <%= row.name(column) %>
 RETURNS <%= column.sql_type %> IMMUTABLE AS $$
-  SELECT <%= row.sql_value(column) %>;
+  SELECT CAST(<%= row.sql_value(column) %> AS <%= column.sql_type %>);
 $$ LANGUAGE SQL;
 
 <%- end -%>

data/lib/masamune/transform/define_table.rb

@@ -24,13 +24,27 @@ module Masamune::Transform
   module DefineTable
     extend ActiveSupport::Concern
 
-    def define_table(target, files = [])
+    def define_table(target, files = [], options = {})
       return if target.implicit
-      Operator.new(__method__, target: target, files: Masamune::Schema::Map.convert_files(files), presenters: { hive: Hive }).tap do |operator|
+      Operator.new(__method__, target: target, files: Masamune::Schema::Map.convert_files(files), **options.slice(:with_index, :with_foreign_key, :with_unique_constraint), presenters: { postgres: Postgres, hive: Hive }).tap do |operator|
         logger.debug("#{target.id}\n" + operator.to_s) if target.debug
       end
     end
 
+    class Postgres < SimpleDelegator
+      def children
+        super.map { |child| self.class.new(child) }
+      end
+
+      def delay_index?
+        type == :fact
+      end
+
+      def delay_constraints?
+        type == :fact
+      end
+    end
+
     class Hive < SimpleDelegator
       def partition_by
         return unless partitions.any?

data/lib/masamune/transform/define_unique.psql.erb

@@ -24,7 +24,7 @@
 <%- constraint_name = "#{target.name}_#{id}_key" -%>
 DO $$ BEGIN
 IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = '<%= constraint_name %>') THEN
-ALTER TABLE <%= target.name %> ADD CONSTRAINT <%= constraint_name %> UNIQUE(<%= column_names.join(', ') %>);
+ALTER TABLE <%= target.name %> ADD CONSTRAINT <%= constraint_name %> UNIQUE(<%= column_names.to_a.join(', ') %>);
 END IF; END $$;
 
 <%- end -%>

data/lib/masamune/transform/denormalize_table.rb

@@ -54,6 +54,10 @@ module Masamune::Transform
       end
       method_with_last_element :select_columns
 
+      def join_alias(reference)
+        reference.label ? "#{reference.name} AS #{[reference.label, reference.name].compact.join('_')}" : reference.name
+      end
+
       def join_conditions(column_names)
         {}.tap do |conditions|
           column_names.each do |column_name|
@@ -63,7 +67,7 @@ module Masamune::Transform
             next unless adjacent_reference
             adjacent_column = columns[adjacent_reference.foreign_key_name]
             next unless adjacent_column
-            conditions[column.reference.name] = "#{column.reference.surrogate_key.qualified_name} = #{adjacent_column.qualified_name}"
+            conditions[join_alias(column.reference)] = "#{column.reference.surrogate_key.qualified_name(column.reference.label)} = #{adjacent_column.qualified_name}"
           end
         end
       end

data/lib/masamune/transform/replace_table.psql.erb

@@ -30,13 +30,12 @@ BEGIN;
 SET TRANSACTION ISOLATION LEVEL SERIALIZABLE;
 
 ALTER TABLE <%= target.name %> DROP CONSTRAINT IF EXISTS <%= target.name %>_time_key_check;
-<%- target.foreign_key_columns.each do |column| -%>
-<%- if column.reference_constraint -%>
-ALTER TABLE <%= target.name %> DROP CONSTRAINT IF EXISTS <%= target.name %>_<%= column.name %>_fkey CASCADE;
-<%- end -%>
+<%- target.foreign_key_constraints.each do |id, _, _, _| -%>
+<%- foreign_key_name = "#{target.name}_#{id}_fkey" -%>
+ALTER TABLE <%= target.name %> DROP CONSTRAINT IF EXISTS <%= foreign_key_name %> CASCADE;
 <%- end -%>
 
-<%- target.index_columns.each do |column_names, unique, id| -%>
+<%- target.index_columns.each do |_, _, id| -%>
 <%- index_name = "#{target.name}_#{id}_index" -%>
 DROP INDEX IF EXISTS <%= index_name %>;
 <%- end -%>
@@ -44,18 +43,15 @@ DROP INDEX IF EXISTS <%= index_name %>;
 ALTER TABLE <%= target.name %> RENAME TO <%= target_tmp.name %>;
 ALTER TABLE <%= source.name %> RENAME TO <%= target.name %>;
 
+<%- if target.parent -%>
 ALTER TABLE <%= target.name %> INHERIT <%= target.parent.name %>;
-ALTER TABLE <%= target.name %> ADD CONSTRAINT <%= target.name %>_time_key_check <%= target.constraints %>;
-<%- target.foreign_key_columns.each do |column| -%>
-<%- if column.reference_constraint -%>
-ALTER TABLE <%= target.name %> ADD CONSTRAINT <%= target.name %>_<%= column.name %>_fkey FOREIGN KEY (<%= column.name %>) <%= column.reference_constraint %> NOT VALID DEFERRABLE INITIALLY DEFERRED;
 <%- end -%>
+<%- if target.constraints -%>
+ALTER TABLE <%= target.name %> ADD CONSTRAINT <%= target.name %>_time_key_check <%= target.constraints %>;
 <%- end -%>
+<%= render 'define_foreign_key.psql.erb', target: target, skip_check_exist: true, skip_check_valid: true %>
 
-<%- target.index_columns.each do |column_names, unique, id| -%>
-<%- index_name = "#{target.name}_#{id}_index" -%>
-CREATE <%= unique ? 'UNIQUE INDEX' : 'INDEX' %> <%= index_name %> ON <%= target.name %> (<%= column_names.join(', ') %>);
-<%- end -%>
+<%= render 'define_index.psql.erb', target: target, skip_check_exist: true %>
 
 ANALYZE <%= target.name %>;
 

data/lib/masamune/transform/stage_fact.rb

@@ -48,7 +48,7 @@ module Masamune::Transform
         shared_columns(source).values.map do |columns|
           column = columns.first
           if !column.degenerate? && reference = column.reference
-            reference.surrogate_key.qualified_name
+            reference.surrogate_key.qualified_name(column.reference.label)
           else
             column.qualified_name
           end
@@ -56,6 +56,10 @@ module Masamune::Transform
       end
       method_with_last_element :insert_values
 
+      def join_alias(reference)
+        reference.label ? "#{reference.name} AS #{[reference.label, reference.name].compact.join('_')}" : reference.name
+      end
+
       def join_conditions(source)
         join_columns = shared_columns(source).values.flatten
         join_columns = join_columns.select { |column| column.reference }
@@ -64,15 +68,16 @@ module Masamune::Transform
         dependencies = Masamune::TopologicalHash.new
         conditions = Hash.new { |h,k| h[k] = [] }
         join_columns.each do |reference, columns|
+          reference_name = join_alias(reference)
           columns.each do |column|
             next if column.degenerate?
-            dependencies[reference.name] = []
+            dependencies[reference_name] ||= []
             cross_references = cross_references(column)
             coalesce_values = []
 
             if cross_references.any?
-              dependencies[reference.name] += cross_references.map { |reference, _| reference.name }
-              coalesce_values << cross_references.map { |_, column| column.qualified_name }
+              dependencies[reference_name] += cross_references.map { |reference, _| join_alias(reference) }
+              coalesce_values << cross_references.map { |reference, column| column.qualified_name(reference.label) }
             end
 
             if column.reference && !column.reference.default.nil?
@@ -81,14 +86,14 @@ module Masamune::Transform
               coalesce_values << column.adjacent.sql_value(column.adjacent.default)
             end
 
-            conditions[reference.name] << (coalesce_values.any? ?
+            conditions[reference_name] << (coalesce_values.any? ?
               "#{column.foreign_key_name} = COALESCE(#{column.qualified_name}, #{coalesce_values.join(', ')})" :
               "#{column.foreign_key_name} = #{column.qualified_name}")
           end
           if reference.type == :two || reference.type == :four
-            join_key_a = "TO_TIMESTAMP(#{source.time_key.qualified_name}) BETWEEN #{reference.start_key.qualified_name} AND COALESCE(#{reference.end_key.qualified_name}, 'INFINITY')"
-            join_key_b = "TO_TIMESTAMP(#{source.time_key.qualified_name}) < #{reference.start_key.qualified_name} AND #{reference.version_key.qualified_name} = 1"
-            conditions[reference.name] << "((#{join_key_a}) OR (#{join_key_b}))"
+            join_key_a = "TO_TIMESTAMP(#{source.time_key.qualified_name}) BETWEEN #{reference.start_key.qualified_name(reference.label)} AND COALESCE(#{reference.end_key.qualified_name(reference.label)}, 'INFINITY')"
+            join_key_b = "TO_TIMESTAMP(#{source.time_key.qualified_name}) < #{reference.start_key.qualified_name(reference.label)} AND #{reference.version_key.qualified_name(reference.label)} = 1"
+            conditions[reference_name] << "((#{join_key_a}) OR (#{join_key_b}))"
           end
         end
         conditions.slice(*dependencies.tsort)
@@ -99,8 +104,9 @@ module Masamune::Transform
       def cross_references(column)
         return {} unless column.natural_key || column.adjacent.try(:natural_key)
         {}.tap do |result|
-          references.each do |_, reference|
-            if reference.id != column.reference.id && reference.columns[column.id]
+          column.reference.through.each do |reference_id|
+            reference = references[reference_id]
+            if reference.columns[column.id]
               result[reference] = reference.columns[column.id]
             end
           end

data/lib/masamune/version.rb

@@ -21,5 +21,5 @@
 #  THE SOFTWARE.
 
 module Masamune
-  VERSION = '0.12.3'
+  VERSION = '0.13.0'
 end

data/spec/masamune/schema/map_spec.rb

@@ -223,7 +223,7 @@ describe Masamune::Schema::Map do
       end
 
       before do
-        expect(environment.logger).to receive(:warn).with(/failed to process '{.*}' for #{target.name}/).ordered
+        expect(environment.logger).to receive(:warn).with(/failed to process row for #{target.name}/).ordered
         expect(environment.logger).to receive(:warn).with(/failed to parse '{.*}' for #{source.name}/).ordered
       end
 

data/spec/masamune/tasks/dump_thor_spec.rb

@@ -0,0 +1,42 @@
+#  The MIT License (MIT)
+#
+#  Copyright (c) 2014-2015, VMware, Inc. All Rights Reserved.
+#
+#  Permission is hereby granted, free of charge, to any person obtaining a copy
+#  of this software and associated documentation files (the "Software"), to deal
+#  in the Software without restriction, including without limitation the rights
+#  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+#  copies of the Software, and to permit persons to whom the Software is
+#  furnished to do so, subject to the following conditions:
+#
+#  The above copyright notice and this permission notice shall be included in
+#  all copies or substantial portions of the Software.
+#
+#  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+#  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+#  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+#  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+#  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+#  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+#  THE SOFTWARE.
+
+require 'spec_helper'
+require 'thor'
+
+require 'masamune/tasks/dump_thor'
+
+describe Masamune::Tasks::DumpThor do
+  context 'with help command ' do
+    let(:command) { 'help' }
+    it_behaves_like 'command usage'
+  end
+
+  context 'with no arguments' do
+    it 'exits with status code 0 and prints catalog' do
+      expect { cli_invocation }.to raise_error { |e|
+        expect(e).to be_a(SystemExit)
+        expect(e.status).to eq(0)
+      }
+    end
+  end
+end

data/spec/masamune/tasks/shell_thor_spec.rb

@@ -37,15 +37,4 @@ describe Masamune::Tasks::ShellThor do
       cli_invocation
     end
   end
-
-  context 'with --dump' do
-    let(:options) { ['--dump'] }
-
-    it 'exits with status code 0 and prints catalog' do
-      expect { cli_invocation }.to raise_error { |e|
-        expect(e).to be_a(SystemExit)
-        expect(e.status).to eq(0)
-      }
-    end
-  end
 end

data/spec/masamune/template_spec.rb

@@ -73,5 +73,10 @@ describe Masamune::Template do
       let(:template) { File.join(File.dirname(__FILE__), '..', 'fixtures', 'relative.sql.erb') }
       it { is_expected.to eq("SELECT * FROM relative;\n") }
     end
+
+    context 'with packaged template' do
+      let(:template) { 'define_schema.hql.erb' }
+      it { is_expected.to_not be_nil }
+    end
   end
 end

data/spec/masamune/transform/define_table.dimension_spec.rb

@@ -157,11 +157,16 @@ describe Masamune::Transform::DefineTable do
       is_expected.to eq <<-EOS.strip_heredoc
         CREATE TABLE IF NOT EXISTS user_dimension
         (
-          id SERIAL PRIMARY KEY,
+          id SERIAL,
           tenant_id INTEGER NOT NULL,
           user_id INTEGER NOT NULL,
           last_modified_at TIMESTAMP NOT NULL DEFAULT NOW()
         );
+
+        DO $$ BEGIN
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_pkey') THEN
+        ALTER TABLE user_dimension ADD PRIMARY KEY (id);
+        END IF; END $$;
       EOS
     end
   end
@@ -170,8 +175,8 @@ describe Masamune::Transform::DefineTable do
     before do
       catalog.schema :postgres do
         dimension 'user', type: :two do
-          column 'tenant_id', index: true, natural_key: true
-          column 'user_id', index: true, natural_key: true
+          column 'tenant_id', natural_key: true
+          column 'user_id', natural_key: true
         end
       end
     end
@@ -182,7 +187,7 @@ describe Masamune::Transform::DefineTable do
       is_expected.to eq <<-EOS.strip_heredoc
         CREATE TABLE IF NOT EXISTS user_dimension
         (
-          id SERIAL PRIMARY KEY,
+          id SERIAL,
           tenant_id INTEGER NOT NULL,
           user_id INTEGER NOT NULL,
           start_at TIMESTAMP NOT NULL DEFAULT TO_TIMESTAMP(0),
@@ -192,18 +197,18 @@ describe Masamune::Transform::DefineTable do
         );
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_e6c3d91_key') THEN
-        ALTER TABLE user_dimension ADD CONSTRAINT user_dimension_e6c3d91_key UNIQUE(tenant_id, user_id, start_at);
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_pkey') THEN
+        ALTER TABLE user_dimension ADD PRIMARY KEY (id);
         END IF; END $$;
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_3854361_index') THEN
-        CREATE INDEX user_dimension_3854361_index ON user_dimension (tenant_id);
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_e6c3d91_key') THEN
+        ALTER TABLE user_dimension ADD CONSTRAINT user_dimension_e6c3d91_key UNIQUE(tenant_id, user_id, start_at);
         END IF; END $$;
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_e8701ad_index') THEN
-        CREATE INDEX user_dimension_e8701ad_index ON user_dimension (user_id);
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_2c8e908_index') THEN
+        CREATE INDEX user_dimension_2c8e908_index ON user_dimension (end_at);
         END IF; END $$;
 
         DO $$ BEGIN
@@ -212,13 +217,18 @@ describe Masamune::Transform::DefineTable do
         END IF; END $$;
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_2c8e908_index') THEN
-        CREATE INDEX user_dimension_2c8e908_index ON user_dimension (end_at);
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_3854361_index') THEN
+        CREATE INDEX user_dimension_3854361_index ON user_dimension (tenant_id);
+        END IF; END $$;
+
+        DO $$ BEGIN
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_e8701ad_index') THEN
+        CREATE INDEX user_dimension_e8701ad_index ON user_dimension (user_id);
         END IF; END $$;
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_2af72f1_index') THEN
-        CREATE INDEX user_dimension_2af72f1_index ON user_dimension (version);
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_e6c3d91_index') THEN
+        CREATE UNIQUE INDEX user_dimension_e6c3d91_index ON user_dimension (tenant_id, user_id, start_at);
         END IF; END $$;
       EOS
     end
@@ -242,8 +252,8 @@ describe Masamune::Transform::DefineTable do
         dimension 'user', type: :four do
           references :cluster
           references :user_account_state
-          column 'tenant_id', index: true, natural_key: true
-          column 'user_id', index: true, natural_key: true
+          column 'tenant_id', natural_key: true
+          column 'user_id', natural_key: true
           column 'preferences', type: :key_value, null: true
         end
       end
@@ -255,9 +265,9 @@ describe Masamune::Transform::DefineTable do
       is_expected.to eq <<-EOS.strip_heredoc
         CREATE TABLE IF NOT EXISTS user_dimension_ledger
         (
-          id SERIAL PRIMARY KEY,
-          cluster_type_id INTEGER NOT NULL REFERENCES cluster_type(id) DEFAULT default_cluster_type_id(),
-          user_account_state_type_id INTEGER REFERENCES user_account_state_type(id),
+          id SERIAL,
+          cluster_type_id INTEGER NOT NULL DEFAULT default_cluster_type_id(),
+          user_account_state_type_id INTEGER,
           tenant_id INTEGER NOT NULL,
           user_id INTEGER NOT NULL,
           preferences HSTORE,
@@ -269,18 +279,23 @@ describe Masamune::Transform::DefineTable do
         );
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_ledger_370d6dd_key') THEN
-        ALTER TABLE user_dimension_ledger ADD CONSTRAINT user_dimension_ledger_370d6dd_key UNIQUE(tenant_id, user_id, source_kind, source_uuid, start_at);
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_ledger_pkey') THEN
+        ALTER TABLE user_dimension_ledger ADD PRIMARY KEY (cluster_type_id, id);
         END IF; END $$;
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_ledger_d6b9b38_index') THEN
-        CREATE INDEX user_dimension_ledger_d6b9b38_index ON user_dimension_ledger (cluster_type_id);
+        IF NOT EXISTS (SELECT 1 FROM pg_constraint c WHERE c.conname = 'user_dimension_ledger_d6b9b38_fkey') THEN
+        ALTER TABLE user_dimension_ledger ADD CONSTRAINT user_dimension_ledger_d6b9b38_fkey FOREIGN KEY (cluster_type_id) REFERENCES cluster_type(id);
         END IF; END $$;
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_ledger_7988187_index') THEN
-        CREATE INDEX user_dimension_ledger_7988187_index ON user_dimension_ledger (user_account_state_type_id);
+        IF NOT EXISTS (SELECT 1 FROM pg_constraint c WHERE c.conname = 'user_dimension_ledger_7988187_fkey') THEN
+        ALTER TABLE user_dimension_ledger ADD CONSTRAINT user_dimension_ledger_7988187_fkey FOREIGN KEY (user_account_state_type_id) REFERENCES user_account_state_type(id);
+        END IF; END $$;
+
+        DO $$ BEGIN
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_ledger_ff54dba_key') THEN
+        ALTER TABLE user_dimension_ledger ADD CONSTRAINT user_dimension_ledger_ff54dba_key UNIQUE(cluster_type_id, tenant_id, user_id, source_kind, source_uuid, start_at);
         END IF; END $$;
 
         DO $$ BEGIN
@@ -294,20 +309,20 @@ describe Masamune::Transform::DefineTable do
         END IF; END $$;
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_ledger_23563d3_index') THEN
-        CREATE INDEX user_dimension_ledger_23563d3_index ON user_dimension_ledger (start_at);
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_ledger_370d6dd_index') THEN
+        CREATE INDEX user_dimension_ledger_370d6dd_index ON user_dimension_ledger (tenant_id, user_id, source_kind, source_uuid, start_at);
         END IF; END $$;
 
         CREATE TABLE IF NOT EXISTS user_dimension
         (
-          id SERIAL PRIMARY KEY,
-          cluster_type_id INTEGER NOT NULL REFERENCES cluster_type(id) DEFAULT default_cluster_type_id(),
-          user_account_state_type_id INTEGER NOT NULL REFERENCES user_account_state_type(id) DEFAULT default_user_account_state_type_id(),
+          id SERIAL,
+          cluster_type_id INTEGER NOT NULL DEFAULT default_cluster_type_id(),
+          user_account_state_type_id INTEGER NOT NULL DEFAULT default_user_account_state_type_id(),
           tenant_id INTEGER NOT NULL,
           user_id INTEGER NOT NULL,
           preferences HSTORE,
-          parent_id INTEGER REFERENCES user_dimension_ledger(id),
-          record_id INTEGER REFERENCES user_dimension_ledger(id),
+          parent_id INTEGER,
+          record_id INTEGER,
           start_at TIMESTAMP NOT NULL DEFAULT TO_TIMESTAMP(0),
           end_at TIMESTAMP,
           version INTEGER DEFAULT 1,
@@ -315,33 +330,33 @@ describe Masamune::Transform::DefineTable do
         );
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_e6c3d91_key') THEN
-        ALTER TABLE user_dimension ADD CONSTRAINT user_dimension_e6c3d91_key UNIQUE(tenant_id, user_id, start_at);
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_pkey') THEN
+        ALTER TABLE user_dimension ADD PRIMARY KEY (cluster_type_id, id);
         END IF; END $$;
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_d6b9b38_index') THEN
-        CREATE INDEX user_dimension_d6b9b38_index ON user_dimension (cluster_type_id);
+        IF NOT EXISTS (SELECT 1 FROM pg_constraint c WHERE c.conname = 'user_dimension_d6b9b38_fkey') THEN
+        ALTER TABLE user_dimension ADD CONSTRAINT user_dimension_d6b9b38_fkey FOREIGN KEY (cluster_type_id) REFERENCES cluster_type(id);
         END IF; END $$;
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_7988187_index') THEN
-        CREATE INDEX user_dimension_7988187_index ON user_dimension (user_account_state_type_id);
+        IF NOT EXISTS (SELECT 1 FROM pg_constraint c WHERE c.conname = 'user_dimension_7988187_fkey') THEN
+        ALTER TABLE user_dimension ADD CONSTRAINT user_dimension_7988187_fkey FOREIGN KEY (user_account_state_type_id) REFERENCES user_account_state_type(id);
         END IF; END $$;
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_3854361_index') THEN
-        CREATE INDEX user_dimension_3854361_index ON user_dimension (tenant_id);
+        IF NOT EXISTS (SELECT 1 FROM pg_constraint c WHERE c.conname = 'user_dimension_e0538bc_fkey') THEN
+        ALTER TABLE user_dimension ADD CONSTRAINT user_dimension_e0538bc_fkey FOREIGN KEY (cluster_type_id, parent_id) REFERENCES user_dimension_ledger(cluster_type_id, id);
         END IF; END $$;
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_e8701ad_index') THEN
-        CREATE INDEX user_dimension_e8701ad_index ON user_dimension (user_id);
+        IF NOT EXISTS (SELECT 1 FROM pg_constraint c WHERE c.conname = 'user_dimension_824002d_fkey') THEN
+        ALTER TABLE user_dimension ADD CONSTRAINT user_dimension_824002d_fkey FOREIGN KEY (cluster_type_id, record_id) REFERENCES user_dimension_ledger(cluster_type_id, id);
         END IF; END $$;
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_23563d3_index') THEN
-        CREATE INDEX user_dimension_23563d3_index ON user_dimension (start_at);
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_3fcebfa_key') THEN
+        ALTER TABLE user_dimension ADD CONSTRAINT user_dimension_3fcebfa_key UNIQUE(cluster_type_id, tenant_id, user_id, start_at);
         END IF; END $$;
 
         DO $$ BEGIN
@@ -350,8 +365,23 @@ describe Masamune::Transform::DefineTable do
         END IF; END $$;
 
         DO $$ BEGIN
-        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_2af72f1_index') THEN
-        CREATE INDEX user_dimension_2af72f1_index ON user_dimension (version);
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_23563d3_index') THEN
+        CREATE INDEX user_dimension_23563d3_index ON user_dimension (start_at);
+        END IF; END $$;
+
+        DO $$ BEGIN
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_3854361_index') THEN
+        CREATE INDEX user_dimension_3854361_index ON user_dimension (tenant_id);
+        END IF; END $$;
+
+        DO $$ BEGIN
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_e8701ad_index') THEN
+        CREATE INDEX user_dimension_e8701ad_index ON user_dimension (user_id);
+        END IF; END $$;
+
+        DO $$ BEGIN
+        IF NOT EXISTS (SELECT 1 FROM pg_class c WHERE c.relname = 'user_dimension_e6c3d91_index') THEN
+        CREATE INDEX user_dimension_e6c3d91_index ON user_dimension (tenant_id, user_id, start_at);
         END IF; END $$;
       EOS
     end
@@ -368,8 +398,8 @@ describe Masamune::Transform::DefineTable do
 
         dimension 'user', type: :four do
           references :user_account_state
-          column 'tenant_id', index: true, natural_key: true
-          column 'user_id', index: true, natural_key: true
+          column 'tenant_id', natural_key: true
+          column 'user_id', natural_key: true
           column 'preferences', type: :key_value, null: true
         end
       end
@@ -393,12 +423,11 @@ describe Masamune::Transform::DefineTable do
           last_modified_at TIMESTAMP DEFAULT NOW()
         );
 
-        CREATE INDEX user_consolidated_forward_dimension_stage_7988187_index ON user_consolidated_forward_dimension_stage (user_account_state_type_id);
+        CREATE INDEX user_consolidated_forward_dimension_stage_2c8e908_index ON user_consolidated_forward_dimension_stage (end_at);
+        CREATE INDEX user_consolidated_forward_dimension_stage_23563d3_index ON user_consolidated_forward_dimension_stage (start_at);
         CREATE INDEX user_consolidated_forward_dimension_stage_3854361_index ON user_consolidated_forward_dimension_stage (tenant_id);
         CREATE INDEX user_consolidated_forward_dimension_stage_e8701ad_index ON user_consolidated_forward_dimension_stage (user_id);
-        CREATE INDEX user_consolidated_forward_dimension_stage_23563d3_index ON user_consolidated_forward_dimension_stage (start_at);
-        CREATE INDEX user_consolidated_forward_dimension_stage_2c8e908_index ON user_consolidated_forward_dimension_stage (end_at);
-        CREATE INDEX user_consolidated_forward_dimension_stage_2af72f1_index ON user_consolidated_forward_dimension_stage (version);
+        CREATE INDEX user_consolidated_forward_dimension_stage_e6c3d91_index ON user_consolidated_forward_dimension_stage (tenant_id, user_id, start_at);
       EOS
     end
   end