135 lines
2.0 KiB
Go
135 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"sync"
|
|
"sync/atomic"
|
|
|
|
"github.com/lightningnetwork/lnd/lnwallet"
|
|
|
|
"github.com/btcsuite/btcd/chaincfg"
|
|
"github.com/btcsuite/btcwallet/walletdb"
|
|
)
|
|
|
|
// server...
|
|
type server struct {
|
|
listeners []net.Listener
|
|
peers map[int32]*peer
|
|
|
|
started int32 // atomic
|
|
shutdown int32 // atomic
|
|
|
|
bitcoinNet *chaincfg.Params
|
|
|
|
rpcServer *rpcServer
|
|
lnwallet *lnwallet.LightningWallet
|
|
|
|
db walletdb.DB
|
|
|
|
newPeers chan *peer
|
|
donePeers chan *peer
|
|
|
|
wg sync.WaitGroup
|
|
quit chan struct{}
|
|
}
|
|
|
|
// addPeer...
|
|
func (s *server) addPeer(p *peer) {
|
|
}
|
|
|
|
// removePeer...
|
|
func (s *server) removePeer(p *peer) {
|
|
}
|
|
|
|
// peerManager...
|
|
func (s *server) peerManager() {
|
|
out:
|
|
for {
|
|
select {
|
|
// New peers.
|
|
case p := <-s.newPeers:
|
|
s.addPeer(p)
|
|
// Finished peers.
|
|
case p := <-s.donePeers:
|
|
s.removePeer(p)
|
|
case <-s.quit:
|
|
break out
|
|
}
|
|
}
|
|
s.wg.Done()
|
|
}
|
|
|
|
func (s *server) queryHandler() {
|
|
out:
|
|
for {
|
|
select {
|
|
// TODO(roasbeef): meta-rpc-stuff
|
|
case <-s.quit:
|
|
break out
|
|
}
|
|
}
|
|
|
|
s.wg.Done()
|
|
}
|
|
|
|
// AddPeer...
|
|
func (s *server) AddPeer(p *peer) {
|
|
s.newPeers <- p
|
|
}
|
|
|
|
// listener...
|
|
func (s *server) listener(l net.Listener) {
|
|
for atomic.LoadInt32(&s.shutdown) == 0 {
|
|
conn, err := l.Accept()
|
|
if err != nil {
|
|
// TODO(roasbeef): log
|
|
continue
|
|
}
|
|
// TODO(roasbeef): create new peer, start it's goroutines
|
|
fmt.Println(conn)
|
|
}
|
|
|
|
s.wg.Done()
|
|
}
|
|
|
|
// Start...
|
|
func (s *server) Start() {
|
|
// Already running?
|
|
if atomic.AddInt32(&s.started, 1) != 1 {
|
|
return
|
|
}
|
|
|
|
// Start all the listeners.
|
|
for _, l := range s.listeners {
|
|
s.wg.Add(1)
|
|
go s.listener(l)
|
|
}
|
|
|
|
s.wg.Add(1)
|
|
go s.peerManager()
|
|
|
|
}
|
|
|
|
// Stop...
|
|
func (s *server) Stop() error {
|
|
// Bail if we're already shutting down.
|
|
if atomic.AddInt32(&s.shutdown, 1) != 1 {
|
|
return nil
|
|
}
|
|
|
|
// Stop all the listeners.
|
|
for _, listener := range s.listeners {
|
|
if err := listener.Close(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
s.rpcServer.Stop()
|
|
s.lnwallet.Stop()
|
|
|
|
// Signal all the lingering goroutines to quit.
|
|
close(s.quit)
|
|
return nil
|
|
}
|