common

Documentation

Index

• Constants
• Variables
• func GetRequestBody(req *http.Request) ([]byte, error)
• func InitRedisClient() (err error)
• 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
  • func (l *LimitedReader) Read(p []byte) (n int, err error)
• type RequestBodyKey

Constants

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

Variables

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

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

var (
	RDB          *redis.Client
	RedisEnabled = false
)

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

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

Functions

func GetRequestBody

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

func InitRedisClient

func InitRedisClient() (err error)

InitRedisClient This function is called after init()

func LimitReader

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

func NeedColor

func NeedColor() bool

func RedisDel

func RedisDel(key string) error

func RedisGet

func RedisGet(key string) (string, error)

func RedisSet

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

func SetEventStreamHeaders

func SetEventStreamHeaders(c *gin.Context)

func TruncateByRune

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

func TruncateBytesByRune

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

func UnmarshalBody2Node

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

func UnmarshalBodyReusable

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

Types

type LimitedReader

type LimitedReader struct {
	R io.Reader
	N int64
}

func (*LimitedReader) Read

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

type RequestBodyKey

type RequestBodyKey struct{}