package app

import (
	"common/log"
	"context"
	"errors"
	"fmt"
	"gateway/config"
	"gateway/net/http_gateway"
	"github.com/gin-gonic/gin"
	"net/http"
	"sync"
)

// ModuleWebServer Web服务模块
type ModuleWebServer struct {
	wg     *sync.WaitGroup
	server *http.Server
	router *gin.Engine
}

func (m *ModuleWebServer) Init() error {
	m.wg = &sync.WaitGroup{}
	m.router = http_gateway.InitRouter()
	return nil
}

func (m *ModuleWebServer) Start() error {
	m.wg.Add(1)
	go func() {
		defer m.wg.Done()
		m.server = &http.Server{
			Addr:    fmt.Sprintf("%v:%v", config.Get().Serve.Http.Address, config.Get().Serve.Http.Port),
			Handler: m.router,
		}
		if err := m.server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
			log.Errorf("http server failed: %v", err.Error())
		}
		log.Infof("http server stop.")
	}()
	return nil
}

func (m *ModuleWebServer) Stop() error {
	if err := m.server.Shutdown(context.Background()); err != nil {
		log.Errorf("stop http server failed: %v", err)
	}
	m.wg.Wait()
	return nil
}