36 lines
1.1 KiB
Dart
36 lines
1.1 KiB
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/migrations/$s.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]) {
|
|
if (returningFields != null) {
|
|
var fields = returningFields.join(', ');
|
|
var returning = 'RETURNING $fields';
|
|
query = '$query $returning';
|
|
}
|
|
|
|
print('Running: $query');
|
|
return connection.query(query);
|
|
}
|
|
}
|