mirrors/woodpecker
1
0
Fork 0
mirror of https://github.com/woodpecker-ci/woodpecker.git synced 2025-01-11 18:15:28 +00:00
Code Issues Projects Releases Wiki Activity

push common index functions to util.go for reuse

Browse source
This commit is contained in:
Brad Rydzewski 2015-04-13 15:51:15 -07:00
parent 57188ee4f2
commit a50d972d8e
2 changed files with 29 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
datastore/bolt/token.go
View file

@ -1,8 +1,6 @@
package bolt package bolt
import ( import (
"bytes"
"github.com/boltdb/bolt" "github.com/boltdb/bolt"
"github.com/drone/drone/common" "github.com/drone/drone/common"
) )
@ -24,19 +22,10 @@ func (db *DB) GetToken(user, label string) (*common.Token, error) {
func (db *DB) InsertToken(token *common.Token) error { func (db *DB) InsertToken(token *common.Token) error {
key := []byte(token.Login + "/" + token.Label) key := []byte(token.Login + "/" + token.Label)
return db.Update(func(t *bolt.Tx) error { return db.Update(func(t *bolt.Tx) error {
// gets an index err := push(t, bucketUserTokens, []byte(token.Login), key)
var idx [][]byte
err := get(t, bucketUserTokens, []byte(token.Login), &idx)
if err != nil && err != ErrKeyNotFound {
return err
}
// pushes to the index
idx = append(idx, key)
err = update(t, bucketUserTokens, []byte(token.Login), &idx)
if err != nil { if err != nil {
return err return err
} }
return insert(t, bucketTokens, key, token) return insert(t, bucketTokens, key, token)
}) })
} }
@ -45,20 +34,7 @@ func (db *DB) InsertToken(token *common.Token) error {
func (db *DB) DeleteToken(token *common.Token) error { func (db *DB) DeleteToken(token *common.Token) error {
key := []byte(token.Login + "/" + token.Label) key := []byte(token.Login + "/" + token.Label)
return db.Update(func(t *bolt.Tx) error { return db.Update(func(t *bolt.Tx) error {
// gets an index err := splice(t, bucketUserTokens, []byte(token.Login), key)
var idx [][]byte
err := get(t, bucketUserTokens, []byte(token.Login), &idx)
if err != nil && err != ErrKeyNotFound {
return err
}
// pushes to the index
for i, val := range idx {
if bytes.Equal(val, key) {
idx = idx[:i+copy(idx[i:], idx[i+1:])]
break
}
}
err = update(t, bucketUserTokens, []byte(token.Login), &idx)
if err != nil { if err != nil {
return err return err
} }

27
datastore/bolt/util.go
View file

@ -1,6 +1,8 @@
package bolt package bolt
import ( import (
"bytes"
"github.com/boltdb/bolt" "github.com/boltdb/bolt"
"github.com/youtube/vitess/go/bson" "github.com/youtube/vitess/go/bson"
) )
@ -55,3 +57,28 @@ func insert(t *bolt.Tx, bucket, key []byte, v interface{}) error {
func delete(t *bolt.Tx, bucket, key []byte) error { func delete(t *bolt.Tx, bucket, key []byte) error {
return t.Bucket(bucket).Delete(key) return t.Bucket(bucket).Delete(key)
} }
func push(t *bolt.Tx, bucket, index, value []byte) error {
var keys [][]byte
err := get(t, bucket, index, &keys)
if err != nil && err != ErrKeyNotFound {
return err
}
keys = append(keys, value)
return update(t, bucket, index, &keys)
}
func splice(t *bolt.Tx, bucket, index, value []byte) error {
var keys [][]byte
err := get(t, bucket, index, &keys)
if err != nil && err != ErrKeyNotFound {
return err
}
for i, key := range keys {
if bytes.Equal(key, value) {
keys = keys[:i+copy(keys[i:], keys[i+1:])]
break
}
}
return update(t, bucket, index, &keys)
}