|
|
@@ -0,0 +1,251 @@
|
|
|
+import "dart:core";
|
|
|
+import "dart:io";
|
|
|
+import 'dart:async' show Future;
|
|
|
+import 'dart:math' show Random;
|
|
|
+import "package:redstone/server.dart" as app;
|
|
|
+import "package:redstone_mapper/mapper.dart";
|
|
|
+import "package:redstone_mapper/plugin.dart";
|
|
|
+import "package:redstone_mapper_mongo/manager.dart";
|
|
|
+import "package:redstone_mapper_pg/manager.dart";
|
|
|
+import "package:postgresql/postgresql.dart" as pg;
|
|
|
+import "package:di/di.dart";
|
|
|
+import "package:args/args.dart";
|
|
|
+import 'package:yaml/yaml.dart' as yaml;
|
|
|
+import 'package:mustache/mustache.dart' as mustache;
|
|
|
+
|
|
|
+const _WORLD_TABLE_SIZE = 10000;
|
|
|
+
|
|
|
+final _RANDOM = new Random();
|
|
|
+
|
|
|
+class Fortune implements Comparable<Fortune> {
|
|
|
+
|
|
|
+ @Field()
|
|
|
+ int id;
|
|
|
+
|
|
|
+ @Field()
|
|
|
+ String message;
|
|
|
+
|
|
|
+ Fortune([this.id, this.message]);
|
|
|
+
|
|
|
+ compareTo(Fortune other) => message.compareTo(other.message);
|
|
|
+
|
|
|
+}
|
|
|
+
|
|
|
+class World {
|
|
|
+
|
|
|
+ @Field()
|
|
|
+ int id;
|
|
|
+
|
|
|
+ @Field(model: "randomnumber")
|
|
|
+ int randomNumber;
|
|
|
+
|
|
|
+}
|
|
|
+
|
|
|
+class MongoFortune implements Comparable<Fortune> {
|
|
|
+
|
|
|
+ @Field(model: "_id")
|
|
|
+ int id;
|
|
|
+
|
|
|
+ @Field()
|
|
|
+ String message;
|
|
|
+
|
|
|
+ MongoFortune([this.id, this.message]);
|
|
|
+
|
|
|
+ compareTo(Fortune other) => message.compareTo(other.message);
|
|
|
+
|
|
|
+}
|
|
|
+
|
|
|
+class MongoWorld {
|
|
|
+
|
|
|
+ @Field(model: "_id")
|
|
|
+ int id;
|
|
|
+
|
|
|
+ @Field()
|
|
|
+ int randomNumber;
|
|
|
+
|
|
|
+}
|
|
|
+
|
|
|
+///Handle PostgreSql connections
|
|
|
[email protected](r'/pg/.+')
|
|
|
+pgSqlManager(PostgreSqlManager pgSql) {
|
|
|
+ pgSql.getConnection().then((conn) {
|
|
|
+ app.request.attributes["dbConn"] = conn;
|
|
|
+ app.chain.next(() {
|
|
|
+ pgSql.closeConnection(conn, error: app.chain.error);
|
|
|
+ });
|
|
|
+ });
|
|
|
+}
|
|
|
+
|
|
|
+///Handle MongoDb connections
|
|
|
[email protected](r'/mongo/.+')
|
|
|
+mongoDbManager(MongoDbManager mongoDb) {
|
|
|
+ mongoDb.getConnection().then((conn) {
|
|
|
+ app.request.attributes["dbConn"] = conn;
|
|
|
+ app.chain.next(() {
|
|
|
+ mongoDb.closeConnection(conn, error: app.chain.error);
|
|
|
+ });
|
|
|
+ });
|
|
|
+}
|
|
|
+
|
|
|
+///JSON test
|
|
|
[email protected]("/json")
|
|
|
+getJson() => {"message": "Hello, World!"};
|
|
|
+
|
|
|
+///PlainText test
|
|
|
[email protected]("/plaintext")
|
|
|
+getPlainText() => "Hello, World!";
|
|
|
+
|
|
|
+///PostgreSql tests
|
|
|
[email protected]("/pg")
|
|
|
+@Encode()
|
|
|
+class PgTests {
|
|
|
+
|
|
|
+ static const worldQuery = 'SELECT id, randomnumber FROM world WHERE id = @id;';
|
|
|
+ static const worldUpdt = 'UPDATE world SET randomnumber = @randomnumber WHERE id = @id;';
|
|
|
+
|
|
|
+ static const fortuneQuery = 'SELECT id, message FROM fortune;';
|
|
|
+
|
|
|
+ PostgreSql get pgSql => app.request.attributes["dbConn"];
|
|
|
+
|
|
|
+ @app.Route("/db")
|
|
|
+ Future<World> queryTest() {
|
|
|
+ var params = { 'id': _RANDOM.nextInt(_WORLD_TABLE_SIZE) + 1 };
|
|
|
+ return pgSql.query(worldQuery, World, params).then((list) => list[0]);
|
|
|
+ }
|
|
|
+
|
|
|
+ @app.Route("/queries")
|
|
|
+ Future<List<World>> queriesTest() {
|
|
|
+ var queries = _parseQueriesParam(app.request.queryParams.queries);
|
|
|
+ return Future.wait(new List.generate(queries, (_) => queryTest()));
|
|
|
+ }
|
|
|
+
|
|
|
+ @app.Route("/updates")
|
|
|
+ Future<List<World>> updateTest() {
|
|
|
+ var queries = _parseQueriesParam(app.request.queryParams.queries);
|
|
|
+ return Future.wait(new List.generate(queries, (_) => queryTest().then((world) {
|
|
|
+ world.randomNumber = _RANDOM.nextInt(_WORLD_TABLE_SIZE) + 1;
|
|
|
+ return pgSql.execute(worldUpdt, world).then((_) => world);
|
|
|
+ })));
|
|
|
+ }
|
|
|
+
|
|
|
+ @app.Route("/fortunes", responseType: "text/html")
|
|
|
+ Future<String> fortunesTest(@app.Inject() mustache.Template template) {
|
|
|
+ return pgSql.query(fortuneQuery, Fortune).then((values) {
|
|
|
+ values
|
|
|
+ ..add(new Fortune(0, 'Additional fortune added at request time.'))
|
|
|
+ ..sort();
|
|
|
+
|
|
|
+ return template.renderString({
|
|
|
+ "fortunes": encode(values)
|
|
|
+ });
|
|
|
+ });
|
|
|
+ }
|
|
|
+
|
|
|
+}
|
|
|
+
|
|
|
+///MongoDb tests
|
|
|
[email protected]("/mongo")
|
|
|
+@Encode()
|
|
|
+class MongoTests {
|
|
|
+
|
|
|
+ static const worldCollection = "World";
|
|
|
+ static const fortuneCollection = "Fortune";
|
|
|
+
|
|
|
+ MongoDb get mongoDb => app.request.attributes["dbConn"];
|
|
|
+
|
|
|
+ @app.Route("/db")
|
|
|
+ Future<MongoWorld> queryTest() {
|
|
|
+ return mongoDb.findOne(worldCollection, MongoWorld, {
|
|
|
+ "_id": _RANDOM.nextInt(_WORLD_TABLE_SIZE) + 1
|
|
|
+ });
|
|
|
+ }
|
|
|
+
|
|
|
+ @app.Route("/queries")
|
|
|
+ Future<List<MongoWorld>> queriesTest() {
|
|
|
+ var queries = _parseQueriesParam(app.request.queryParams.queries);
|
|
|
+ return Future.wait(new List.generate(queries, (_) => queryTest()));
|
|
|
+ }
|
|
|
+
|
|
|
+ @app.Route("/updates")
|
|
|
+ Future<List<MongoWorld>> updateTest() {
|
|
|
+ var queries = _parseQueriesParam(app.request.queryParams.queries);
|
|
|
+ return Future.wait(new List.generate(queries, (_) => queryTest().then((world) {
|
|
|
+ world.randomNumber = _RANDOM.nextInt(_WORLD_TABLE_SIZE) + 1;
|
|
|
+ return mongoDb.update(worldCollection, { "_id": world.id }, world)
|
|
|
+ .then((_) => world);
|
|
|
+ })));
|
|
|
+ }
|
|
|
+
|
|
|
+ @app.Route("/fortunes", responseType: "text/html")
|
|
|
+ Future<String> fortunesTest(@app.Inject() mustache.Template template) {
|
|
|
+ return mongoDb.find(fortuneCollection, MongoFortune).then((values) {
|
|
|
+ values
|
|
|
+ ..add(new MongoFortune(0, 'Additional fortune added at request time.'))
|
|
|
+ ..sort();
|
|
|
+
|
|
|
+ return template.renderString({
|
|
|
+ "fortunes": encode(values)
|
|
|
+ });
|
|
|
+ });
|
|
|
+ }
|
|
|
+
|
|
|
+}
|
|
|
+
|
|
|
+main(List<String> args) {
|
|
|
+ var parser = new ArgParser();
|
|
|
+ parser.addOption('address', abbr: 'a', defaultsTo: '0.0.0.0');
|
|
|
+ parser.addOption('port', abbr: 'p', defaultsTo: '8080');
|
|
|
+ parser.addOption('dbconnections', abbr: 'd', defaultsTo: '256');
|
|
|
+
|
|
|
+ var arguments = parser.parse(args);
|
|
|
+ var dbConnections = int.parse(arguments["dbconnections"]);
|
|
|
+
|
|
|
+ MongoDbManager mongoDbManager;
|
|
|
+ PostgreSqlManager pgSqlManager;
|
|
|
+ mustache.Template fortunesTemplate;
|
|
|
+
|
|
|
+ Future.wait([
|
|
|
+
|
|
|
+ //load PostgreSql configuration
|
|
|
+ new File("postgresql.yaml").readAsString().then((config){
|
|
|
+ pgSqlManager = new PostgreSqlManager(
|
|
|
+ new pg.Settings.fromMap(yaml.loadYaml(config)).toUri(),
|
|
|
+ min: dbConnections,
|
|
|
+ max: dbConnections);
|
|
|
+ }),
|
|
|
+
|
|
|
+ //load MongoDb configuration
|
|
|
+ new File("mongodb.yaml").readAsString().then((config) {
|
|
|
+ var mongoConfig = yaml.loadYaml(config);
|
|
|
+ mongoDbManager = new MongoDbManager(
|
|
|
+ "mongodb://${mongoConfig["host"]}/${mongoConfig["database"]}",
|
|
|
+ poolSize: dbConnections);
|
|
|
+ }),
|
|
|
+
|
|
|
+ //load fortunes mustache template
|
|
|
+ new File('fortunes.mustache').readAsString().then((template) {
|
|
|
+ fortunesTemplate = mustache.parse(template);
|
|
|
+ })
|
|
|
+
|
|
|
+ ]).then((_) {
|
|
|
+
|
|
|
+ //app.setupConsoleLog();
|
|
|
+
|
|
|
+ //install module for dependency injection
|
|
|
+ app.addModule(new Module()
|
|
|
+ ..bind(MongoDbManager, toValue: mongoDbManager)
|
|
|
+ ..bind(PostgreSqlManager, toValue: pgSqlManager)
|
|
|
+ ..bind(mustache.Template, toValue: fortunesTemplate));
|
|
|
+
|
|
|
+ //initialize mapper plugin
|
|
|
+ app.addPlugin(getMapperPlugin());
|
|
|
+
|
|
|
+ //start the server
|
|
|
+ app.start(address: arguments["address"], port: int.parse(arguments["port"]));
|
|
|
+
|
|
|
+ });
|
|
|
+}
|
|
|
+
|
|
|
+_parseQueriesParam(param) {
|
|
|
+ return param.isEmpty ? 1 : int.parse(param, radix: 10, onError: (_) => 1).clamp(1, 500);
|
|
|
+}
|
Luiz Mineo