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.
87 lines
1.9 KiB
87 lines
1.9 KiB
6 years ago
|
package main
|
||
|
|
|
||
|
|
var _singleTemplate = `
|
||
|
|
// 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 VALUE, err error) {
|
||
|
|
addCache := true
|
||
|
|
res, err = CACHEFUNC(c, {{.IDName}} {{.ExtraCacheArgs}})
|
||
|
|
if err != nil {
|
||
|
|
addCache = false
|
||
|
|
err = nil
|
||
|
|
}
|
||
|
|
{{if .EnableNullCache}}
|
||
|
|
defer func() {
|
||
|
|
{{if .SimpleValue}} if res == {{.NullCache}} { {{else}} if {{.CheckNullCode}} { {{end}}
|
||
|
|
res = {{.ZeroValue}}
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
{{end}}
|
||
|
|
{{if .GoValue}}
|
||
|
|
if len(res) != 0 {
|
||
|
|
{{else}}
|
||
|
|
if res != {{.ZeroValue}} {
|
||
|
|
{{end}}
|
||
|
|
prom.CacheHit.Incr("NAME")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
{{if .EnablePaging}}
|
||
|
|
var miss 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.Incr("NAME")
|
||
|
|
{{if .EnablePaging}}
|
||
|
|
var rrs [2]interface{}
|
||
|
|
rrs[0], rrs[1], e = RAWFUNC(c, {{.IDName}} {{.ExtraRawArgs}})
|
||
|
|
r = rrs
|
||
|
|
{{else}}
|
||
|
|
r, e = RAWFUNC(c, {{.IDName}} {{.ExtraRawArgs}})
|
||
|
|
{{end}}
|
||
|
|
return
|
||
|
|
})
|
||
|
|
{{if .EnablePaging}}
|
||
|
|
res = rr.([2]interface{})[0].(VALUE)
|
||
|
|
miss = rr.([2]interface{})[1].(VALUE)
|
||
|
|
{{else}}
|
||
|
|
res = rr.(VALUE)
|
||
|
|
{{end}}
|
||
|
|
{{else}}
|
||
|
|
prom.CacheMiss.Incr("NAME")
|
||
|
|
{{if .EnablePaging}}
|
||
|
|
res, miss, err = RAWFUNC(c, {{.IDName}} {{.ExtraRawArgs}})
|
||
|
|
{{else}}
|
||
|
|
res, err = RAWFUNC(c, {{.IDName}} {{.ExtraRawArgs}})
|
||
|
|
{{end}}
|
||
|
|
{{end}}
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
{{if .EnablePaging}}
|
||
|
|
{{else}}
|
||
|
|
miss := res
|
||
|
|
{{end}}
|
||
|
|
{{if .EnableNullCache}}
|
||
|
|
{{if .GoValue}}
|
||
|
|
if len(miss) == 0 {
|
||
|
|
{{else}}
|
||
|
|
if miss == {{.ZeroValue}} {
|
||
|
|
{{end}}
|
||
|
|
miss = {{.NullCache}}
|
||
|
|
}
|
||
|
|
{{end}}
|
||
|
|
if !addCache {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
{{if .Sync}}
|
||
|
|
ADDCACHEFUNC(c, {{.IDName}}, miss {{.ExtraAddCacheArgs}})
|
||
|
|
{{else}}
|
||
|
|
d.cache.Do(c, func(c context.Context) {
|
||
|
|
ADDCACHEFUNC(c, {{.IDName}}, miss {{.ExtraAddCacheArgs}})
|
||
|
|
})
|
||
|
|
{{end}}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
`
|