通用包
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
utils/sls/producer.go

124 lines
2.7 KiB

package sls
import (
"gitea.drugeyes.vip/pharnexbase/utils/enum"
sls "github.com/aliyun/aliyun-log-go-sdk"
"github.com/aliyun/aliyun-log-go-sdk/producer"
)
type options struct {
accessKey string
accessSecret string
endpoint string
project string
logstore string
}
func defaultOptions() *options {
return &options{
project: "bcpm-log",
logstore: "request-develop",
}
}
// WithEndpoint 指定端口
func WithEndpoint(endpoint string) Option {
return func(alc *options) {
alc.endpoint = endpoint
}
}
// WithProject 指定发送普通日志的project
func WithProject(project string) Option {
return func(alc *options) {
alc.project = project
}
}
// WithLogstore 指定发送普通日志的logstore
func WithLogstore(logstore string) Option {
return func(alc *options) {
alc.logstore = logstore
}
}
// WithAccessKey 设置访问access id
func WithAccessKey(ak string) Option {
return func(alc *options) {
alc.accessKey = ak
}
}
// WithAccessSecret 设置访问access secret
func WithAccessSecret(as string) Option {
return func(alc *options) {
alc.accessSecret = as
}
}
type Option func(alc *options)
type slsProducer struct {
pr *producer.Producer
opts *options
}
// NewSLSProducer 创建一个高性能的sls制造者
func NewSLSProducer(opts ...Option) *slsProducer {
opt := defaultOptions()
for _, o := range opts {
o(opt)
}
producerConfig := producer.GetDefaultProducerConfig()
producerConfig.Endpoint = opt.endpoint
producerConfig.AccessKeyID = opt.accessKey
producerConfig.AccessKeySecret = opt.accessSecret
producerInstance := producer.InitProducer(producerConfig)
producerInstance.Start()
return &slsProducer{
pr: producerInstance,
opts: opt,
}
}
// Close 程序结束时,必须调用该方法,不然可能导致部分日志丢失
func (s *slsProducer) Close() error {
return s.pr.Close(5000)
}
// SendLog 发送普通日志
func (s *slsProducer) SendLog(topic, source string, log *sls.Log) error {
err := s.pr.SendLog(s.opts.project, s.opts.logstore, topic, source, log)
if err != nil {
return err
}
return nil
}
// SendRequestLog 发送特定的请求日志,使用该方法时,不需要指定project和logstore
func (s *slsProducer) SendRequestLog(env enum.Env, req *Request) error {
log := GenerateLog(req)
err := s.pr.SendLog("bcpm-log", getLogStore(env), "request-go", req.RemoteAddr, log)
if err != nil {
return err
}
return nil
}
func getLogStore(env enum.Env) string {
switch env {
case enum.Env_local, enum.Env_develop:
return "request-develop"
case enum.Env_preview:
return "request-preview"
case enum.Env_production:
return "request"
default:
return "request-develop"
}
}