32 lines
999 B
Dart
32 lines
999 B
Dart
import 'dart:async';
|
|
import 'dart:io';
|
|
import 'package:angel_orm/angel_orm.dart';
|
|
import 'package:postgres/postgres.dart';
|
|
|
|
Future<PostgresExecutor> connectToPostgres(Iterable<String> schemas) async {
|
|
var conn = new PostgreSQLConnection('127.0.0.1', 5432, 'angel_orm_test',
|
|
username: Platform.environment['POSTGRES_USERNAME'] ?? 'postgres',
|
|
password: Platform.environment['POSTGRES_PASSWORD'] ?? 'password');
|
|
await conn.open();
|
|
|
|
for (var s in schemas)
|
|
await conn
|
|
.execute(await new File('test/models/$s.up.g.sql').readAsString());
|
|
|
|
return new PostgresExecutor(conn);
|
|
}
|
|
|
|
class PostgresExecutor extends QueryExecutor {
|
|
final PostgreSQLConnection connection;
|
|
|
|
PostgresExecutor(this.connection);
|
|
|
|
Future close() => connection.close();
|
|
|
|
@override
|
|
Future<List<List>> query(String query, List<String> returningFields) {
|
|
var fields = returningFields.join(', ');
|
|
var returning = 'RETURNING ($fields)';
|
|
return connection.query('$query $returning');
|
|
}
|
|
}
|