microservice/socket_client.go

package microservice

import (
	"context"
	"fmt"
	"net"
	"time"
)

func NewSocketClient(host string, port uint16, bufSize int, ctx context.Context) *SocketClient {
	cctx, cancel := context.WithCancel(ctx)
	socket := &SocketClient{
		host:    host,
		port:    port,
		bufSize: bufSize,
		ctx:     cctx,
		cancel:  cancel,
	}
	go socket.connect()
	return socket
}

type SocketClient struct {
	host    string
	port    uint16
	bufSize int
	ctx     context.Context
	cancel  context.CancelFunc
	socket  *Socket
}

func (s *SocketClient) start() {
	//log.Println("Start(X)...")
	for {
		select {
		case <-s.ctx.Done():
			//log.Println("XDONE--------------------------")
			return
		case <-s.socket.Context().Done():
			//log.Println("DONE>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>")
			s.connect()
			return
		}
	}
}

func (s *SocketClient) connect() {
	//log.Println("CONNECT(X)..............")
	for {
		select {
		case <-s.ctx.Done():
			s.socket.Close()
			return
		default:
			//log.Println("CONNECT...")
			if conn, err := net.Dial("tcp", fmt.Sprintf("%v:%v", s.host, s.port)); err == nil {
				s.socket = NewSocket(conn, s.bufSize, s.ctx)
				s.start()
				return
			}
			time.Sleep(time.Second)
		}
	}
}

func (s *SocketClient) Close() {
	//log.Println("CLIENT-CLOSE...")
	s.socket.Close()
	s.cancel()
}