ghproxy/proxy/httpc.go

package proxy

import (
	"fmt"
	"ghproxy/config"
	"net/http"
	"time"

	"github.com/WJQSERVER-STUDIO/httpc"
)

var BufferSize int = 32 * 1024 // 32KB

var (
	tr         *http.Transport
	gittr      *http.Transport
	client     *httpc.Client
	gitclient  *httpc.Client
	ghcrtr     *http.Transport
	ghcrclient *httpc.Client
)

func InitReq(cfg *config.Config) error {
	initHTTPClient(cfg)
	if cfg.GitClone.Mode == "cache" {
		initGitHTTPClient(cfg)
	}
	initGhcrHTTPClient(cfg)
	err := SetGlobalRateLimit(cfg)
	if err != nil {
		return err
	}
	return nil

}

func initHTTPClient(cfg *config.Config) {
	var proTolcols = new(http.Protocols)
	proTolcols.SetHTTP1(true)
	proTolcols.SetHTTP2(true)
	proTolcols.SetUnencryptedHTTP2(true)
	if cfg.Httpc.Mode == "auto" || cfg.Httpc.Mode == "" {

		tr = &http.Transport{
			IdleConnTimeout: 30 * time.Second,
			WriteBufferSize: 32 * 1024, // 32KB
			ReadBufferSize:  32 * 1024, // 32KB
			Protocols:       proTolcols,
		}
	} else if cfg.Httpc.Mode == "advanced" {
		tr = &http.Transport{
			MaxIdleConns:        cfg.Httpc.MaxIdleConns,
			MaxConnsPerHost:     cfg.Httpc.MaxConnsPerHost,
			MaxIdleConnsPerHost: cfg.Httpc.MaxIdleConnsPerHost,
			WriteBufferSize:     32 * 1024, // 32KB
			ReadBufferSize:      32 * 1024, // 32KB
			Protocols:           proTolcols,
		}
	} else {
		panic("unknown httpc mode: " + cfg.Httpc.Mode)
	}
	if cfg.Outbound.Enabled {
		initTransport(cfg, tr)
	}
	if cfg.Server.Debug {
		client = httpc.New(
			httpc.WithTransport(tr),
			httpc.WithDumpLog(),
		)
	} else {
		client = httpc.New(
			httpc.WithTransport(tr),
		)
	}

}

func initGitHTTPClient(cfg *config.Config) {

	if cfg.Httpc.Mode == "auto" || cfg.Httpc.Mode == "" {
		gittr = &http.Transport{
			IdleConnTimeout: 30 * time.Second,
			WriteBufferSize: 32 * 1024, // 32KB
			ReadBufferSize:  32 * 1024, // 32KB
		}
	} else if cfg.Httpc.Mode == "advanced" {
		gittr = &http.Transport{
			MaxIdleConns:        cfg.Httpc.MaxIdleConns,
			MaxConnsPerHost:     cfg.Httpc.MaxConnsPerHost,
			MaxIdleConnsPerHost: cfg.Httpc.MaxIdleConnsPerHost,
			WriteBufferSize:     32 * 1024, // 32KB
			ReadBufferSize:      32 * 1024, // 32KB
		}
	} else {
		panic("unknown httpc mode: " + cfg.Httpc.Mode)
	}
	if cfg.Outbound.Enabled {
		initTransport(cfg, gittr)
	}
	if cfg.Server.Debug && cfg.GitClone.ForceH2C {
		gitclient = httpc.New(
			httpc.WithTransport(gittr),
			httpc.WithDumpLog(),
			httpc.WithProtocols(httpc.ProtocolsConfig{
				ForceH2C: true,
			}),
		)
	} else if !cfg.Server.Debug && cfg.GitClone.ForceH2C {
		gitclient = httpc.New(
			httpc.WithTransport(gittr),
			httpc.WithProtocols(httpc.ProtocolsConfig{
				ForceH2C: true,
			}),
		)
	} else if cfg.Server.Debug && !cfg.GitClone.ForceH2C {
		gitclient = httpc.New(
			httpc.WithTransport(gittr),
			httpc.WithDumpLog(),
			httpc.WithProtocols(httpc.ProtocolsConfig{
				Http1:           true,
				Http2:           true,
				Http2_Cleartext: true,
			}),
		)
	} else {
		gitclient = httpc.New(
			httpc.WithTransport(gittr),
			httpc.WithProtocols(httpc.ProtocolsConfig{
				Http1:           true,
				Http2:           true,
				Http2_Cleartext: true,
			}),
		)
	}
}

func initGhcrHTTPClient(cfg *config.Config) {
	var proTolcols = new(http.Protocols)
	proTolcols.SetHTTP1(true)
	proTolcols.SetHTTP2(true)
	if cfg.Httpc.Mode == "auto" || cfg.Httpc.Mode == "" {

		ghcrtr = &http.Transport{
			IdleConnTimeout: 30 * time.Second,
			WriteBufferSize: 32 * 1024, // 32KB
			ReadBufferSize:  32 * 1024, // 32KB
			Protocols:       proTolcols,
		}
	} else if cfg.Httpc.Mode == "advanced" {
		ghcrtr = &http.Transport{
			MaxIdleConns:        cfg.Httpc.MaxIdleConns,
			MaxConnsPerHost:     cfg.Httpc.MaxConnsPerHost,
			MaxIdleConnsPerHost: cfg.Httpc.MaxIdleConnsPerHost,
			WriteBufferSize:     32 * 1024, // 32KB
			ReadBufferSize:      32 * 1024, // 32KB
			Protocols:           proTolcols,
		}
	} else {
		panic(fmt.Sprintf("unknown httpc mode: %s", cfg.Httpc.Mode))
	}
	if cfg.Outbound.Enabled {
		initTransport(cfg, ghcrtr)
	}
	if cfg.Server.Debug {
		ghcrclient = httpc.New(
			httpc.WithTransport(ghcrtr),
			httpc.WithDumpLog(),
		)
	} else {
		ghcrclient = httpc.New(
			httpc.WithTransport(ghcrtr),
		)
	}
}