common

package

v0.1.7 Latest Latest Go to latest Published: Apr 11, 2025 License: MIT Imports: 19 Imported by: 0

Details

View Source

const (
	MaxRequestBodySize = 1024 * 1024 * 50 // 50MB
)

View Source

var (
	UsingSQLite     = false
	UsingPostgreSQL = false
	UsingMySQL      = false
)

View Source

var (
	SQLitePath        = env.String("SQLITE_PATH", "aiproxy.db")
	SQLiteBusyTimeout = env.Int64("SQLITE_BUSY_TIMEOUT", 3000)
)

View Source

var (
	RDB          *redis.Client
	RedisEnabled = false
)

View Source

var ErrLimitedReaderExceeded = errors.New("limited reader exceeded")

View Source

var StartTime = time.Now().UnixMilli() // unit: millisecond

func GetRequestBody(req *http.Request) ([]byte, error)

func InitRedisClient() (err error)

InitRedisClient This function is called after init()

func LimitReader(r io.Reader, n int64) io.Reader

func NeedColor() bool

func RedisDel(key string) error

func RedisGet(key string) (string, error)

func RedisSet(key string, value string, expiration time.Duration) error

func SetEventStreamHeaders(c *gin.Context)

func TruncateByRune[T ~string](s T, length int) T

func TruncateBytesByRune(b []byte, length int) []byte

func UnmarshalBody2Node(req *http.Request) (ast.Node, error)

func UnmarshalBodyReusable(req *http.Request, v any) error

type LimitedReader struct {
	R io.Reader
	N int64
}

func (l *LimitedReader) Read(p []byte) (n int, err error)

type RequestBodyKey struct{}