wartank/pkg/components/arena_net/http_worker/http_worker.go

// package http_worker -- веб-воркер бота
package http_worker

import (
	"context"
	"fmt"
	"io"
	"net/http"
	"strings"
	"sync"
	"time"
	"wartank/pkg/components/arena_net/http_worker/net_stat"
	"wartank/pkg/components/kernel/logger"
	"wartank/pkg/types"
)

var (
	блок sync.Mutex
)

// ХттпВоркер -- объект сетевого исполнителя бота
type ХттпВоркер struct {
	botNet     types.ИБотСеть
	conn       *http.Client
	статистика *net_stat.СетьСтата
	лог        types.ИЛоггер
}

// НовХттпВоркер -- возвращает веб-воркера бота
func НовХттпВоркер(botNet types.ИБотСеть, conn *http.Client) *ХттпВоркер {
	лог := logger.НовЛоггер("ХттпВоркер")
	лог.Инфо("НовХттпВоркер()\n")
	сам := &ХттпВоркер{
		botNet:     botNet,
		conn:       conn,
		статистика: net_stat.НовСетьСтата(botNet),
		лог:        лог,
	}
	_ = types.ИХттпВоркер(сам)
	return сам
}

// Получ -- потокобезопасно возвращает список строк по ссылке
func (сам *ХттпВоркер) Получ(strLink string) ([]string, error) {
	блок.Lock()
	defer блок.Unlock()
	// if strLink == "https://wartank.ru/production/Mine" {
	сам.лог.Отладка("Получ(): link=%v\n", strLink)
	// }
	ctxCancel, fnCancel := context.WithTimeout(сам.botNet.Контекст(), time.Second*10)
	defer fnCancel()
	запрос, ош := http.NewRequestWithContext(ctxCancel, "GET", strLink, nil)
	if ош != nil {
		сам.статистика.IncErr()
		return nil, fmt.Errorf("ХттпВоркер.Получ(): при создании запроса, err=\n\t%w", ош)
	}
	запрос.Header.Set("User-Agent", "Mozilla Firefox 106.1")
	ответ, ош := сам.conn.Do(запрос)
	if ош != nil {
		сам.статистика.IncErr()
		return nil, fmt.Errorf("ХттпВоркер.Получ(): при выполнении GET-запроса, err=\n\t%w", ош)
	}
	defer ответ.Body.Close()

	if ответ.StatusCode != http.StatusOK {
		сам.статистика.IncErr()
		return nil, fmt.Errorf("ХттпВоркер.Получ(): code=%v, status=%v", ответ.StatusCode, ответ.Status)
	}
	binData, ош := io.ReadAll(ответ.Body)
	if ош != nil {
		сам.статистика.IncErr()
		return nil, fmt.Errorf("ХттпВоркер.Получ(): при чтении тела ответа, err=\n\t%w", ош)
	}
	if len(binData) == 0 {
		сам.статистика.IncErr()
		return nil, fmt.Errorf("ХттпВоркер.Получ(): пустое тело ответа, err=\n\t%w", ош)
	}
	lenData := len(binData) + len(strLink)
	сам.статистика.AddByte(lenData)

	lstString := strings.Split(string(binData), "\n")
	if len(lstString) == 0 {
		сам.статистика.IncErr()
		return nil, fmt.Errorf("ХттпВоркер.Получ(): lstString is empty")
	}
	return lstString, nil
}