Add Peer.mustRequest

diff --git a/peerconn.go b/peerconn.go
index c7a4e8cfceddacfda1f16d4e3ff8296990e5f0a3..91ef37e601f27bf37a51b886f1658b88eb4f5d6f 100644 (file)
--- a/peerconn.go
+++ b/peerconn.go
@@ -585,6 +585,14 @@ func (cn *Peer) shouldRequest(r RequestIndex) error {
        return nil
 }
 
+func (cn *Peer) mustRequest(r RequestIndex) bool {
+       more, err := cn.request(r)
+       if err != nil {
+               panic(err)
+       }
+       return more
+}
+
 func (cn *Peer) request(r RequestIndex) (more bool, err error) {
        if err := cn.shouldRequest(r); err != nil {
                panic(err)

diff --git a/requesting.go b/requesting.go
index 462d681836e44a7d051ef8c535463614aa28fe56..adb1c51a102875ddcd400428c6e3f25e058fdd39 100644 (file)
--- a/requesting.go
+++ b/requesting.go
@@ -304,10 +304,7 @@ func (p *Peer) applyRequestState(next requestState) bool {
                        //)
                        break
                }
-               more, err = p.request(req)
-               if err != nil {
-                       panic(err)
-               }
+               more = p.mustRequest(req)
                if !more {
                        break
                }