Skip to content

Commit

Permalink
Merge pull request #28 from coinbase/michael/linting
Browse files Browse the repository at this point in the history
Fix linting + salus
  • Loading branch information
mdehoog authored Apr 26, 2023
2 parents 4867b70 + d7f9c7d commit 20e5a31
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 22 deletions.
41 changes: 22 additions & 19 deletions proxy/proxy.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,10 +17,11 @@ import (
"github.com/coinbase/memcachedbetween/listener"
"github.com/coinbase/memcachedbetween/pool"
"github.com/coinbase/mongobetween/util"
"github.com/mediocregopher/radix/v3"

"github.com/coinbase/redisbetween/config"
"github.com/coinbase/redisbetween/handlers"
"github.com/coinbase/redisbetween/redis"
"github.com/mediocregopher/radix/v3"

"github.com/DataDog/datadog-go/statsd"
"go.uber.org/zap"
Expand Down Expand Up @@ -154,9 +155,8 @@ func (p *Proxy) run() error {
ls, err := p.createListener(p.localConfigHost, p.upstreamConfigHost)
if err != nil {
return err
} else {
p.log.Info("Created Listener", zap.String("localHost", p.localConfigHost), zap.String("upstreamHost", p.upstreamConfigHost))
}
p.log.Info("Created Listener", zap.String("localHost", p.localConfigHost), zap.String("upstreamHost", p.upstreamConfigHost))
defer func() {
p.listenerWg.Wait()
}()
Expand Down Expand Up @@ -397,17 +397,6 @@ func (p *Proxy) healthCheckSingleConnection(key string, wg *sync.WaitGroup) {
}
}

// Safely grab an entry for a given key from the listeners map
func (p *Proxy) getListener(key string) *listener.Listener {
p.listenerLock.Lock()
defer p.listenerLock.Unlock()
ls, ok := p.listeners[key]
if ok {
return ls
}
return nil
}

// Safely delete a listener from the map
func (p *Proxy) removeListener(key string) {
p.listenerLock.Lock()
Expand All @@ -425,7 +414,7 @@ func (p *Proxy) getListenerKeys() []string {
p.listenerLock.Lock()
defer p.listenerLock.Unlock()
var keys []string
for key, _ := range p.listeners {
for key := range p.listeners {
keys = append(keys, key)
}
return keys
Expand Down Expand Up @@ -500,7 +489,7 @@ func compareNewNodesWithExisting(newNodes []string, existingNodes []string) (nod
}
nodesToRemove = append(nodesToRemove, node)
}
for node, _ := range newNodesMap {
for node := range newNodesMap {
nodesToAdd = append(nodesToAdd, node)
}
return
Expand All @@ -516,16 +505,30 @@ func pingServer(network, address string, readTimeout, writeTimeout time.Duration
}
return false
}
defer conn.Close()
conn.SetWriteDeadline(time.Now().Add(writeTimeout))
defer func() {
_ = conn.Close()
}()
err = conn.SetWriteDeadline(time.Now().Add(writeTimeout))
if err != nil {
if logger != nil {
logger.Error("failed to set write deadline", zap.Error(err))
}
return false
}
_, err = conn.Write([]byte(ping))
if err != nil {
if logger != nil {
logger.Error("failed to write PING", zap.Error(err))
}
return false
}
conn.SetReadDeadline(time.Now().Add(readTimeout))
err = conn.SetReadDeadline(time.Now().Add(readTimeout))
if err != nil {
if logger != nil {
logger.Error("failed to set read deadline", zap.Error(err))
}
return false
}
resp := make([]byte, 7)
_, err = io.ReadFull(conn, resp)
if err != nil || !strings.Contains(string(resp), pong) {
Expand Down
7 changes: 4 additions & 3 deletions redisbetween.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,10 +9,11 @@ import (
"time"

"github.com/DataDog/datadog-go/statsd"
"github.com/coinbase/redisbetween/config"
"github.com/coinbase/redisbetween/proxy"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"

"github.com/coinbase/redisbetween/config"
"github.com/coinbase/redisbetween/proxy"
)

func main() {
Expand Down Expand Up @@ -90,7 +91,7 @@ func proxies(c *config.Config, log *zap.Logger) (proxies []*proxy.Proxy, err err
if err != nil {
return nil, err
}
for index, _ := range c.Upstreams {
for index := range c.Upstreams {
p, err := proxy.NewProxy(log, s, c, index)

if err != nil {
Expand Down

0 comments on commit 20e5a31

Please sign in to comment.