# PostgreSQLIndexing.java

    /*
     * Copyright (c) 2010, Stanislav Muhametsin. All Rights Reserved.
     *
     * Licensed  under the  Apache License,  Version 2.0  (the "License");
     * you may not use  this file  except in  compliance with the License.
     * You may obtain a copy of the License at
     *
     *   http://www.apache.org/licenses/LICENSE-2.0
     *
     * Unless required by applicable law or agreed to in writing, software
     * distributed  under the  License is distributed on an "AS IS" BASIS,
     * WITHOUT  WARRANTIES OR CONDITIONS  OF ANY KIND, either  express  or
     * implied.
     *
     * See the License for the specific language governing permissions and
     * limitations under the License.
     */
    package org.qi4j.index.sql.support.postgresql;

    import java.sql.Connection;
    import java.sql.PreparedStatement;
    import java.sql.ResultSet;
    import java.sql.SQLException;
    import org.qi4j.index.sql.support.common.DBNames;
    import org.qi4j.index.sql.support.skeletons.AbstractSQLIndexing;
    import org.qi4j.library.sql.common.SQLUtil;
    import org.qi4j.spi.entity.EntityState;
    import org.sql.generation.api.grammar.builders.modification.ColumnSourceByValuesBuilder;
    import org.sql.generation.api.grammar.builders.modification.pgsql.PgSQLInsertStatementBuilder;
    import org.sql.generation.api.grammar.factories.ColumnsFactory;
    import org.sql.generation.api.grammar.factories.LiteralFactory;
    import org.sql.generation.api.grammar.factories.ModificationFactory;
    import org.sql.generation.api.grammar.factories.TableReferenceFactory;
    import org.sql.generation.api.grammar.modification.InsertStatement;
    import org.sql.generation.api.grammar.modification.ValueSource;
    import org.sql.generation.api.vendor.SQLVendor;

    public class PostgreSQLIndexing
        extends AbstractSQLIndexing
    {

        @Override
        protected long getPKFromAutoGeneratedIDInsert( EntityState state,
                                                       PreparedStatement autoGeneratedIDStatement,
                                                       SQLVendor vendor, Connection connection )
            throws SQLException
        {
            this.addEntityInfoToInsertToEntityTablePS( state, autoGeneratedIDStatement, 1 );
            ResultSet rs = autoGeneratedIDStatement.executeQuery();
            try
            {
                rs.next();
                return rs.getLong( 1 );
            }
            finally
            {
                SQLUtil.closeQuietly( rs );
            }
        }

        @Override
        protected InsertStatement createInsertStatementWithAutoGeneratedIDForEntitiesTable(
            String schemaName, String tableName, SQLVendor vendor )
        {
            ModificationFactory m = vendor.getModificationFactory();
            LiteralFactory l = vendor.getLiteralFactory();
            TableReferenceFactory t = vendor.getTableReferenceFactory();
            ColumnsFactory c = vendor.getColumnsFactory();

            ColumnSourceByValuesBuilder columnBuilder = m.columnSourceByValues();
            columnBuilder.addValues( ValueSource.Default.INSTANCE );
            for( Integer x = 1; x < AMOUNT_OF_COLUMNS_IN_ENTITY_TABLE; ++x )
            {
                columnBuilder.addValues( l.param() );
            }

            return ( (PgSQLInsertStatementBuilder) m.insert() )
                .setReturningClause(
                    vendor.getQueryFactory().columnsBuilder()
                    .addUnnamedColumns( c.colName( DBNames.ENTITY_TABLE_PK_COLUMN_NAME ) )
                    .createExpression()
                )
                .setTableName( t.tableName( schemaName, tableName ) )
                .setColumnSource( columnBuilder.createExpression() )
                .createExpression();
        }
    }