Changeset 652 in code for trunk/db_sqlite.go


Ignore:
Timestamp:
Oct 18, 2021, 5:15:15 PM (4 years ago)
Author:
contact
Message:

Add context args to Database interface

This is a mecanical change, which just lifts up the context.TODO()
calls from inside the DB implementations to the callers.

Future work involves properly wiring up the contexts when it makes
sense.

File:
1 edited

Legend:

Unmodified
Added
Removed

  • trunk/db_sqlite.go

    r645 r652  
    209209}
    210210
    211 func (db *SqliteDB) Stats() (*DatabaseStats, error) {
     211func (db *SqliteDB) Stats(ctx context.Context) (*DatabaseStats, error) {
    212212        db.lock.RLock()
    213213        defer db.lock.RUnlock()
    214214
    215         ctx, cancel := context.WithTimeout(context.TODO(), sqliteQueryTimeout)
     215        ctx, cancel := context.WithTimeout(ctx, sqliteQueryTimeout)
    216216        defer cancel()
    217217
    … …  
    235235}
    236236
    237 func (db *SqliteDB) ListUsers() ([]User, error) {
     237func (db *SqliteDB) ListUsers(ctx context.Context) ([]User, error) {
    238238        db.lock.RLock()
    239239        defer db.lock.RUnlock()
    240240
    241         ctx, cancel := context.WithTimeout(context.TODO(), sqliteQueryTimeout)
     241        ctx, cancel := context.WithTimeout(ctx, sqliteQueryTimeout)
    242242        defer cancel()
    243243
    … …  
    267267}
    268268
    269 func (db *SqliteDB) GetUser(username string) (*User, error) {
     269func (db *SqliteDB) GetUser(ctx context.Context, username string) (*User, error) {
    270270        db.lock.RLock()
    271271        defer db.lock.RUnlock()
    272272
    273         ctx, cancel := context.WithTimeout(context.TODO(), sqliteQueryTimeout)
     273        ctx, cancel := context.WithTimeout(ctx, sqliteQueryTimeout)
    274274        defer cancel()
    275275
    … …  
    288288}
    289289
    290 func (db *SqliteDB) StoreUser(user *User) error {
    291         db.lock.Lock()
    292         defer db.lock.Unlock()
    293 
    294         ctx, cancel := context.WithTimeout(context.TODO(), sqliteQueryTimeout)
     290func (db *SqliteDB) StoreUser(ctx context.Context, user *User) error {
     291        db.lock.Lock()
     292        defer db.lock.Unlock()
     293
     294        ctx, cancel := context.WithTimeout(ctx, sqliteQueryTimeout)
    295295        defer cancel()
    296296
    … …  
    324324}
    325325
    326 func (db *SqliteDB) DeleteUser(id int64) error {
    327         db.lock.Lock()
    328         defer db.lock.Unlock()
    329 
    330         ctx, cancel := context.WithTimeout(context.TODO(), sqliteQueryTimeout)
     326func (db *SqliteDB) DeleteUser(ctx context.Context, id int64) error {
     327        db.lock.Lock()
     328        defer db.lock.Unlock()
     329
     330        ctx, cancel := context.WithTimeout(ctx, sqliteQueryTimeout)
    331331        defer cancel()
    332332
    … …  
    372372}
    373373
    374 func (db *SqliteDB) ListNetworks(userID int64) ([]Network, error) {
     374func (db *SqliteDB) ListNetworks(ctx context.Context, userID int64) ([]Network, error) {
    375375        db.lock.RLock()
    376376        defer db.lock.RUnlock()
    377377
    378         ctx, cancel := context.WithTimeout(context.TODO(), sqliteQueryTimeout)
     378        ctx, cancel := context.WithTimeout(ctx, sqliteQueryTimeout)
    379379        defer cancel()
    380380
    … …  
    421421}
    422422
    423 func (db *SqliteDB) StoreNetwork(userID int64, network *Network) error {
    424         db.lock.Lock()
    425         defer db.lock.Unlock()
    426 
    427         ctx, cancel := context.WithTimeout(context.TODO(), sqliteQueryTimeout)
     423func (db *SqliteDB) StoreNetwork(ctx context.Context, userID int64, network *Network) error {
     424        db.lock.Lock()
     425        defer db.lock.Unlock()
     426
     427        ctx, cancel := context.WithTimeout(ctx, sqliteQueryTimeout)
    428428        defer cancel()
    429429
    … …  
    491491}
    492492
    493 func (db *SqliteDB) DeleteNetwork(id int64) error {
    494         db.lock.Lock()
    495         defer db.lock.Unlock()
    496 
    497         ctx, cancel := context.WithTimeout(context.TODO(), sqliteQueryTimeout)
     493func (db *SqliteDB) DeleteNetwork(ctx context.Context, id int64) error {
     494        db.lock.Lock()
     495        defer db.lock.Unlock()
     496
     497        ctx, cancel := context.WithTimeout(ctx, sqliteQueryTimeout)
    498498        defer cancel()
    499499
    … …  
    522522}
    523523
    524 func (db *SqliteDB) ListChannels(networkID int64) ([]Channel, error) {
     524func (db *SqliteDB) ListChannels(ctx context.Context, networkID int64) ([]Channel, error) {
    525525        db.lock.RLock()
    526526        defer db.lock.RUnlock()
    527527
    528         ctx, cancel := context.WithTimeout(context.TODO(), sqliteQueryTimeout)
     528        ctx, cancel := context.WithTimeout(ctx, sqliteQueryTimeout)
    529529        defer cancel()
    530530
    … …  
    559559}
    560560
    561 func (db *SqliteDB) StoreChannel(networkID int64, ch *Channel) error {
    562         db.lock.Lock()
    563         defer db.lock.Unlock()
    564 
    565         ctx, cancel := context.WithTimeout(context.TODO(), sqliteQueryTimeout)
     561func (db *SqliteDB) StoreChannel(ctx context.Context, networkID int64, ch *Channel) error {
     562        db.lock.Lock()
     563        defer db.lock.Unlock()
     564
     565        ctx, cancel := context.WithTimeout(ctx, sqliteQueryTimeout)
    566566        defer cancel()
    567567
    … …  
    599599}
    600600
    601 func (db *SqliteDB) DeleteChannel(id int64) error {
    602         db.lock.Lock()
    603         defer db.lock.Unlock()
    604 
    605         ctx, cancel := context.WithTimeout(context.TODO(), sqliteQueryTimeout)
     601func (db *SqliteDB) DeleteChannel(ctx context.Context, id int64) error {
     602        db.lock.Lock()
     603        defer db.lock.Unlock()
     604
     605        ctx, cancel := context.WithTimeout(ctx, sqliteQueryTimeout)
    606606        defer cancel()
    607607
    … …  
    610610}
    611611
    612 func (db *SqliteDB) ListDeliveryReceipts(networkID int64) ([]DeliveryReceipt, error) {
     612func (db *SqliteDB) ListDeliveryReceipts(ctx context.Context, networkID int64) ([]DeliveryReceipt, error) {
    613613        db.lock.RLock()
    614614        defer db.lock.RUnlock()
    615615
    616         ctx, cancel := context.WithTimeout(context.TODO(), sqliteQueryTimeout)
     616        ctx, cancel := context.WithTimeout(ctx, sqliteQueryTimeout)
    617617        defer cancel()
    618618
    … …  
    643643}
    644644
    645 func (db *SqliteDB) StoreClientDeliveryReceipts(networkID int64, client string, receipts []DeliveryReceipt) error {
    646         db.lock.Lock()
    647         defer db.lock.Unlock()
    648 
    649         ctx, cancel := context.WithTimeout(context.TODO(), sqliteQueryTimeout)
     645func (db *SqliteDB) StoreClientDeliveryReceipts(ctx context.Context, networkID int64, client string, receipts []DeliveryReceipt) error {
     646        db.lock.Lock()
     647        defer db.lock.Unlock()
     648
     649        ctx, cancel := context.WithTimeout(ctx, sqliteQueryTimeout)
    650650        defer cancel()
    651651
Note: See TracChangeset for help on using the changeset viewer.