aboutsummaryrefslogtreecommitdiff
path: root/middleware/singleflight/singleflight.go
diff options
context:
space:
mode:
authorGravatar Miek Gieben <miek@miek.nl> 2016-08-16 09:24:37 -0700
committerGravatar GitHub <noreply@github.com> 2016-08-16 09:24:37 -0700
commit21ddfe8290bb9f863f3f9175737d80d5a2782b39 (patch)
treed913285ba5b39a880a0a2d3e1f1ff68788f56c4e /middleware/singleflight/singleflight.go
parent9aaeef6376e677395f31ab0432c90cbe415c12c1 (diff)
downloadcoredns-21ddfe8290bb9f863f3f9175737d80d5a2782b39.tar.gz
coredns-21ddfe8290bb9f863f3f9175737d80d5a2782b39.tar.zst
coredns-21ddfe8290bb9f863f3f9175737d80d5a2782b39.zip
Move singleflight out of middleware/ (#217)
It is not middleware, so move it up one level. It was also use from core/ showing that indeed its use is not limited to middlewares.
Diffstat (limited to 'middleware/singleflight/singleflight.go')
-rw-r--r--middleware/singleflight/singleflight.go64
1 files changed, 0 insertions, 64 deletions
diff --git a/middleware/singleflight/singleflight.go b/middleware/singleflight/singleflight.go
deleted file mode 100644
index ff2c2ee4f..000000000
--- a/middleware/singleflight/singleflight.go
+++ /dev/null
@@ -1,64 +0,0 @@
-/*
-Copyright 2012 Google Inc.
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
- http://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-*/
-
-// Package singleflight provides a duplicate function call suppression
-// mechanism.
-package singleflight
-
-import "sync"
-
-// call is an in-flight or completed Do call
-type call struct {
- wg sync.WaitGroup
- val interface{}
- err error
-}
-
-// Group represents a class of work and forms a namespace in which
-// units of work can be executed with duplicate suppression.
-type Group struct {
- mu sync.Mutex // protects m
- m map[string]*call // lazily initialized
-}
-
-// Do executes and returns the results of the given function, making
-// sure that only one execution is in-flight for a given key at a
-// time. If a duplicate comes in, the duplicate caller waits for the
-// original to complete and receives the same results.
-func (g *Group) Do(key string, fn func() (interface{}, error)) (interface{}, error) {
- g.mu.Lock()
- if g.m == nil {
- g.m = make(map[string]*call)
- }
- if c, ok := g.m[key]; ok {
- g.mu.Unlock()
- c.wg.Wait()
- return c.val, c.err
- }
- c := new(call)
- c.wg.Add(1)
- g.m[key] = c
- g.mu.Unlock()
-
- c.val, c.err = fn()
- c.wg.Done()
-
- g.mu.Lock()
- delete(g.m, key)
- g.mu.Unlock()
-
- return c.val, c.err
-}