Implement package cache API
This commit is contained in:
parent
ac74e450c3
commit
d04a9de932
30
distro/debian/cache.go
Normal file
30
distro/debian/cache.go
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
package debian
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/rapidloop/skv"
|
||||||
|
|
||||||
|
"code.dumpstack.io/tools/out-of-tree/distro/debian/snapshot"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Cache struct {
|
||||||
|
store *skv.KVStore
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewCache(path string) (c *Cache, err error) {
|
||||||
|
c = &Cache{}
|
||||||
|
c.store, err = skv.Open(path)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c Cache) Put(p snapshot.Package) error {
|
||||||
|
return c.store.Put(p.Version, p)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c Cache) Get(version string) (p snapshot.Package, err error) {
|
||||||
|
err = c.store.Get(version, &p)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c Cache) Close() error {
|
||||||
|
return c.store.Close()
|
||||||
|
}
|
68
distro/debian/cache_test.go
Normal file
68
distro/debian/cache_test.go
Normal file
@ -0,0 +1,68 @@
|
|||||||
|
package debian
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/rapidloop/skv"
|
||||||
|
|
||||||
|
"code.dumpstack.io/tools/out-of-tree/distro/debian/snapshot"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestCache(t *testing.T) {
|
||||||
|
dir, err := os.MkdirTemp("", "out-of-tree_cache_test_")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer os.RemoveAll(dir)
|
||||||
|
|
||||||
|
path := filepath.Join(dir, "debian.cache")
|
||||||
|
|
||||||
|
c, err := NewCache(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
packages, err := snapshot.Packages("linux", "4.17.14-1", "amd64",
|
||||||
|
`^linux-(image|headers)-[0-9\.\-]*-(amd64|amd64-unsigned)$`)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = c.Put(packages[0])
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
p, err := c.Get(packages[0].Version)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if p.Deb.Hash != packages[0].Deb.Hash {
|
||||||
|
t.Fatalf("mismatch")
|
||||||
|
}
|
||||||
|
|
||||||
|
c.Close()
|
||||||
|
|
||||||
|
c, err = NewCache(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer c.Close()
|
||||||
|
|
||||||
|
p, err = c.Get(packages[0].Version)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if p.Deb.Hash != packages[0].Deb.Hash {
|
||||||
|
t.Fatalf("mismatch")
|
||||||
|
}
|
||||||
|
|
||||||
|
p, err = c.Get("key not exist")
|
||||||
|
if err == nil || err != skv.ErrNotFound {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user