Changeset 652 in code for trunk


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.

Location:
trunk
Files:
11 edited

Legend:

Unmodified
Added
Removed

  • trunk/cmd/sojuctl/main.go

    r620 r652  
    33import (
    44        "bufio"
     5        "context"
    56        "flag"
    67        "fmt"
    … …  
    7677                        Admin:    *admin,
    7778                }
    78                 if err := db.StoreUser(&user); err != nil {
     79                if err := db.StoreUser(context.TODO(), &user); err != nil {
    7980                        log.Fatalf("failed to create user: %v", err)
    8081                }
    … …  
    8687                }
    8788
    88                 user, err := db.GetUser(username)
     89                user, err := db.GetUser(context.TODO(), username)
    8990                if err != nil {
    9091                        log.Fatalf("failed to get user: %v", err)
    … …  
    102103
    103104                user.Password = string(hashed)
    104                 if err := db.StoreUser(user); err != nil {
     105                if err := db.StoreUser(context.TODO(), user); err != nil {
    105106                        log.Fatalf("failed to update password: %v", err)
    106107                }

  • trunk/contrib/znc-import.go

    r620 r652  
    33import (
    44        "bufio"
     5        "context"
    56        "flag"
    67        "fmt"
    … …  
    8081        }
    8182
    82         l, err := db.ListUsers()
     83        l, err := db.ListUsers(context.TODO())
    8384        if err != nil {
    8485                log.Fatalf("failed to list users in DB: %v", err)
    … …  
    112113                u.Admin = section.Values.Get("Admin") == "true"
    113114
    114                 if err := db.StoreUser(u); err != nil {
     115                if err := db.StoreUser(context.TODO(), u); err != nil {
    115116                        log.Fatalf("failed to store user %q: %v", username, err)
    116117                }
    117118                userID := u.ID
    118119
    119                 l, err := db.ListNetworks(userID)
     120                l, err := db.ListNetworks(context.TODO(), userID)
    120121                if err != nil {
    121122                        log.Fatalf("failed to list networks for user %q: %v", username, err)
    … …  
    184185                        n.Enabled = section.Values.Get("IRCConnectEnabled") != "false"
    185186
    186                         if err := db.StoreNetwork(userID, n); err != nil {
     187                        if err := db.StoreNetwork(context.TODO(), userID, n); err != nil {
    187188                                logger.Fatalf("failed to store network: %v", err)
    188189                        }
    189190
    190                         l, err := db.ListChannels(n.ID)
     191                        l, err := db.ListChannels(context.TODO(), n.ID)
    191192                        if err != nil {
    192193                                logger.Fatalf("failed to list channels: %v", err)
    … …  
    218219                                ch.Detached = section.Values.Get("Detached") == "true"
    219220
    220                                 if err := db.StoreChannel(n.ID, ch); err != nil {
     221                                if err := db.StoreChannel(context.TODO(), n.ID, ch); err != nil {
    221222                                        logger.Printf("channel %q: failed to store channel: %v", chName, err)
    222223                                }

  • trunk/db.go

    r620 r652  
    22
    33import (
     4        "context"
    45        "fmt"
    56        "net/url"
    … …  
    1011type Database interface {
    1112        Close() error
    12         Stats() (*DatabaseStats, error)
     13        Stats(ctx context.Context) (*DatabaseStats, error)
    1314
    14         ListUsers() ([]User, error)
    15         GetUser(username string) (*User, error)
    16         StoreUser(user *User) error
    17         DeleteUser(id int64) error
     15        ListUsers(ctx context.Context) ([]User, error)
     16        GetUser(ctx context.Context, username string) (*User, error)
     17        StoreUser(ctx context.Context, user *User) error
     18        DeleteUser(ctx context.Context, id int64) error
    1819
    19         ListNetworks(userID int64) ([]Network, error)
    20         StoreNetwork(userID int64, network *Network) error
    21         DeleteNetwork(id int64) error
    22         ListChannels(networkID int64) ([]Channel, error)
    23         StoreChannel(networKID int64, ch *Channel) error
    24         DeleteChannel(id int64) error
     20        ListNetworks(ctx context.Context, userID int64) ([]Network, error)
     21        StoreNetwork(ctx context.Context, userID int64, network *Network) error
     22        DeleteNetwork(ctx context.Context, id int64) error
     23        ListChannels(ctx context.Context, networkID int64) ([]Channel, error)
     24        StoreChannel(ctx context.Context, networKID int64, ch *Channel) error
     25        DeleteChannel(ctx context.Context, id int64) error
    2526
    26         ListDeliveryReceipts(networkID int64) ([]DeliveryReceipt, error)
    27         StoreClientDeliveryReceipts(networkID int64, client string, receipts []DeliveryReceipt) error
     27        ListDeliveryReceipts(ctx context.Context, networkID int64) ([]DeliveryReceipt, error)
     28        StoreClientDeliveryReceipts(ctx context.Context, networkID int64, client string, receipts []DeliveryReceipt) error
    2829}
    2930

  • trunk/db_postgres.go

    r645 r652  
    148148}
    149149
    150 func (db *PostgresDB) Stats() (*DatabaseStats, error) {
    151         ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
     150func (db *PostgresDB) Stats(ctx context.Context) (*DatabaseStats, error) {
     151        ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
    152152        defer cancel()
    153153
    … …  
    164164}
    165165
    166 func (db *PostgresDB) ListUsers() ([]User, error) {
    167         ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
     166func (db *PostgresDB) ListUsers(ctx context.Context) ([]User, error) {
     167        ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
    168168        defer cancel()
    169169
    … …  
    193193}
    194194
    195 func (db *PostgresDB) GetUser(username string) (*User, error) {
    196         ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
     195func (db *PostgresDB) GetUser(ctx context.Context, username string) (*User, error) {
     196        ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
    197197        defer cancel()
    198198
    … …  
    211211}
    212212
    213 func (db *PostgresDB) StoreUser(user *User) error {
    214         ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
     213func (db *PostgresDB) StoreUser(ctx context.Context, user *User) error {
     214        ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
    215215        defer cancel()
    216216
    … …  
    235235}
    236236
    237 func (db *PostgresDB) DeleteUser(id int64) error {
    238         ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
     237func (db *PostgresDB) DeleteUser(ctx context.Context, id int64) error {
     238        ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
    239239        defer cancel()
    240240
    … …  
    243243}
    244244
    245 func (db *PostgresDB) ListNetworks(userID int64) ([]Network, error) {
    246         ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
     245func (db *PostgresDB) ListNetworks(ctx context.Context, userID int64) ([]Network, error) {
     246        ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
    247247        defer cancel()
    248248
    … …  
    287287}
    288288
    289 func (db *PostgresDB) StoreNetwork(userID int64, network *Network) error {
    290         ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
     289func (db *PostgresDB) StoreNetwork(ctx context.Context, userID int64, network *Network) error {
     290        ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
    291291        defer cancel()
    292292
    … …  
    339339}
    340340
    341 func (db *PostgresDB) DeleteNetwork(id int64) error {
    342         ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
     341func (db *PostgresDB) DeleteNetwork(ctx context.Context, id int64) error {
     342        ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
    343343        defer cancel()
    344344
    … …  
    347347}
    348348
    349 func (db *PostgresDB) ListChannels(networkID int64) ([]Channel, error) {
    350         ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
     349func (db *PostgresDB) ListChannels(ctx context.Context, networkID int64) ([]Channel, error) {
     350        ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
    351351        defer cancel()
    352352
    … …  
    381381}
    382382
    383 func (db *PostgresDB) StoreChannel(networkID int64, ch *Channel) error {
    384         ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
     383func (db *PostgresDB) StoreChannel(ctx context.Context, networkID int64, ch *Channel) error {
     384        ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
    385385        defer cancel()
    386386
    … …  
    409409}
    410410
    411 func (db *PostgresDB) DeleteChannel(id int64) error {
    412         ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
     411func (db *PostgresDB) DeleteChannel(ctx context.Context, id int64) error {
     412        ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
    413413        defer cancel()
    414414
    … …  
    417417}
    418418
    419 func (db *PostgresDB) ListDeliveryReceipts(networkID int64) ([]DeliveryReceipt, error) {
    420         ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
     419func (db *PostgresDB) ListDeliveryReceipts(ctx context.Context, networkID int64) ([]DeliveryReceipt, error) {
     420        ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
    421421        defer cancel()
    422422
    … …  
    445445}
    446446
    447 func (db *PostgresDB) StoreClientDeliveryReceipts(networkID int64, client string, receipts []DeliveryReceipt) error {
    448         ctx, cancel := context.WithTimeout(context.TODO(), postgresQueryTimeout)
     447func (db *PostgresDB) StoreClientDeliveryReceipts(ctx context.Context, networkID int64, client string, receipts []DeliveryReceipt) error {
     448        ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
    449449        defer cancel()
    450450

  • 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

  • trunk/downstream.go

    r651 r652  
    22
    33import (
     4        "context"
    45        "crypto/tls"
    56        "encoding/base64"
    … …  
    977978        username, clientName, networkName := unmarshalUsername(username)
    978979
    979         u, err := dc.srv.db.GetUser(username)
     980        u, err := dc.srv.db.GetUser(context.TODO(), username)
    980981        if err != nil {
    981982                dc.logger.Printf("failed authentication for %q: user not found: %v", username, err)
    … …  
    13781379                        }
    13791380                        n.Nick = nick
    1380                         err = dc.srv.db.StoreNetwork(dc.user.ID, &n.Network)
     1381                        err = dc.srv.db.StoreNetwork(context.TODO(), dc.user.ID, &n.Network)
    13811382                })
    13821383                if err != nil {
    … …  
    14281429
    14291430                                n.Realname = storeRealname
    1430                                 if err := dc.srv.db.StoreNetwork(dc.user.ID, &n.Network); err != nil {
     1431                                if err := dc.srv.db.StoreNetwork(context.TODO(), dc.user.ID, &n.Network); err != nil {
    14311432                                        dc.logger.Printf("failed to store network realname: %v", err)
    14321433                                        storeErr = err
    … …  
    15171518                                uc.network.channels.SetValue(upstreamName, ch)
    15181519                        }
    1519                         if err := dc.srv.db.StoreChannel(uc.network.ID, ch); err != nil {
     1520                        if err := dc.srv.db.StoreChannel(context.TODO(), uc.network.ID, ch); err != nil {
    15201521                                dc.logger.Printf("failed to create or update channel %q: %v", upstreamName, err)
    15211522                        }
    … …  
    15491550                                        uc.network.channels.SetValue(upstreamName, ch)
    15501551                                }
    1551                                 if err := dc.srv.db.StoreChannel(uc.network.ID, ch); err != nil {
     1552                                if err := dc.srv.db.StoreChannel(context.TODO(), uc.network.ID, ch); err != nil {
    15521553                                        dc.logger.Printf("failed to create or update channel %q: %v", upstreamName, err)
    15531554                                }
    … …  
    24462447        n.SASL.Plain.Username = username
    24472448        n.SASL.Plain.Password = password
    2448         if err := dc.srv.db.StoreNetwork(dc.user.ID, &n.Network); err != nil {
     2449        if err := dc.srv.db.StoreNetwork(context.TODO(), dc.user.ID, &n.Network); err != nil {
    24492450                dc.logger.Printf("failed to save NickServ credentials: %v", err)
    24502451        }

  • trunk/server.go

    r636 r652  
    22
    33import (
     4        "context"
    45        "fmt"
    56        "log"
    … …  
    8687
    8788func (s *Server) Start() error {
    88         users, err := s.db.ListUsers()
     89        users, err := s.db.ListUsers(context.TODO())
    8990        if err != nil {
    9091                return err
    … …  
    127128        }
    128129
    129         err := s.db.StoreUser(user)
     130        err := s.db.StoreUser(context.TODO(), user)
    130131        if err != nil {
    131132                return nil, fmt.Errorf("could not create user in db: %v", err)

  • trunk/server_test.go

    r623 r652  
    22
    33import (
     4        "context"
    45        "net"
    56        "testing"
    … …  
    4445
    4546        record := &User{Username: testUsername, Password: string(hashed)}
    46         if err := db.StoreUser(record); err != nil {
     47        if err := db.StoreUser(context.TODO(), record); err != nil {
    4748                t.Fatalf("failed to store test user: %v", err)
    4849        }
    … …  
    6970                Enabled: true,
    7071        }
    71         if err := db.StoreNetwork(user.ID, network); err != nil {
     72        if err := db.StoreNetwork(context.TODO(), user.ID, network); err != nil {
    7273                t.Fatalf("failed to store test network: %v", err)
    7374        }

  • trunk/service.go

    r625 r652  
    22
    33import (
     4        "context"
    45        "crypto/sha1"
    56        "crypto/sha256"
    … …  
    658659        net.SASL.Mechanism = "EXTERNAL"
    659660
    660         if err := dc.srv.db.StoreNetwork(dc.user.ID, &net.Network); err != nil {
     661        if err := dc.srv.db.StoreNetwork(context.TODO(), dc.user.ID, &net.Network); err != nil {
    661662                return err
    662663        }
    … …  
    699700        net.SASL.Mechanism = "PLAIN"
    700701
    701         if err := dc.srv.db.StoreNetwork(dc.user.ID, &net.Network); err != nil {
     702        if err := dc.srv.db.StoreNetwork(context.TODO(), dc.user.ID, &net.Network); err != nil {
    702703                return err
    703704        }
    … …  
    723724        net.SASL.Mechanism = ""
    724725
    725         if err := dc.srv.db.StoreNetwork(dc.user.ID, &net.Network); err != nil {
     726        if err := dc.srv.db.StoreNetwork(context.TODO(), dc.user.ID, &net.Network); err != nil {
    726727                return err
    727728        }
    … …  
    861862        u.stop()
    862863
    863         if err := dc.srv.db.DeleteUser(u.ID); err != nil {
     864        if err := dc.srv.db.DeleteUser(context.TODO(), u.ID); err != nil {
    864865                return fmt.Errorf("failed to delete user: %v", err)
    865866        }
    … …  
    10161017        uc.updateChannelAutoDetach(upstreamName)
    10171018
    1018         if err := dc.srv.db.StoreChannel(uc.network.ID, ch); err != nil {
     1019        if err := dc.srv.db.StoreChannel(context.TODO(), uc.network.ID, ch); err != nil {
    10191020                return fmt.Errorf("failed to update channel: %v", err)
    10201021        }
    … …  
    10251026
    10261027func handleServiceServerStatus(dc *downstreamConn, params []string) error {
    1027         dbStats, err := dc.user.srv.db.Stats()
     1028        dbStats, err := dc.user.srv.db.Stats(context.TODO())
    10281029        if err != nil {
    10291030                return err

  • trunk/upstream.go

    r649 r652  
    22
    33import (
     4        "context"
    45        "crypto"
    56        "crypto/sha256"
    … …  
    15171518        if ch.ReattachOn == FilterMessage || (ch.ReattachOn == FilterHighlight && uc.network.isHighlight(msg)) {
    15181519                uc.network.attach(ch)
    1519                 if err := uc.srv.db.StoreChannel(uc.network.ID, ch); err != nil {
     1520                if err := uc.srv.db.StoreChannel(context.TODO(), uc.network.ID, ch); err != nil {
    15201521                        uc.logger.Printf("failed to update channel %q: %v", ch.Name, err)
    15211522                }

  • trunk/user.go

    r644 r652  
    22
    33import (
     4        "context"
    45        "crypto/sha256"
    56        "encoding/binary"
    … …  
    331332        }
    332333
    333         if err := net.user.srv.db.DeleteChannel(ch.ID); err != nil {
     334        if err := net.user.srv.db.DeleteChannel(context.TODO(), ch.ID); err != nil {
    334335                return err
    335336        }
    … …  
    368369        })
    369370
    370         if err := net.user.srv.db.StoreClientDeliveryReceipts(net.ID, clientName, receipts); err != nil {
     371        if err := net.user.srv.db.StoreClientDeliveryReceipts(context.TODO(), net.ID, clientName, receipts); err != nil {
    371372                net.logger.Printf("failed to store delivery receipts for client %q: %v", clientName, err)
    372373        }
    … …  
    488489        }()
    489490
    490         networks, err := u.srv.db.ListNetworks(u.ID)
     491        networks, err := u.srv.db.ListNetworks(context.TODO(), u.ID)
    491492        if err != nil {
    492493                u.logger.Printf("failed to list networks for user %q: %v", u.Username, err)
    … …  
    496497        for _, record := range networks {
    497498                record := record
    498                 channels, err := u.srv.db.ListChannels(record.ID)
     499                channels, err := u.srv.db.ListChannels(context.TODO(), record.ID)
    499500                if err != nil {
    500501                        u.logger.Printf("failed to list channels for user %q, network %q: %v", u.Username, record.GetName(), err)
    … …  
    506507
    507508                if u.hasPersistentMsgStore() {
    508                         receipts, err := u.srv.db.ListDeliveryReceipts(record.ID)
     509                        receipts, err := u.srv.db.ListDeliveryReceipts(context.TODO(), record.ID)
    509510                        if err != nil {
    510511                                u.logger.Printf("failed to load delivery receipts for user %q, network %q: %v", u.Username, network.GetName(), err)
    … …  
    591592                        }
    592593                        uc.network.detach(c)
    593                         if err := uc.srv.db.StoreChannel(uc.network.ID, c); err != nil {
     594                        if err := uc.srv.db.StoreChannel(context.TODO(), uc.network.ID, c); err != nil {
    594595                                u.logger.Printf("failed to store updated detached channel %q: %v", c.Name, err)
    595596                        }
    … …  
    780781
    781782        network := newNetwork(u, record, nil)
    782         err := u.srv.db.StoreNetwork(u.ID, &network.Network)
     783        err := u.srv.db.StoreNetwork(context.TODO(), u.ID, &network.Network)
    783784        if err != nil {
    784785                return nil, err
    … …  
    822823        }
    823824
    824         if err := u.srv.db.StoreNetwork(u.ID, record); err != nil {
     825        if err := u.srv.db.StoreNetwork(context.TODO(), u.ID, record); err != nil {
    825826                return nil, err
    826827        }
    … …  
    889890        }
    890891
    891         if err := u.srv.db.DeleteNetwork(network.ID); err != nil {
     892        if err := u.srv.db.DeleteNetwork(context.TODO(), network.ID); err != nil {
    892893                return err
    893894        }
    … …  
    915916
    916917        realnameUpdated := u.Realname != record.Realname
    917         if err := u.srv.db.StoreUser(record); err != nil {
     918        if err := u.srv.db.StoreUser(context.TODO(), record); err != nil {
    918919                return fmt.Errorf("failed to update user %q: %v", u.Username, err)
    919920        }
Note: See TracChangeset for help on using the changeset viewer.