84 lines
1.5 KiB
Go
84 lines
1.5 KiB
Go
package tcpclient
|
|
|
|
import (
|
|
"bufio"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
)
|
|
|
|
type ClientConfiguration struct {
|
|
Status uint32
|
|
ErrorResolver func(chan error)
|
|
}
|
|
|
|
func GetDefaultConfig() *ClientConfiguration {
|
|
return &ClientConfiguration{
|
|
ErrorResolver: func(c chan error) {
|
|
for err := range c {
|
|
fmt.Printf("DefConfig:Error: %v\n", err)
|
|
}
|
|
},
|
|
}
|
|
}
|
|
|
|
type ErrorResolverFunc func(errors chan error)
|
|
|
|
type Client struct {
|
|
addr string
|
|
Status uint32
|
|
exit chan bool
|
|
Server net.Conn
|
|
IsConnected bool
|
|
//
|
|
Messages chan []byte
|
|
ErrorsChannel chan error
|
|
ErrorResolver ErrorResolverFunc
|
|
}
|
|
|
|
func CreateClient(addr string, options ...ClientOption) *Client {
|
|
conf := GetDefaultConfig()
|
|
|
|
for _, opt := range options {
|
|
opt(conf)
|
|
}
|
|
|
|
return &Client{
|
|
addr: addr,
|
|
Messages: make(chan []byte, 16),
|
|
ErrorResolver: conf.ErrorResolver,
|
|
ErrorsChannel: make(chan error, 8),
|
|
exit: make(chan bool, 1),
|
|
}
|
|
}
|
|
|
|
func (c *Client) StartClient() error {
|
|
server, err := net.Dial("tcp", c.addr)
|
|
if err != nil {
|
|
return fmt.Errorf("net.Dial: %w", err)
|
|
}
|
|
serverReader := bufio.NewReader(server)
|
|
c.IsConnected = true
|
|
c.Server = server
|
|
loop:
|
|
for {
|
|
select {
|
|
case <-c.exit:
|
|
break loop
|
|
default:
|
|
msg, err := serverReader.ReadString('\n')
|
|
if err != nil {
|
|
if errors.Is(err, io.EOF) {
|
|
c.exit <- true
|
|
break
|
|
}
|
|
c.ErrorsChannel <- fmt.Errorf("serverReader.ReadString: %w", err)
|
|
}
|
|
fmt.Printf("Server send us a message: %s", msg)
|
|
c.Messages <- []byte(msg)
|
|
}
|
|
}
|
|
return nil
|
|
}
|