rename (#465)

[bytom/vapor.git] / protocol / bc / types / block_witness.go

diff --git a/protocol/bc/types/block_witness.go b/protocol/bc/types/block_witness.go
index e4bb2ac..a756cd3 100644 (file)
--- a/protocol/bc/types/block_witness.go
+++ b/protocol/bc/types/block_witness.go
@@ -3,7 +3,7 @@ package types
 import (
        "io"
 
-       "github.com/vapor/encoding/blockchain"
+       "github.com/bytom/vapor/encoding/blockchain"
 )
 
 type BlockWitness struct {
@@ -11,12 +11,34 @@ type BlockWitness struct {
        Witness [][]byte
 }
 
+func (bw *BlockWitness) readFrom(r *blockchain.Reader) (err error) {
+       bw.Witness, err = blockchain.ReadVarstrList(r)
+       return err
+}
+
 func (bw *BlockWitness) writeTo(w io.Writer) error {
        _, err := blockchain.WriteVarstrList(w, bw.Witness)
        return err
 }
 
-func (bw *BlockWitness) readFrom(r *blockchain.Reader) (err error) {
-       bw.Witness, err = blockchain.ReadVarstrList(r)
-       return err
+func (bw *BlockWitness) Set(index uint64, data []byte) {
+       if uint64(len(bw.Witness)) <= index {
+               newWitness := make([][]byte, index+1, index+1)
+               copy(newWitness, bw.Witness)
+               bw.Witness = newWitness
+       }
+       bw.Witness[index] = data
+}
+
+func (bw *BlockWitness) Delete(index uint64) {
+       if uint64(len(bw.Witness)) > index {
+               bw.Witness[index] = nil
+       }
+}
+
+func (bw *BlockWitness) Get(index uint64) []byte {
+       if uint64(len(bw.Witness)) > index {
+               return bw.Witness[index]
+       }
+       return nil
 }