Skip to content

Commit

Permalink
updated datastore (keytransform)
Browse files Browse the repository at this point in the history
  • Loading branch information
jbenet committed Oct 21, 2014
1 parent f7c1ed3 commit 8d7850e
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 7 deletions.
2 changes: 1 addition & 1 deletion core/datastore.go
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ func makeDatastore(cfg config.Datastore) (ds.ThreadSafeDatastore, error) {
if err != nil {
return nil, err
}
ktd := ktds.WrapDatastore(d, u.DsKeyB58Encode)
ktd := ktds.Wrap(d, u.B58KeyConverter)
return syncds.MutexWrap(ktd), nil
}

Expand Down
18 changes: 12 additions & 6 deletions util/key.go
Original file line number Diff line number Diff line change
Expand Up @@ -29,24 +29,30 @@ func KeyFromDsKey(dsk ds.Key) Key {
return Key(dsk.BaseNamespace())
}

// DsKeyB58Encode returns a B58 encoded Datastore key
// B58KeyConverter -- for KeyTransform datastores
// (static as only one obj needed)
var B58KeyConverter = b58KeyConverter{}

type b58KeyConverter struct{}

// ConvertKey returns a B58 encoded Datastore key
// TODO: this is hacky because it encodes every path component. some
// path components may be proper strings already...
func DsKeyB58Encode(dsk ds.Key) ds.Key {
func (b58KeyConverter) ConvertKey(dsk ds.Key) ds.Key {
k := ds.NewKey("/")
for _, n := range dsk.Namespaces() {
k = k.Child(b58.Encode([]byte(n)))
k = k.ChildString(b58.Encode([]byte(n)))
}
return k
}

// DsKeyB58Decode returns a b58 decoded Datastore key
// InvertKey returns a b58 decoded Datastore key
// TODO: this is hacky because it encodes every path component. some
// path components may be proper strings already...
func DsKeyB58Decode(dsk ds.Key) ds.Key {
func (b58KeyConverter) InvertKey(dsk ds.Key) ds.Key {
k := ds.NewKey("/")
for _, n := range dsk.Namespaces() {
k = k.Child(string(b58.Decode(n)))
k = k.ChildString(string(b58.Decode(n)))
}
return k
}
Expand Down

0 comments on commit 8d7850e

Please sign in to comment.