app.pl 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111
  1. use Mojolicious::Lite;
  2. use Mango;
  3. use JSON::XS 'encode_json';
  4. use Scalar::Util 'looks_like_number';
  5. # configuration
  6. plugin JSONConfig => {
  7. file => 'app.conf',
  8. default => {
  9. database_host => 'localhost',
  10. workers => 8,
  11. },
  12. };
  13. app->config->{hypnotoad}{workers} = app->config->{workers};
  14. # Database connections
  15. helper mango => sub { state $mango = Mango->new('mongodb://'. shift->config->{database_host} . ':27017') };
  16. helper db => sub { state $db = shift->mango->db('hello_world') };
  17. helper world => sub { shift->db->collection('World') };
  18. helper fortune => sub { shift->db->collection('Fortune') };
  19. # JSON::XS renderer
  20. helper render_json => sub { shift->render( data => encode_json(shift), format => 'json' ) };
  21. # Routes
  22. get '/json' => sub { shift->render_json({message => 'Hello, World!'}) };
  23. get '/db' => sub { shift->render_query(1) };
  24. get '/queries' => sub {
  25. my $c = shift;
  26. $c->render_query($c->param('queries'));
  27. };
  28. get '/fortunes' => sub {
  29. my $c = shift->render_later;
  30. my $tx = $c->tx;
  31. $c->fortune->find->all(sub{
  32. my ($cursor, $err, $docs) = @_;
  33. push @$docs, { _id => 0, message => 'Additional fortune added at request time.' };
  34. $c->render( fortunes => docs => $docs ) unless $tx->is_finished;
  35. });
  36. };
  37. get '/updates' => sub {
  38. my $c = shift;
  39. $c->render_query($c->param('queries'), 1);
  40. };
  41. get '/plaintext' => sub { shift->render( text => 'Hello, World!' ) };
  42. # Additional helpers (shared code)
  43. helper 'render_query' => sub {
  44. my ($self, $q, $update) = @_;
  45. $self->render_later;
  46. $q = 1 unless looks_like_number($q);
  47. $q = 500 if $q > 500;
  48. my $r = [];
  49. my $tx = $self->tx;
  50. my $delay = Mojo::IOLoop->delay;
  51. $delay->on(finish => sub{
  52. $self->render_json($r) unless $tx->is_finished;
  53. });
  54. my $world = $self->world;
  55. foreach (1 .. $q) {
  56. my $id = int rand 10_000;
  57. my $end = $delay->begin;
  58. $world->find_one({_id => $id} => sub {
  59. my ($world, $err, $doc) = @_;
  60. if ($update) { $doc->{randomNumber} = 1 + int rand 10_000 };
  61. push @$r, { id => $id, randomNumber => $doc->{randomNumber} };
  62. $update ? $world->save($doc, $end) : $end->();
  63. });
  64. }
  65. # use this line if not running under a Mojolicious server
  66. # $delay->wait unless $delay->ioloop->is_running;
  67. };
  68. app->start;
  69. __DATA__
  70. @@ fortunes.html.ep
  71. <!DOCTYPE html>
  72. <html>
  73. <head><title>Fortunes</title></head>
  74. <body>
  75. <table>
  76. <tr><th>id</th><th>message</th></tr>
  77. % foreach my $doc (sort { $a->{message} cmp $b->{message} } @$docs) {
  78. <tr>
  79. <td><%= $doc->{_id} %></td>
  80. <td><%= $doc->{message} %></td>
  81. </tr>
  82. % }
  83. </table>
  84. </body>
  85. </html>