more work done

pkg/raft/kv.go

@@ -2,13 +2,18 @@ package raft
 
 import (
 	"errors"
+	"strings"
 	"sync"
 
 	"git.giftfish.de/ston1th/raftbbolt/msgpack"
 	"github.com/hashicorp/raft"
 )
 
-var ErrNotLeader = errors.New("not leader")
+var (
+	ErrNotLeader      = errors.New("raft: not leader")
+	ErrKeyNotFound    = errors.New("raft: key not found")
+	ErrPrefixNotFound = errors.New("raft: prefix not found")
+)
 
 type kvm map[string][]byte
 
@@ -21,7 +26,11 @@ type kv struct {
 func (kv *kv) Get(k string) ([]byte, error) {
 	kv.mu.Lock()
 	defer kv.mu.Unlock()
-	return kv.m[k], nil
+	v, ok := kv.m[k]
+	if !ok {
+		return nil, ErrKeyNotFound
+	}
+	return v, nil
 }
 
 func (kv *kv) GetPrefix(pk string) (m map[string][]byte, err error) {
@@ -33,6 +42,9 @@ func (kv *kv) GetPrefix(pk string) (m map[string][]byte, err error) {
 			m[k] = v
 		}
 	}
+	if len(m) == 0 {
+		err = ErrPrefixNotFound
+	}
 	return
 }