nxshock/zkv
1
0
Fork 0
mirror of https://github.com/nxshock/zkv.git synced 2024-11-27 11:21:02 +05:00
Code Issues Projects Releases Wiki Activity

Make constant key length

Browse Source
This commit is contained in:
nxshock 2022-12-02 21:37:34 +05:00
parent d40b88eebb
commit db25842429
3 changed files with 12 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
record.go
View File

@ -16,7 +16,7 @@ const (
type Record struct {
Type RecordType
KeyHash []byte
KeyHash [28]byte
ValueBytes []byte
}

11
utils.go
View File

@ -17,20 +17,17 @@ func decode(b []byte, value interface{}) error {
return gob.NewDecoder(bytes.NewReader(b)).Decode(value)
}
func hashInterface(value interface{}) ([]byte, error) {
func hashInterface(value interface{}) ([sha256.Size224]byte, error) {
valueBytes, err := encode(value)
if err != nil {
return nil, err
return [sha256.Size224]byte{}, err
}
return hashBytes(valueBytes), nil
}
func hashBytes(b []byte) []byte {
bytes := sha256.Sum224(b)
return bytes[:]
func hashBytes(b []byte) [sha256.Size224]byte {
return sha256.Sum224(b)
}
func skip(r io.Reader, count int64) (err error) {

14
zkv.go
View File

@ -47,7 +47,7 @@ func (db *Database) Set(key, value interface{}) error {
return err
}
db.dataOffset[string(record.KeyHash)] = db.offset // TODO: удалить хеш и откатить запись в случае ошибки
db.dataOffset[string(record.KeyHash[:])] = db.offset // TODO: удалить хеш и откатить запись в случае ошибки
_, err = db.compressor.Write(b)
if err != nil {
@ -68,7 +68,7 @@ func (db *Database) Get(key, value interface{}) error {
return err
}
offset, exists := db.dataOffset[string(hashToFind)]
offset, exists := db.dataOffset[string(hashToFind[:])]
if !exists {
return ErrNotExists
}
@ -95,8 +95,8 @@ func (db *Database) Get(key, value interface{}) error {
return err
}
if !bytes.Equal(record.KeyHash, hashToFind) {
return fmt.Errorf("wrong hash on this offset: expected %s, got %s", base64.StdEncoding.EncodeToString(hashToFind), base64.StdEncoding.EncodeToString(record.KeyHash)) // TODO: заменить на константную ошибку
if !bytes.Equal(record.KeyHash[:], hashToFind[:]) {
return fmt.Errorf("wrong hash on this offset: expected %s, got %s", base64.StdEncoding.EncodeToString(hashToFind[:]), base64.StdEncoding.EncodeToString(record.KeyHash[:])) // TODO: заменить на константную ошибку
}
return decode(record.ValueBytes, value)
@ -147,9 +147,9 @@ func Open(filePath string) (*Database, error) {
switch record.Type {
case RecordTypeSet:
database.dataOffset[string(record.KeyHash)] = offset
database.dataOffset[string(record.KeyHash[:])] = offset
case RecordTypeDelete:
delete(database.dataOffset, string(record.KeyHash))
delete(database.dataOffset, string(record.KeyHash[:]))
}
offset += n
@ -177,7 +177,7 @@ func (db *Database) Delete(key interface{}) error {
return err
}
delete(db.dataOffset, string(record.KeyHash))
delete(db.dataOffset, string(record.KeyHash[:]))
_, err = db.compressor.Write(b)
if err != nil {