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 } `