@@ -88,14 +88,14 @@ func ListGPGKeys(ctx context.Context, uid int64, listOptions db.ListOptions) ([]
8888}
8989
9090// CountUserGPGKeys return number of gpg keys a user own
91- func CountUserGPGKeys (userID int64 ) (int64 , error ) {
92- return db .GetEngine (db . DefaultContext ).Where ("owner_id=? AND primary_key_id=''" , userID ).Count (& GPGKey {})
91+ func CountUserGPGKeys (ctx context. Context , userID int64 ) (int64 , error ) {
92+ return db .GetEngine (ctx ).Where ("owner_id=? AND primary_key_id=''" , userID ).Count (& GPGKey {})
9393}
9494
9595// GetGPGKeyByID returns public key by given ID.
96- func GetGPGKeyByID (keyID int64 ) (* GPGKey , error ) {
96+ func GetGPGKeyByID (ctx context. Context , keyID int64 ) (* GPGKey , error ) {
9797 key := new (GPGKey )
98- has , err := db .GetEngine (db . DefaultContext ).ID (keyID ).Get (key )
98+ has , err := db .GetEngine (ctx ).ID (keyID ).Get (key )
9999 if err != nil {
100100 return nil , err
101101 } else if ! has {
@@ -105,9 +105,9 @@ func GetGPGKeyByID(keyID int64) (*GPGKey, error) {
105105}
106106
107107// GetGPGKeysByKeyID returns public key by given ID.
108- func GetGPGKeysByKeyID (keyID string ) ([]* GPGKey , error ) {
108+ func GetGPGKeysByKeyID (ctx context. Context , keyID string ) ([]* GPGKey , error ) {
109109 keys := make ([]* GPGKey , 0 , 1 )
110- return keys , db .GetEngine (db . DefaultContext ).Where ("key_id=?" , keyID ).Find (& keys )
110+ return keys , db .GetEngine (ctx ).Where ("key_id=?" , keyID ).Find (& keys )
111111}
112112
113113// GPGKeyToEntity retrieve the imported key and the traducted entity
@@ -224,8 +224,8 @@ func deleteGPGKey(ctx context.Context, keyID string) (int64, error) {
224224}
225225
226226// DeleteGPGKey deletes GPG key information in database.
227- func DeleteGPGKey (doer * user_model.User , id int64 ) (err error ) {
228- key , err := GetGPGKeyByID (id )
227+ func DeleteGPGKey (ctx context. Context , doer * user_model.User , id int64 ) (err error ) {
228+ key , err := GetGPGKeyByID (ctx , id )
229229 if err != nil {
230230 if IsErrGPGKeyNotExist (err ) {
231231 return nil
@@ -238,7 +238,7 @@ func DeleteGPGKey(doer *user_model.User, id int64) (err error) {
238238 return ErrGPGKeyAccessDenied {doer .ID , key .ID }
239239 }
240240
241- ctx , committer , err := db .TxContext (db . DefaultContext )
241+ ctx , committer , err := db .TxContext (ctx )
242242 if err != nil {
243243 return err
244244 }
0 commit comments