common

Documentation

Index

• Variables
• func GetRequestBody(req *http.Request) ([]byte, error)
• func Init()
• func InitRedisClient() (err error)
• 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 UnmarshalBodyReusable(req *http.Request, v any) error
• type CustomEvent
  • func (r CustomEvent) Render(w http.ResponseWriter) error
  • func (r CustomEvent) WriteContentType(w http.ResponseWriter)
• type InMemoryRateLimiter
  • func (l *InMemoryRateLimiter) Init(expirationDuration time.Duration)
  • func (l *InMemoryRateLimiter) Request(key string, maxRequestNum int, duration time.Duration) bool
• type RateLimitWindow
• type RequestBodyKey

Variables

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

var (
	SQLitePath        = "aiproxy.db"
	SQLiteBusyTimeout = env.Int("SQLITE_BUSY_TIMEOUT", 3000)
)

var (
	Port   = flag.Int("port", 3000, "the listening port")
	LogDir = flag.String("log-dir", "", "specify the log directory")
)

var (
	RDB          *redis.Client
	RedisEnabled = false
)

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

Functions

func GetRequestBody

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

func Init

func Init()

func InitRedisClient

func InitRedisClient() (err error)

InitRedisClient This function is called after init()

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 UnmarshalBodyReusable

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

Types

type CustomEvent

type CustomEvent struct {
	Data  string
	Event string
	ID    string
	Retry uint
}

func (CustomEvent) Render

func (r CustomEvent) Render(w http.ResponseWriter) error

func (CustomEvent) WriteContentType

func (r CustomEvent) WriteContentType(w http.ResponseWriter)

type InMemoryRateLimiter

type InMemoryRateLimiter struct {
	// contains filtered or unexported fields
}

func (*InMemoryRateLimiter) Init

func (l *InMemoryRateLimiter) Init(expirationDuration time.Duration)

func (*InMemoryRateLimiter) Request

func (l *InMemoryRateLimiter) Request(key string, maxRequestNum int, duration time.Duration) bool

Request parameter duration's unit is seconds

type RateLimitWindow

type RateLimitWindow struct {
	// contains filtered or unexported fields
}

type RequestBodyKey

type RequestBodyKey struct{}