package AWS::XRay::Segment;
use 5.012000;
use strict;
use warnings;
use JSON::XS    ();
use Time::HiRes ();
my $header = qq|{"format":"json","version":1}\n|;
my $json   = JSON::XS->new->utf8;
sub new {
    my $class = shift;
    my $src   = shift;
    return bless {}, "${class}::NoTrace" if !$AWS::XRay::ENABLED;
    my $segment = {
        id         => AWS::XRay::new_id(),
        start_time => Time::HiRes::time(),
        trace_id   => $AWS::XRay::TRACE_ID,
        %$src,
    };
    if (my $parent_id = $AWS::XRay::SEGMENT_ID) {
        # This is a sub segment.
        $segment->{parent_id} = $parent_id;
        $segment->{type}      = "subsegment";
        $segment->{namespace} = "remote";
    }
    bless $segment, $class;
}
# alias for backward compatibility
*send = \&close;
sub close {
    my $self = shift;
    $self->{end_time} //= Time::HiRes::time();
    my $sock = AWS::XRay::sock() or return;
    $sock->print($header, $json->encode({%$self}));
}
sub trace_header {
    my $self = shift;
    my $h    = sprintf("Root=%s;Parent=%s", $self->{trace_id}, $self->{id});
    if (defined $AWS::XRay::SAMPLED) {
        $h .= ";Sampled=$AWS::XRay::SAMPLED";
    }
    return $h;
}
package AWS::XRay::Segment::NoTrace;
sub new {
    my $class = shift;
    bless {}, $class;
}
sub close {
    # do not anything
}
sub trace_header {
    "";
}
1;