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.
100 lines
2.3 KiB
100 lines
2.3 KiB
6 years ago
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _noneGetTemplate = `
|
||
|
|
// NAME {{or .Comment "get data from mc"}}
|
||
|
|
func (d *{{.StructName}}) NAME(c context.Context) (res VALUE, err error) {
|
||
|
|
key := {{.KeyMethod}}()
|
||
|
|
{{if .GetSimpleValue}}
|
||
|
|
var v string
|
||
|
|
err = d.mc.Get(c, key).Scan(&v)
|
||
|
|
{{else}}
|
||
|
|
{{if .GetDirectValue}}
|
||
|
|
err = d.mc.Get(c, key).Scan(&res)
|
||
|
|
{{else}}
|
||
|
|
{{if .PointType}}
|
||
|
|
res = &{{.OriginValueType}}{}
|
||
|
|
if err = d.mc.Get(c, key).Scan(res); err != nil {
|
||
|
|
res = nil
|
||
|
|
if err == memcache.ErrNotFound {
|
||
|
|
err = nil
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
{{else}}
|
||
|
|
err = d.mc.Get(c, key).Scan(&res)
|
||
|
|
{{end}}
|
||
|
|
{{end}}
|
||
|
|
{{end}}
|
||
|
|
if err != nil {
|
||
|
|
{{if .PointType}}
|
||
|
|
{{else}}
|
||
|
|
if err == memcache.ErrNotFound {
|
||
|
|
err = nil
|
||
|
|
return
|
||
|
|
}
|
||
|
|
{{end}}
|
||
|
|
log.Errorv(c, log.KV("NAME", fmt.Sprintf("%+v", err)), log.KV("key", key))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
{{if .GetSimpleValue}}
|
||
|
|
r, err := {{.ConvertBytes2Value}}
|
||
|
|
if err != nil {
|
||
|
|
log.Errorv(c, log.KV("NAME", fmt.Sprintf("%+v", err)), log.KV("key", key))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
res = {{.ValueType}}(r)
|
||
|
|
{{end}}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
`
|
||
|
|
|
||
|
|
var _noneSetTemplate = `
|
||
|
|
// NAME {{or .Comment "Set data to mc"}}
|
||
|
|
func (d *{{.StructName}}) NAME(c context.Context, val VALUE) (err error) {
|
||
|
|
{{if .PointType}}
|
||
|
|
if val == nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
{{end}}
|
||
|
|
{{if .LenType}}
|
||
|
|
if len(val) == 0 {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
{{end}}
|
||
|
|
key := {{.KeyMethod}}()
|
||
|
|
{{if .SimpleValue}}
|
||
|
|
bs := {{.ConvertValue2Bytes}}
|
||
|
|
item := &memcache.Item{Key: key, Value: bs, Expiration: {{.ExpireCode}}, Flags: {{.Encode}}}
|
||
|
|
{{else}}
|
||
|
|
item := &memcache.Item{Key: key, Object: val, Expiration: {{.ExpireCode}}, Flags: {{.Encode}}}
|
||
|
|
{{end}}
|
||
|
|
if err = d.mc.Set(c, item); err != nil {
|
||
|
|
log.Errorv(c, log.KV("NAME", fmt.Sprintf("%+v", err)), log.KV("key", key))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
`
|
||
|
|
var _noneAddTemplate = strings.Replace(_noneSetTemplate, "Set", "Add", -1)
|
||
|
|
var _noneReplaceTemplate = strings.Replace(_noneSetTemplate, "Set", "Replace", -1)
|
||
|
|
|
||
|
|
var _noneDelTemplate = `
|
||
|
|
// NAME {{or .Comment "delete data from mc"}}
|
||
|
|
func (d *{{.StructName}}) NAME(c context.Context) (err error) {
|
||
|
|
key := {{.KeyMethod}}()
|
||
|
|
if err = d.mc.Delete(c, key); err != nil {
|
||
|
|
if err == memcache.ErrNotFound {
|
||
|
|
err = nil
|
||
|
|
return
|
||
|
|
}
|
||
|
|
log.Errorv(c, log.KV("NAME", fmt.Sprintf("%+v", err)), log.KV("key", key))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
`
|