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 Record struct {
Type RecordType Type RecordType
KeyHash []byte KeyHash [28]byte
ValueBytes []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) 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) valueBytes, err := encode(value)
if err != nil { if err != nil {
return nil, err return [sha256.Size224]byte{}, err
} }
return hashBytes(valueBytes), nil return hashBytes(valueBytes), nil
} }
func hashBytes(b []byte) []byte { func hashBytes(b []byte) [sha256.Size224]byte {
bytes := sha256.Sum224(b) return sha256.Sum224(b)
return bytes[:]
} }
func skip(r io.Reader, count int64) (err error) { 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 return err
} }
db.dataOffset[string(record.KeyHash)] = db.offset // TODO: удалить хеш и откатить запись в случае ошибки db.dataOffset[string(record.KeyHash[:])] = db.offset // TODO: удалить хеш и откатить запись в случае ошибки
_, err = db.compressor.Write(b) _, err = db.compressor.Write(b)
if err != nil { if err != nil {
@ -68,7 +68,7 @@ func (db *Database) Get(key, value interface{}) error {
return err return err
} }
offset, exists := db.dataOffset[string(hashToFind)] offset, exists := db.dataOffset[string(hashToFind[:])]
if !exists { if !exists {
return ErrNotExists return ErrNotExists
} }
@ -95,8 +95,8 @@ func (db *Database) Get(key, value interface{}) error {
return err return err
} }
if !bytes.Equal(record.KeyHash, hashToFind) { 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 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) return decode(record.ValueBytes, value)
@ -147,9 +147,9 @@ func Open(filePath string) (*Database, error) {
switch record.Type { switch record.Type {
case RecordTypeSet: case RecordTypeSet:
database.dataOffset[string(record.KeyHash)] = offset database.dataOffset[string(record.KeyHash[:])] = offset
case RecordTypeDelete: case RecordTypeDelete:
delete(database.dataOffset, string(record.KeyHash)) delete(database.dataOffset, string(record.KeyHash[:]))
} }
offset += n offset += n
@ -177,7 +177,7 @@ func (db *Database) Delete(key interface{}) error {
return err return err
} }
delete(db.dataOffset, string(record.KeyHash)) delete(db.dataOffset, string(record.KeyHash[:]))
_, err = db.compressor.Write(b) _, err = db.compressor.Write(b)
if err != nil { if err != nil {