Commit | Line | Data |
0a663589 |
1 | package Catalyst::Controller::MessageDriven; |
2 | use Moose; |
a5ae1e8c |
3 | use Data::Serializer; |
0a663589 |
4 | |
5 | BEGIN { extends 'Catalyst::Controller' } |
6 | |
d78bb739 |
7 | =head1 NAME |
8 | |
9 | Catalyst::Controller::MessageDriven |
10 | |
11 | =head1 SYNOPSIS |
12 | |
13 | package MyApp::Controller::Queue; |
14 | use Moose; |
15 | BEGIN { extends 'Catalyst::Controller::MessageDriven' } |
16 | |
17 | sub some_action : Local { |
bf8937b7 |
18 | my ($self, $c, $message) = @_; |
19 | |
20 | # Handle message |
21 | |
d78bb739 |
22 | # Reply with a minimal response message |
23 | my $response = { type => 'testaction_response' }; |
24 | $c->stash->{response} = $response; |
25 | } |
26 | |
27 | =head1 DESCRIPTION |
28 | |
29 | A Catalyst controller base class for use with Catalyst::Engine::Stomp, |
30 | which handles YAML-serialized messages. A top-level "type" key in the |
31 | YAML determines the action dispatched to. |
32 | |
33 | =cut |
34 | |
a5ae1e8c |
35 | __PACKAGE__->config( serializer => 'YAML' ); |
36 | |
bf8937b7 |
37 | sub begin : Private { |
38 | my ($self, $c) = @_; |
39 | |
40 | # Deserialize the request message |
bf8937b7 |
41 | my $message; |
a5ae1e8c |
42 | my $serializer = $self->config->{serializer}; |
43 | my $s = Data::Serializer->new( serializer => $serializer ); |
bf8937b7 |
44 | eval { |
45 | my $body = $c->request->body; |
a5ae1e8c |
46 | open my $IN, "$body" or die "can't open temp file $body"; |
47 | $message = $s->raw_deserialize(do { local $/; <$IN> }); |
bf8937b7 |
48 | }; |
49 | if ($@) { |
50 | # can't reply - reply_to is embedded in the message |
51 | $c->error("exception in deserialize: $@"); |
52 | } |
53 | else { |
54 | $c->stash->{request} = $message; |
55 | } |
56 | } |
0a663589 |
57 | |
bf8937b7 |
58 | sub end : Private { |
0a663589 |
59 | my ($self, $c) = @_; |
60 | |
61 | # Engine will send our reply based on the value of this header. |
bf8937b7 |
62 | $c->response->headers->header( 'X-Reply-Address' => $c->stash->{request}->{reply_to} ); |
480eb46f |
63 | |
64 | # The wire response |
65 | my $output; |
66 | |
67 | # Load a serializer |
68 | my $serializer = $self->config->{serializer}; |
69 | my $s = Data::Serializer->new( serializer => $serializer ); |
0a663589 |
70 | |
71 | # Custom error handler - steal errors from catalyst and dump them into |
72 | # the stash, to get them serialized out as the reply. |
73 | if (scalar @{$c->error}) { |
74 | my $error = join "\n", @{$c->error}; |
75 | $c->stash->{response} = { status => 'ERROR', error => $error }; |
480eb46f |
76 | $output = $s->serialize( $c->stash->{response} ); |
0a663589 |
77 | $c->error(0); # clear errors, so our response isn't clobbered |
78 | } |
bf8937b7 |
79 | |
80 | # Serialize the response |
bf8937b7 |
81 | eval { |
a5ae1e8c |
82 | $output = $s->raw_serialize( $c->stash->{response} ); |
bf8937b7 |
83 | }; |
84 | if ($@) { |
85 | my $error = "exception in serialize: $@"; |
86 | $c->error($error); |
87 | $c->stash->{response} = { status => 'ERROR', error => $error }; |
480eb46f |
88 | $output = $s->serialize( $c->stash->{response} ); |
bf8937b7 |
89 | } |
90 | |
91 | $c->response->output( $output ); |
0a663589 |
92 | } |
93 | |
94 | sub default : Private { |
95 | my ($self, $c) = @_; |
bf8937b7 |
96 | |
0a663589 |
97 | # Forward the request to the appropriate action, based on the |
98 | # message type. |
bf8937b7 |
99 | my $action = $c->stash->{request}->{type}; |
100 | $c->forward($action, [$c->stash->{request}]); |
0a663589 |
101 | } |
102 | |
103 | __PACKAGE__->meta->make_immutable; |
104 | |
105 | 1; |