fix packets

cosmos · Jun 13, 2019 · 7445bfd · 7445bfd
1 parent 114f6f3
commit 7445bfd
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/x/ibc/04-channel/manager.go b/x/ibc/04-channel/manager.go
@@ -69,7 +69,7 @@ func (man CounterpartyManager) object(connid, chanid string) CounterObject {
 
 		seqsend: commitment.NewInteger(man.protocol.Value([]byte(key+"/nextSequenceSend")), mapping.Dec),
 		seqrecv: commitment.NewInteger(man.protocol.Value([]byte(key+"/nextSequenceRecv")), mapping.Dec),
-		// packets: commitment.NewIndexer(man.protocol.Prefix([]byte(key + "/packets")), mapping.Dec),
+		packets: commitment.NewIndexer(man.protocol.Prefix([]byte(key+"/packets")), mapping.Dec),
 	}
 }
 
@@ -182,7 +182,7 @@ type CounterObject struct {
 
 	seqsend commitment.Integer
 	seqrecv commitment.Integer
-	//packets commitment.Indexer
+	packets commitment.Indexer
 
 	connection connection.CounterObject
 }

diff --git a/x/ibc/23-commitment/value.go b/x/ibc/23-commitment/value.go
@@ -52,6 +52,28 @@ func (m Mapping) Value(key []byte) Value {
 	}
 }
 
+func (m Mapping) Prefix(prefix []byte) Mapping {
+	return Mapping{
+		base: m.base.Prefix(prefix),
+	}
+}
+
+type Indexer struct {
+	Mapping
+	enc mapping.IntEncoding
+}
+
+func NewIndexer(m Mapping, enc mapping.IntEncoding) Indexer {
+	return Indexer{
+		Mapping: m,
+		enc:     enc,
+	}
+}
+
+func (ix Indexer) Value(index uint64) Value {
+	return ix.Mapping.Value(mapping.EncodeInt(index, ix.enc))
+}
+
 type Value struct {
 	base Base
 	key  []byte