mirror of
https://github.com/dcarrillo/whatismyip.git
synced 2024-11-01 01:31:14 +00:00
47 lines
940 B
Go
47 lines
940 B
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/dcarrillo/whatismyip/internal/setting"
|
|
)
|
|
|
|
type TCP struct {
|
|
server *http.Server
|
|
handler *http.Handler
|
|
ctx context.Context
|
|
}
|
|
|
|
func NewTCPServer(ctx context.Context, handler *http.Handler) *TCP {
|
|
return &TCP{
|
|
handler: handler,
|
|
ctx: ctx,
|
|
}
|
|
}
|
|
|
|
func (t *TCP) Start() {
|
|
t.server = &http.Server{
|
|
Addr: setting.App.BindAddress,
|
|
Handler: *t.handler,
|
|
ReadTimeout: setting.App.Server.ReadTimeout,
|
|
WriteTimeout: setting.App.Server.WriteTimeout,
|
|
}
|
|
|
|
log.Printf("Starting TCP server listening on %s", setting.App.BindAddress)
|
|
go func() {
|
|
if err := t.server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
|
|
log.Fatal(err)
|
|
}
|
|
}()
|
|
}
|
|
|
|
func (t *TCP) Stop() {
|
|
log.Printf("Stopping TCP server...")
|
|
if err := t.server.Shutdown(t.ctx); err != nil {
|
|
log.Printf("TCP server forced to shutdown: %s", err)
|
|
}
|
|
}
|