File Coverage

blib/lib/Pg/SQL/PrettyPrinter/Node/TransactionStmt.pm

Criterion	Covered	Total	%
statement	67	68	98.5
branch	23	26	88.4
condition			n/a
subroutine	16	16	100.0
pod	2	2	100.0
total	108	112	96.4

package Pg::SQL::PrettyPrinter::Node::TransactionStmt;

# UTF8 boilerplace, per http://stackoverflow.com/questions/6162484/why-does-modern-perl-avoid-utf-8-by-default/
use v5.26;
use strict;
use warnings;
use warnings qw( FATAL utf8 );
use utf8;
use open qw( :std :utf8 );
use Unicode::Normalize qw( NFC );
use Unicode::Collate;
use Encode qw( decode );

if ( grep /\P{ASCII}/ => @ARGV ) {
    @ARGV = map { decode( 'UTF-8', $_ ) } @ARGV;
}

# If there is __DATA__,then uncomment next line:
# binmode( DATA, ':encoding(UTF-8)' );
# UTF8 boilerplace, per http://stackoverflow.com/questions/6162484/why-does-modern-perl-avoid-utf-8-by-default/

# Useful common code
use autodie;
use Carp         qw( carp croak confess cluck );
use English      qw( -no_match_vars );
use Data::Dumper qw( Dumper );

# give a full stack dump on any untrapped exceptions
local $SIG{ __DIE__ } = sub {
    confess "Uncaught exception: @_" unless $^S;
};

# now promote run-time warnings into stackdumped exceptions
#   *unless* we're in an try block, in which
#   case just generate a clucking stackdump instead
local $SIG{ __WARN__ } = sub {
    if   ( $^S ) { cluck "Trapped warning: @_" }
    else         { confess "Deadly warning: @_" }
};

# Useful common code

use parent qw( Pg::SQL::PrettyPrinter::Node );

sub new {
    my $class = shift;
    my $self  = $class->SUPER::new( @_ );
    bless $self, $class;

    croak "Invalid TransactionStmt Kind: " . $self->{ 'kind' } unless $self->{ 'kind' } =~ m{\A(?:TRANS_STMT_COMMIT|TRANS_STMT_ROLLBACK|TRANS_STMT_START|TRANS_STMT_BEGIN)\z};

    $self->objectify( 'options' );

    return $self;
}

sub as_text {
    my $self = shift;
    if ( $self->{ 'kind' } eq 'TRANS_STMT_ROLLBACK' ) {
        return 'ROLLBACK' . ( $self->{ 'chain' } ? ' AND CHAIN' : '' );
    }
    if ( $self->{ 'kind' } eq 'TRANS_STMT_COMMIT' ) {
        return 'COMMIT' . ( $self->{ 'chain' } ? ' AND CHAIN' : '' );
    }
    if ( $self->{ 'kind' } =~ /\ATRANS_STMT_(?:START|BEGIN)\z/ ) {
        my $cmd = $self->{ 'kind' } eq 'TRANS_STMT_BEGIN' ? 'BEGIN' : 'START TRANSACTION';
        return $cmd unless $self->{ 'options' };
        my @elements = ();
        for my $opt ( @{ $self->{ 'options' } } ) {
            if ( $opt->{ 'defname' } eq 'transaction_isolation' ) {
                push @elements, 'ISOLATION LEVEL ' . uc( $opt->{ 'arg' }->{ 'val' }->{ 'str' } );
            }
            elsif ( $opt->{ 'defname' } eq 'transaction_read_only' ) {
                if ( $opt->{ 'arg' }->{ 'val' }->{ 'ival' } ) {
                    push @elements, 'READ ONLY';
                }
                else {
                    push @elements, 'READ WRITE';
                }
            }
            elsif ( $opt->{ 'defname' } eq 'transaction_deferrable' ) {
                if ( $opt->{ 'arg' }->{ 'val' }->{ 'ival' } ) {
                    push @elements, 'DEFERRABLE';
                }
                else {
                    push @elements, 'NOT DEFERRABLE';
                }
            }
        }
        return $cmd . ' ' . join( ', ', @elements );
    }
    return 'aaa';
}

1;

# vim: set ft=perl: