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.
126 lines
2.8 KiB
126 lines
2.8 KiB
package main
|
|
|
|
var _multiTemplate = `
|
|
// NAME {{or .Comment "get data from cache if miss will call source method, then add to cache."}}
|
|
func (d *Dao) NAME(c context.Context, {{.IDName}} []KEY{{.ExtraArgsType}}) (res map[KEY]VALUE, err error) {
|
|
if len({{.IDName}}) == 0 {
|
|
return
|
|
}
|
|
addCache := true
|
|
if res, err = CACHEFUNC(c, {{.IDName}} {{.ExtraCacheArgs}});err != nil {
|
|
addCache = false
|
|
res = nil
|
|
err = nil
|
|
}
|
|
var miss []KEY
|
|
for _, key := range {{.IDName}} {
|
|
{{if .GoValue}}
|
|
if (res == nil) || (len(res[key]) == 0) {
|
|
{{else}}
|
|
{{if .NumberValue}}
|
|
if _, ok := res[key]; !ok {
|
|
{{else}}
|
|
if (res == nil) || (res[key] == {{.ZeroValue}}) {
|
|
{{end}}
|
|
{{end}}
|
|
miss = append(miss, key)
|
|
}
|
|
}
|
|
prom.CacheHit.Add("NAME", int64(len({{.IDName}}) - len(miss)))
|
|
{{if .EnableNullCache}}
|
|
for k, v := range res {
|
|
{{if .SimpleValue}} if v == {{.NullCache}} { {{else}} if {{.CheckNullCode}} { {{end}}
|
|
delete(res, k)
|
|
}
|
|
}
|
|
{{end}}
|
|
missLen := len(miss)
|
|
if missLen == 0 {
|
|
return
|
|
}
|
|
{{if .EnableBatch}}
|
|
missData := make(map[KEY]VALUE, missLen)
|
|
{{else}}
|
|
var missData map[KEY]VALUE
|
|
{{end}}
|
|
{{if .EnableSingleFlight}}
|
|
var rr interface{}
|
|
sf := d.cacheSFNAME({{.IDName}} {{.ExtraArgs}})
|
|
rr, err, _ = cacheSingleFlights[SFNUM].Do(sf, func() (r interface{}, e error) {
|
|
prom.CacheMiss.Add("NAME", int64(len(miss)))
|
|
r, e = RAWFUNC(c, miss {{.ExtraRawArgs}})
|
|
return
|
|
})
|
|
missData = rr.(map[KEY]VALUE)
|
|
{{else}}
|
|
{{if .EnableBatch}}
|
|
prom.CacheMiss.Add("NAME", int64(missLen))
|
|
var mutex sync.Mutex
|
|
{{if .BatchErrBreak}}
|
|
group := errgroup.WithCancel(c)
|
|
{{else}}
|
|
group := errgroup.WithContext(c)
|
|
{{end}}
|
|
if missLen > MAXGROUP {
|
|
group.GOMAXPROCS(MAXGROUP)
|
|
}
|
|
var run = func(ms []KEY) {
|
|
group.Go(func(ctx context.Context) (err error) {
|
|
data, err := RAWFUNC(ctx, ms {{.ExtraRawArgs}})
|
|
mutex.Lock()
|
|
for k, v := range data {
|
|
missData[k] = v
|
|
}
|
|
mutex.Unlock()
|
|
return
|
|
})
|
|
}
|
|
var (
|
|
i int
|
|
n = missLen/GROUPSIZE
|
|
)
|
|
for i=0; i< n; i++{
|
|
run(miss[i*GROUPSIZE:(i+1)*GROUPSIZE])
|
|
}
|
|
if len(miss[i*GROUPSIZE:]) > 0 {
|
|
run(miss[i*GROUPSIZE:])
|
|
}
|
|
err = group.Wait()
|
|
{{else}}
|
|
prom.CacheMiss.Add("NAME", int64(len(miss)))
|
|
missData, err = RAWFUNC(c, miss {{.ExtraRawArgs}})
|
|
{{end}}
|
|
{{end}}
|
|
if res == nil {
|
|
res = make(map[KEY]VALUE, len({{.IDName}}))
|
|
}
|
|
for k, v := range missData {
|
|
res[k] = v
|
|
}
|
|
if err != nil {
|
|
return
|
|
}
|
|
{{if .EnableNullCache}}
|
|
for _, key := range miss {
|
|
{{if .GoValue}}
|
|
if len(res[key]) == 0 {
|
|
{{else}}
|
|
if res[key] == {{.ZeroValue}} {
|
|
{{end}}
|
|
missData[key] = {{.NullCache}}
|
|
}
|
|
}
|
|
{{end}}
|
|
if !addCache {
|
|
return
|
|
}
|
|
{{if .Sync}}
|
|
ADDCACHEFUNC(c, missData {{.ExtraAddCacheArgs}})
|
|
{{else}}
|
|
d.cache.Do(c, func(c context.Context) {
|
|
ADDCACHEFUNC(c, missData {{.ExtraAddCacheArgs}})
|
|
})
|
|
{{end}}
|
|
return
|
|
}
|
|
`
|
|
|