Merge branch 'master' of https://github.com/harmony-one/harmony into HAR-67_libp2p_integration_v2
Richard Liu
6 years ago
commit
bc76946bdb
@ -1,23 +0,0 @@ |
||||
// Copyright (c) 2014, Suryandaru Triandana <syndtr@gmail.com>
|
||||
// All rights reserved.
|
||||
//
|
||||
// Use of this source code is governed by a BSD-style license that can be
|
||||
// found in the LICENSE file.
|
||||
|
||||
// Package ida errors provides common error types used throughout leveldb.
|
||||
package ida |
||||
|
||||
import ( |
||||
"errors" |
||||
) |
||||
|
||||
// Common errors.
|
||||
var ( |
||||
ErrRaptorImpNotFound = New("raptor implementation: not found") |
||||
ErrTimeOut = New("timeout: time's up now") |
||||
) |
||||
|
||||
// New returns an error that formats as the given text.
|
||||
func New(text string) error { |
||||
return errors.New(text) |
||||
} |
||||
@ -1,37 +0,0 @@ |
||||
package ida |
||||
|
||||
import ( |
||||
"time" |
||||
|
||||
"github.com/harmony-one/harmony/p2p" |
||||
) |
||||
|
||||
// IDAImp implements IDA interface.
|
||||
type IDAImp struct { |
||||
raptorQImp RaptorQ |
||||
} |
||||
|
||||
// TakeRaptorQ takes RaptorQ implementation.
|
||||
func (ida *IDAImp) TakeRaptorQ(raptorQImp RaptorQ) { |
||||
ida.raptorQImp = raptorQImp |
||||
} |
||||
|
||||
// Process implements very simple IDA logic.
|
||||
func (ida *IDAImp) Process(msg Message, peers []p2p.Peer, done chan struct{}, timeout time.Duration) error { |
||||
if ida.raptorQImp == nil { |
||||
return ErrRaptorImpNotFound |
||||
} |
||||
chunkStream := ida.raptorQImp.Process(msg) |
||||
id := 0 |
||||
for { |
||||
select { |
||||
case <-done: |
||||
return nil |
||||
case <-time.After(timeout): |
||||
return ErrTimeOut |
||||
case chunk := <-chunkStream: |
||||
p2p.SendMessage(peers[id], chunk) |
||||
id++ |
||||
} |
||||
} |
||||
} |
||||
@ -1,77 +0,0 @@ |
||||
package ida |
||||
|
||||
import ( |
||||
"math/rand" |
||||
"testing" |
||||
"time" |
||||
|
||||
"github.com/harmony-one/harmony/p2p" |
||||
) |
||||
|
||||
var ( |
||||
a = Symbol{0, 0} |
||||
b = Symbol{1, 0} |
||||
c = Symbol{0, 1} |
||||
d = Symbol{1, 1} |
||||
) |
||||
|
||||
type FakeRaptor struct { |
||||
symbols []Symbol |
||||
done chan struct{} |
||||
r *rand.Rand |
||||
} |
||||
|
||||
func (raptor *FakeRaptor) Init() { |
||||
// raptor.symbols = make([]Symbol, 4)
|
||||
raptor.symbols = make([]Symbol, 4) |
||||
raptor.symbols[0] = make(Symbol, 2) |
||||
copy(raptor.symbols[0], a) |
||||
raptor.symbols[1] = make(Symbol, 2) |
||||
copy(raptor.symbols[1], b) |
||||
raptor.symbols[2] = make(Symbol, 2) |
||||
copy(raptor.symbols[2], c) |
||||
raptor.symbols[3] = make(Symbol, 2) |
||||
copy(raptor.symbols[3], d) |
||||
raptor.r = rand.New(rand.NewSource(99)) |
||||
raptor.done = make(chan struct{}) |
||||
} |
||||
|
||||
func (raptor *FakeRaptor) generate(res chan Symbol) { |
||||
for { |
||||
select { |
||||
case <-raptor.done: |
||||
return |
||||
default: |
||||
i := raptor.r.Intn(4) |
||||
res <- raptor.symbols[i] |
||||
} |
||||
} |
||||
} |
||||
|
||||
func (raptor *FakeRaptor) Process(msg Message) chan Symbol { |
||||
res := make(chan Symbol) |
||||
go raptor.generate(res) |
||||
return res |
||||
} |
||||
|
||||
func TestShouldReturnErrRaptorImpNotFound(t *testing.T) { |
||||
ida := &IDAImp{} |
||||
done := make(chan struct{}) |
||||
err := ida.Process([]byte{}, []p2p.Peer{}, done, time.Second) |
||||
if err != ErrRaptorImpNotFound { |
||||
t.Fatal("Should return an error") |
||||
} |
||||
} |
||||
|
||||
func TestSimple(t *testing.T) { |
||||
raptor := &FakeRaptor{} |
||||
raptor.Init() |
||||
|
||||
// ida := &IDAImp{}
|
||||
// done := make(chan struct{})
|
||||
// ida.TakeRaptorQ(raptor)
|
||||
// err := ida.Process([]byte{}, []p2p.Peer{}, done, time.Second)
|
||||
// if err == ErrRaptorImpNotFound {
|
||||
// t.Fatal("Should return an error")
|
||||
// }
|
||||
} |
||||
@ -1,25 +0,0 @@ |
||||
package ida |
||||
|
||||
import ( |
||||
"time" |
||||
|
||||
"github.com/harmony-one/harmony/p2p" |
||||
) |
||||
|
||||
// Symbol is produced from a RaptorQ implementation.
|
||||
type Symbol []byte |
||||
|
||||
// Message is type of general message gopssiped
|
||||
type Message []byte |
||||
|
||||
// RaptorQ interface.
|
||||
type RaptorQ interface { |
||||
Init() |
||||
Process(msg Message) chan Symbol |
||||
} |
||||
|
||||
// IDA interface.
|
||||
type IDA interface { |
||||
TakeRaptorQ(raptorQImp *RaptorQ) |
||||
Process(msg Message, peers []p2p.Peer, done chan struct{}, timeout time.Duration) error |
||||
} |
||||
Loading…
Reference in new issue