more refactoring
Nick Dumas
1 year ago
parent
3566309c3d
commit
71ee479873
@ -1,62 +1,44 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
|
||||||
"context"
|
"context"
|
||||||
"log"
|
"log"
|
||||||
"net"
|
"net"
|
||||||
|
|
||||||
"github.com/ThreeDotsLabs/watermill"
|
"github.com/ThreeDotsLabs/watermill"
|
||||||
"github.com/ThreeDotsLabs/watermill/message"
|
"github.com/ThreeDotsLabs/watermill/message"
|
||||||
|
"github.com/ThreeDotsLabs/watermill/message/router/middleware"
|
||||||
|
"github.com/ThreeDotsLabs/watermill/message/router/plugin"
|
||||||
|
"github.com/ThreeDotsLabs/watermill/pubsub/gochannel"
|
||||||
)
|
)
|
||||||
|
|
||||||
type TELNETServer struct {
|
func main() {
|
||||||
l net.Listener
|
l, err := net.Listen("tcp", ":5555")
|
||||||
}
|
|
||||||
|
|
||||||
func NewTELNETServer(l net.Listener) *TELNETServer {
|
|
||||||
ts := TELNETServer{
|
|
||||||
l: l,
|
|
||||||
}
|
|
||||||
|
|
||||||
return &ts
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ts *TELNETServer) Run(ctx context.Context) {
|
|
||||||
for {
|
|
||||||
conn, err := ts.l.Accept()
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
|
defer l.Close()
|
||||||
|
wmLogger := watermill.NewStdLogger(false, false)
|
||||||
|
router, err := message.NewRouter(message.RouterConfig{}, wmLogger)
|
||||||
|
|
||||||
go func(c net.Conn) {
|
router.AddPlugin(plugin.SignalsHandler)
|
||||||
s := bufio.NewScanner(c)
|
|
||||||
defer c.Close()
|
|
||||||
|
|
||||||
for s.Scan() {
|
router.AddMiddleware(
|
||||||
t := s.Text()
|
middleware.CorrelationID,
|
||||||
err := ts.Publish("raw_telnet", message.NewMessage(watermill.NewUUID(), []byte(t)))
|
middleware.Recoverer,
|
||||||
if err != nil {
|
)
|
||||||
log.Fatalln("couldn't write to raw_telnet queue")
|
|
||||||
}
|
|
||||||
log.Printf("Received user input: %q\n", t)
|
|
||||||
}
|
|
||||||
}(conn)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ts *TELNETServer) Publish(topic string, messages ...*message.Message) error {
|
pubSub := gochannel.NewGoChannel(gochannel.Config{}, wmLogger)
|
||||||
|
|
||||||
return nil
|
ctx := context.Background()
|
||||||
}
|
ts := NewTELNETServer(ctx, l, pubSub)
|
||||||
|
|
||||||
func main() {
|
for {
|
||||||
l, err := net.Listen("tcp", ":5555")
|
conn, err := ts.l.Accept()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
defer l.Close()
|
|
||||||
|
|
||||||
ts := NewTELNETServer(l)
|
go ts.Handle(conn)
|
||||||
ts.Run(context.Background())
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@ -0,0 +1,44 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"context"
|
||||||
|
"log"
|
||||||
|
"net"
|
||||||
|
|
||||||
|
"github.com/ThreeDotsLabs/watermill"
|
||||||
|
"github.com/ThreeDotsLabs/watermill/message"
|
||||||
|
)
|
||||||
|
|
||||||
|
type TELNETParser struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
type TELNETServer struct {
|
||||||
|
l net.Listener
|
||||||
|
p message.Publisher
|
||||||
|
c context.Context
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewTELNETServer(c context.Context, l net.Listener, p message.Publisher) *TELNETServer {
|
||||||
|
ts := TELNETServer{
|
||||||
|
c: c,
|
||||||
|
l: l,
|
||||||
|
p: p,
|
||||||
|
}
|
||||||
|
|
||||||
|
return &ts
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ts *TELNETServer) Handle(conn net.Conn) {
|
||||||
|
s := bufio.NewScanner(conn)
|
||||||
|
defer conn.Close()
|
||||||
|
|
||||||
|
for s.Scan() {
|
||||||
|
t := s.Text()
|
||||||
|
err := ts.p.Publish("telnet.raw", message.NewMessage(watermill.NewUUID(), []byte(t)))
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalln("couldn't write to raw_telnet queue")
|
||||||
|
}
|
||||||
|
log.Printf("Received user input: %q\n", t)
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue