Add config to docker, env overrides

docker/Dockerfile

@@ -8,5 +8,6 @@ RUN apk --no-cache add tini
 ENTRYPOINT ["/sbin/tini", "-g", "--"]
 CMD ["godns"]
 
+COPY etc/godns.conf /etc/godns.conf
 COPY godns /usr/local/bin/godns
 RUN chmod +x /usr/local/bin/godns

docker/Dockerfile.arm

@@ -10,5 +10,6 @@ RUN apk --no-cache add tini
 ENTRYPOINT ["/sbin/tini", "-g", "--"]
 CMD ["godns"]
 
+COPY etc/godns.conf /etc/godns.conf
 COPY godns-arm /usr/local/bin/godns
 RUN chmod +x /usr/local/bin/godns

docker/Dockerfile.arm64

@@ -10,5 +10,6 @@ RUN apk --no-cache add tini
 ENTRYPOINT ["/sbin/tini", "-g", "--"]
 CMD ["godns"]
 
+COPY etc/godns.conf /etc/godns.conf
 COPY godns-arm64 /usr/local/bin/godns
 RUN chmod +x /usr/local/bin/godns

etc/godns.conf

@@ -16,7 +16,6 @@ port = 53
 # Semicolon separate multiple files.
 server-list-file = "./etc/apple.china.conf;./etc/google.china.conf"
 resolv-file = "/etc/resolv.conf"
-dns-over-https = "https://cloudflare-dns.com/dns-query"
 timeout = 5  # 5 seconds
 # The concurrency interval request upstream recursive server
 # Match the PR15, https://github.com/kenshinx/godns/pull/15

hosts_redis.go

@@ -48,7 +48,13 @@ func NewRedisProvider(rc *redis.Client, key string) HostProvider {
 					continue
 				}
 
+				rh.mu.RLock()
 				rh.hosts[recordName] = string(b)
+				rh.mu.RUnlock()
+			} else if msg.Channel == "godns:remove_record" {
+				rh.mu.RLock()
+				delete(rh.hosts, string(msg.Message))
+				rh.mu.RUnlock()
 			} else if msg.Channel == keyspaceEvent {
 				rh.Refresh()
 			}

server.go

@@ -6,6 +6,7 @@ import (
 	"time"
 
 	"github.com/miekg/dns"
+	"os"
 )
 
 type Server struct {
@@ -16,42 +17,48 @@ type Server struct {
 }
 
 func (s *Server) Addr() string {
+	if addr := os.Getenv("SERVER_BIND"); addr != "" {
+		return addr
+	}
+
 	return net.JoinHostPort(s.host, strconv.Itoa(s.port))
 }
 
 func (s *Server) Run() {
-	Handler := NewHandler()
+	handler := NewHandler()
 
 	tcpHandler := dns.NewServeMux()
-	tcpHandler.HandleFunc(".", Handler.DoTCP)
+	tcpHandler.HandleFunc(".", handler.DoTCP)
 
 	udpHandler := dns.NewServeMux()
-	udpHandler.HandleFunc(".", Handler.DoUDP)
+	udpHandler.HandleFunc(".", handler.DoUDP)
 
-	tcpServer := &dns.Server{Addr: s.Addr(),
+	tcpServer := &dns.Server{
+		Addr: s.Addr(),
 		Net: "tcp",
 		Handler: tcpHandler,
 		ReadTimeout: s.rTimeout,
-		WriteTimeout: s.wTimeout}
+		WriteTimeout: s.wTimeout,
+	}
 
-	udpServer := &dns.Server{Addr: s.Addr(),
+	udpServer := &dns.Server{
+		Addr: s.Addr(),
 		Net: "udp",
 		Handler: udpHandler,
 		UDPSize: 65535,
 		ReadTimeout: s.rTimeout,
-		WriteTimeout: s.wTimeout}
+		WriteTimeout: s.wTimeout,
+	}
 
 	go s.start(udpServer)
 	go s.start(tcpServer)
-
 }
 
 func (s *Server) start(ds *dns.Server) {
-
 	logger.Info("Start %s listener on %s", ds.Net, s.Addr())
+
 	err := ds.ListenAndServe()
 	if err != nil {
 		logger.Error("Start %s listener on %s failed:%s", ds.Net, s.Addr(), err.Error())
 	}
-
 }

settings.go

@@ -7,6 +7,7 @@ import (
 	"strconv"
 
 	"github.com/BurntSushi/toml"
+	"strings"
 )
 
 var (
@@ -59,6 +60,10 @@ type MemcacheSettings struct {
 }
 
 func (s RedisSettings) Addr() string {
+	if addr := os.Getenv("REDIS_ADDR"); addr != "" {
+		return addr
+	}
+
 	return s.Host + ":" + strconv.Itoa(s.Port)
 }
 
@@ -92,10 +97,9 @@ type HostsSettings struct {
 }
 
 func init() {
-
 	var configFile string
 
-	flag.StringVar(&configFile, "c", "./etc/godns.conf", "Look for godns toml-formatting config file in this directory")
+	flag.StringVar(&configFile, "c", "/etc/godns.conf", "Look for godns toml-formatting config file in this directory")
 	flag.Parse()
 
 	if _, err := toml.DecodeFile(configFile, &settings); err != nil {
@@ -104,4 +108,12 @@ func init() {
 		os.Exit(1)
 	}
 
+	// Env overrides
+	if cacheBackend := os.Getenv("CACHE_BACKEND"); cacheBackend != "" {
+		settings.Cache.Backend = cacheBackend
+	}
+
+	if memcacheServers := os.Getenv("MEMCACHE_SERVERS"); memcacheServers != "" {
+		settings.Memcache.Servers = strings.Split(memcacheServers, ",")
+	}
 }