Index: trunk/db_postgres.go
===================================================================
--- trunk/db_postgres.go	(revision 645)
+++ trunk/db_postgres.go	(revision 652)
@@ -148,6 +148,6 @@
 }
 
-func (db *PostgresDB) Stats() (*DatabaseStats, error) {
-	ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
+func (db *PostgresDB) Stats(ctx context.Context) (*DatabaseStats, error) {
+	ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
 	defer cancel()
 
@@ -164,6 +164,6 @@
 }
 
-func (db *PostgresDB) ListUsers() ([]User, error) {
-	ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
+func (db *PostgresDB) ListUsers(ctx context.Context) ([]User, error) {
+	ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
 	defer cancel()
 
@@ -193,6 +193,6 @@
 }
 
-func (db *PostgresDB) GetUser(username string) (*User, error) {
-	ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
+func (db *PostgresDB) GetUser(ctx context.Context, username string) (*User, error) {
+	ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
 	defer cancel()
 
@@ -211,6 +211,6 @@
 }
 
-func (db *PostgresDB) StoreUser(user *User) error {
-	ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
+func (db *PostgresDB) StoreUser(ctx context.Context, user *User) error {
+	ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
 	defer cancel()
 
@@ -235,6 +235,6 @@
 }
 
-func (db *PostgresDB) DeleteUser(id int64) error {
-	ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
+func (db *PostgresDB) DeleteUser(ctx context.Context, id int64) error {
+	ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
 	defer cancel()
 
@@ -243,6 +243,6 @@
 }
 
-func (db *PostgresDB) ListNetworks(userID int64) ([]Network, error) {
-	ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
+func (db *PostgresDB) ListNetworks(ctx context.Context, userID int64) ([]Network, error) {
+	ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
 	defer cancel()
 
@@ -287,6 +287,6 @@
 }
 
-func (db *PostgresDB) StoreNetwork(userID int64, network *Network) error {
-	ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
+func (db *PostgresDB) StoreNetwork(ctx context.Context, userID int64, network *Network) error {
+	ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
 	defer cancel()
 
@@ -339,6 +339,6 @@
 }
 
-func (db *PostgresDB) DeleteNetwork(id int64) error {
-	ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
+func (db *PostgresDB) DeleteNetwork(ctx context.Context, id int64) error {
+	ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
 	defer cancel()
 
@@ -347,6 +347,6 @@
 }
 
-func (db *PostgresDB) ListChannels(networkID int64) ([]Channel, error) {
-	ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
+func (db *PostgresDB) ListChannels(ctx context.Context, networkID int64) ([]Channel, error) {
+	ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
 	defer cancel()
 
@@ -381,6 +381,6 @@
 }
 
-func (db *PostgresDB) StoreChannel(networkID int64, ch *Channel) error {
-	ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
+func (db *PostgresDB) StoreChannel(ctx context.Context, networkID int64, ch *Channel) error {
+	ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
 	defer cancel()
 
@@ -409,6 +409,6 @@
 }
 
-func (db *PostgresDB) DeleteChannel(id int64) error {
-	ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
+func (db *PostgresDB) DeleteChannel(ctx context.Context, id int64) error {
+	ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
 	defer cancel()
 
@@ -417,6 +417,6 @@
 }
 
-func (db *PostgresDB) ListDeliveryReceipts(networkID int64) ([]DeliveryReceipt, error) {
-	ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
+func (db *PostgresDB) ListDeliveryReceipts(ctx context.Context, networkID int64) ([]DeliveryReceipt, error) {
+	ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
 	defer cancel()
 
@@ -445,6 +445,6 @@
 }
 
-func (db *PostgresDB) StoreClientDeliveryReceipts(networkID int64, client string, receipts []DeliveryReceipt) error {
-	ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
+func (db *PostgresDB) StoreClientDeliveryReceipts(ctx context.Context, networkID int64, client string, receipts []DeliveryReceipt) error {
+	ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
 	defer cancel()
 
