File Coverage

blib/lib/Kelp/Module/WebSocket/AnyEvent/Connection.pm

Criterion	Covered	Total	%
statement	9	27	33.3
branch	0	4	0.0
condition	0	12	0.0
subroutine	3	6	50.0
pod	0	3	0.0
total	12	52	23.0

package Kelp::Module::WebSocket::AnyEvent::Connection;

our $VERSION = '1.03';

use Kelp::Base;
use Carp;
use Scalar::Util qw(weaken blessed);

attr "-id";
attr "-manager";
attr "-connection";
attr "data" => sub { {} };

sub new
{
	my $class = shift;
	my $self = $class->SUPER::new(@_);
	$self->manager->connections->{$self->id} = $self;
	weaken($self->{connection});
	return $self;
}

sub send
{
	my ($self, $message) = @_;
	my $serializer = $self->manager->get_serializer;
	my $is_inst = blessed $message && $message->isa("AnyEvent::WebSocket::Message");

	if ($serializer && (!blessed $message || !$is_inst)) {
		$message = $serializer->encode($message);
	}

	if (ref $message && !$is_inst) {
		carp "invalid data sent to websocket peer, disconnecting";
		$self->connection->close;
		return;
	}

	$self->connection->send($message);
}

sub close
{
	my ($self) = @_;
	delete $self->manager->connections->{$self->id};
	$self->connection->close;
}

1;
__END__