Moved ledis implementation to a new package

persistence/ledis/checksum_repository.go

@@ -0,0 +1,62 @@
+package ledis
+
+import (
+	"errors"
+
+	"github.com/cloudsonic/sonic-server/log"
+	"github.com/cloudsonic/sonic-server/scanner"
+	"github.com/siddontang/ledisdb/ledis"
+)
+
+var (
+	checkSumKeyName = []byte("checksums")
+)
+
+type checkSumRepository struct {
+	data map[string]string
+}
+
+func NewCheckSumRepository() scanner.CheckSumRepository {
+	r := &checkSumRepository{}
+	r.loadData()
+	return r
+}
+
+func (r *checkSumRepository) loadData() {
+	r.data = make(map[string]string)
+
+	pairs, err := Db().HGetAll(checkSumKeyName)
+	if err != nil {
+		log.Error("Error loading CheckSums", err)
+	}
+	for _, p := range pairs {
+		r.data[string(p.Field)] = string(p.Value)
+	}
+	log.Debug("Loaded checksums", "total", len(r.data))
+}
+
+func (r *checkSumRepository) Put(id, sum string) error {
+	if id == "" {
+		return errors.New("Id is required")
+	}
+	_, err := Db().HSet(checkSumKeyName, []byte(id), []byte(sum))
+	return err
+}
+
+func (r *checkSumRepository) Get(id string) (string, error) {
+	return r.data[id], nil
+}
+
+func (r *checkSumRepository) SetData(newSums map[string]string) error {
+	Db().HClear(checkSumKeyName)
+	pairs := make([]ledis.FVPair, len(newSums))
+	r.data = make(map[string]string)
+	i := 0
+	for id, sum := range newSums {
+		p := ledis.FVPair{Field: []byte(id), Value: []byte(sum)}
+		pairs[i] = p
+		r.data[id] = sum
+		i++
+	}
+	return Db().HMset(checkSumKeyName, pairs...)
+}