123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123 |
- use Mojolicious::Lite;
- use Mango;
- use JSON::XS 'encode_json';
- use Scalar::Util 'looks_like_number';
- # configuration
- plugin JSONConfig => {
- file => 'app.conf',
- default => {
- database_host => 'localhost',
- hypnotoad => {
- graceful_timeout => 1,
- workers => 8,
- },
- hypnotoad_merge => {},
- },
- };
- {
- my $merge = app->config->{hypnotoad_merge};
- @{app->config->{hypnotoad}}{keys %$merge} = values %$merge;
- }
- # Database connections
- helper mango => sub { state $mango = Mango->new('mongodb://'. shift->config->{database_host}) };
- helper db => sub { state $db = shift->mango->db('hello_world') };
- helper world => sub { state $world = shift->db->collection('world') };
- helper fortune => sub { state $fortune = shift->db->collection('fortune') };
- # JSON::XS renderer
- helper render_json => sub { shift->render( data => encode_json(shift), format => 'json' ) };
- # Routes
- get '/json' => sub { shift->helpers->render_json({message => 'Hello, World!'}) };
- get '/db' => sub { shift->helpers->render_query(1, {single => 1}) };
- get '/queries' => sub {
- my $c = shift;
- $c->helpers->render_query(scalar $c->param('queries'));
- };
- get '/fortunes' => sub {
- my $c = shift->render_later;
- my $tx = $c->tx;
- $c->helpers->fortune->find->all(sub{
- my ($cursor, $err, $docs) = @_;
- push @$docs, { _id => 0, message => 'Additional fortune added at request time.' };
- $c->render( fortunes => docs => $docs ) unless $tx->is_finished;
- });
- };
- get '/updates' => sub {
- my $c = shift;
- $c->helpers->render_query(scalar $c->param('queries'), {update => 1});
- };
- get '/plaintext' => sub { shift->render( text => 'Hello, World!' ) };
- # Additional helpers (shared code)
- helper 'render_query' => sub {
- my ($self, $q, $args) = @_;
- $args ||= {};
- my $update = $args->{update};
- $self->render_later;
- $q = 1 unless looks_like_number($q);
- $q = 1 if $q < 1;
- $q = 500 if $q > 500;
- my $r = [];
- my $tx = $self->tx;
- my $delay = Mojo::IOLoop->delay;
- $delay->on(finish => sub{
- $r = $r->[0] if $args->{single};
- $self->helpers->render_json($r) unless $tx->is_finished;
- });
- my $world = $self->helpers->world;
- foreach (1 .. $q) {
- my $id = int rand 10_000;
- my $end = $delay->begin;
- $world->find_one({_id => $id} => sub {
- my ($world, $err, $doc) = @_;
- if ($update) { $doc->{randomNumber} = 1 + int rand 10_000 };
- push @$r, { id => $id, randomNumber => $doc->{randomNumber} };
- $update ? $world->save($doc, $end) : $end->();
- });
- }
- # use this line if not running under a Mojolicious server
- # $delay->wait unless $delay->ioloop->is_running;
- };
- app->start;
- __DATA__
- @@ fortunes.html.ep
- <!DOCTYPE html>
- <html>
- <head><title>Fortunes</title></head>
- <body>
- <table>
- <tr><th>id</th><th>message</th></tr>
- % foreach my $doc (sort { $a->{message} cmp $b->{message} } @$docs) {
- <tr>
- <td><%= $doc->{_id} %></td>
- <td><%= $doc->{message} %></td>
- </tr>
- % }
- </table>
- </body>
- </html>
|