fix: chunk key when delete

Author: yxq

system/p2p/dht/protocol/p2pstore/p2pstore.go

@@ -200,7 +200,7 @@ func (p *Protocol) processLocalChunk() {
 			}
 		case info := <-p.chunkToDelete:
 			if localInfo, ok := p.getChunkInfoByHash(info.ChunkHash); ok && time.Since(localInfo.Time) > types2.RefreshInterval*3 {
-				if err := p.deleteChunkBlock(localInfo.ChunkHash); err != nil {
+				if err := p.deleteChunkBlock(localInfo.ChunkInfoMsg); err != nil {
 					log.Error("processLocalChunk", "deleteChunkBlock error", err, "chunkHash", hex.EncodeToString(localInfo.ChunkHash), "start", localInfo.Start)
 				}
 			}

system/p2p/dht/protocol/p2pstore/p2pstore_test.go

@@ -119,7 +119,11 @@ func TestInit(t *testing.T) {
 	require.Equal(t, 223, len(msg.Data.(*types.BlockBodys).Items))
 
 	// 删除数据后应该找不到数据
-	err = p2.deleteChunkBlock([]byte("test0"))
+	err = p2.deleteChunkBlock(&types.ChunkInfoMsg{
+		ChunkHash: []byte("test0"),
+		Start:	   0,
+		End:       999,
+	})
 	if err != nil {
 		t.Fatal(err)
 	}

system/p2p/dht/protocol/p2pstore/store.go

@@ -53,16 +53,16 @@ func (p *Protocol) updateChunk(req *types.ChunkInfoMsg) error {
 	return types2.ErrNotFound
 }
 
-func (p *Protocol) deleteChunkBlock(hash []byte) error {
-	exist, err := p.deleteLocalChunkInfo(hash)
+func (p *Protocol) deleteChunkBlock(msg *types.ChunkInfoMsg) error {
+	exist, err := p.deleteLocalChunkInfo(msg.ChunkHash)
 	if err != nil {
 		return err
 	}
 	if !exist {
 		return nil
 	}
 	batch := p.DB.NewBatch(true)
-	it := p.DB.Iterator(hash, append(hash, ':'+1), false)
+	it := p.DB.Iterator(genChunkDBKey(msg.Start), genChunkDBKey(msg.End+1), false)
 	defer it.Close()
 	for it.Next(); it.Valid(); it.Next() {
 		batch.Delete(it.Key())