代码优化

This commit is contained in:
1340691923@qq.com 2022-03-02 19:01:19 +08:00
parent ffbcd7e790
commit 783c289abe
4 changed files with 11 additions and 53 deletions

View File

@ -55,7 +55,6 @@ func MysqlConsumer() {
}
}
func AddMetaEvent(kafkaData model.KafkaData) (err error) {
if kafkaData.ReportType == model.EventReportType {
redisConn := db.RedisPool.Get()
@ -81,8 +80,6 @@ func AddMetaEvent(kafkaData model.KafkaData) (err error) {
return nil
}
func AddTableColumn(kafkaData model.KafkaData, failFunc func(data consumer_data.ReportAcceptStatusData), tableName string, ReqDataObject *parser.FastjsonMetric) (err error) {
dims, err := sinker.GetDims(model.GlobConfig.Comm.ClickHouse.DbName, tableName, nil, db.ClickHouseSqlx)

View File

@ -109,7 +109,7 @@ func main() {
reportData2CKSarama := realTimeDataSarama.Clone()
go action.MysqlConsumer()
var json = jsoniter.ConfigCompatibleWithStandardLibrary
pp, err := parser.NewParserPool("fastjson", nil, "", "")
pp, err := parser.NewParserPool()
if err != nil {
panic(err)
}
@ -274,7 +274,6 @@ func main() {
Status: consumer_data.SuccessStatus,
}); err != nil {
logs.Logger.Error("reportAcceptStatus Add SuccessStatus err", zap.Error(err))
return
}
//添加数据到ck用于后台统计

View File

@ -28,7 +28,7 @@ type ReportController struct {
var pp *parser.Pool
func init(){
pp, _ = parser.NewParserPool("fastjson", nil, "", "")
pp, _ = parser.NewParserPool()
}
//上报接口

View File

@ -18,10 +18,6 @@ var (
ErrParseDateTime = errors.Errorf("value doesn't contain DateTime")
)
var typeFormatMap = map[string]string{
}
// Parse is the Parser interface
type Parser interface {
Parse(bs []byte) (metric *FastjsonMetric, err error)
@ -29,34 +25,18 @@ type Parser interface {
// Pool may be used for pooling Parsers for similarly typed JSONs.
type Pool struct {
name string
csvFormat map[string]int
delimiter string
timeZone *time.Location
knownLayouts sync.Map
pool sync.Pool
}
// NewParserPool creates a parser pool
func NewParserPool(name string, csvFormat []string, delimiter string, timezone string) (pp *Pool, err error) {
var tz *time.Location
if timezone == "" {
tz = time.Local
} else if tz, err = time.LoadLocation(timezone); err != nil {
err = errors.Wrapf(err, "")
return
}
func NewParserPool() (pp *Pool, err error) {
var tz = time.Local
pp = &Pool{
name: name,
delimiter: delimiter,
timeZone: tz,
}
if csvFormat != nil {
pp.csvFormat = make(map[string]int)
for i, title := range csvFormat {
pp.csvFormat[title] = i
}
}
return
}
@ -73,12 +53,7 @@ func ParseKafkaData(pool *Pool, data []byte) (metric *FastjsonMetric, err error)
func (pp *Pool) Get() Parser {
v := pp.pool.Get()
if v == nil {
switch pp.name {
case "fastjson":
return &FastjsonParser{pp: pp}
default:
return &FastjsonParser{pp: pp}
}
}
return v.(Parser)
}
@ -92,28 +67,15 @@ func (pp *Pool) Put(p Parser) {
}
func (pp *Pool) ParseDateTime(key string, val string) (t time.Time, err error) {
var layout string
var lay interface{}
var ok bool
var t2 time.Time
if val == "" {
err = ErrParseDateTime
return
}
if lay, ok = pp.knownLayouts.Load(key); !ok {
t2, layout = parseInLocation(val, pp.timeZone)
if layout == "" {
err = ErrParseDateTime
return
}
t = t2
return
}
if layout, ok = lay.(string); !ok {
err = ErrParseDateTime
return
}
if t2, err = time.ParseInLocation(layout, val, pp.timeZone); err != nil {
if t2, err = time.ParseInLocation("2006-01-02 15:04:05", val, pp.timeZone); err != nil {
err = ErrParseDateTime
return
}