parent
406c9f09f8
commit
2031ec09df
@ -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