types cleanup
This commit is contained in:
@@ -9,7 +9,7 @@ import (
|
||||
|
||||
type PacketQueue struct {
|
||||
mu sync.RWMutex
|
||||
expectedPackages map[uint32]*Listener
|
||||
expectedPackages map[PoolMessage]*Listener
|
||||
}
|
||||
|
||||
type CallResult struct {
|
||||
@@ -24,7 +24,7 @@ type Listener struct {
|
||||
|
||||
func NewPacketQueue(connection *PersistentConnection) *PacketQueue {
|
||||
queue := &PacketQueue{
|
||||
expectedPackages: make(map[uint32]*Listener),
|
||||
expectedPackages: make(map[PoolMessage]*Listener),
|
||||
}
|
||||
go queue.HandleConnection(connection)
|
||||
return queue
|
||||
@@ -36,7 +36,7 @@ func (p *PacketQueue) RemoveListeners() {
|
||||
for _, l := range p.expectedPackages {
|
||||
close(l.Chan)
|
||||
}
|
||||
p.expectedPackages = make(map[uint32]*Listener)
|
||||
p.expectedPackages = make(map[PoolMessage]*Listener)
|
||||
}
|
||||
|
||||
func (p *PacketQueue) HandleConnection(connection *PersistentConnection) error {
|
||||
@@ -74,7 +74,7 @@ func (p *PacketQueue) HandleConnection(connection *PersistentConnection) error {
|
||||
}
|
||||
}
|
||||
|
||||
func (p *PacketQueue) HandleData(t uint32, data CallResult) {
|
||||
func (p *PacketQueue) HandleData(t PoolMessage, data CallResult) {
|
||||
p.mu.Lock()
|
||||
defer p.mu.Unlock()
|
||||
l, ok := p.expectedPackages[t]
|
||||
@@ -89,7 +89,7 @@ func (p *PacketQueue) HandleData(t uint32, data CallResult) {
|
||||
}
|
||||
}
|
||||
|
||||
func (p *PacketQueue) Expect(messageType uint32) <-chan CallResult {
|
||||
func (p *PacketQueue) Expect(messageType PoolMessage) <-chan CallResult {
|
||||
p.mu.Lock()
|
||||
defer p.mu.Unlock()
|
||||
l, ok := p.expectedPackages[messageType]
|
||||
|
||||
Reference in New Issue
Block a user