Skip to content

Commit 229dad8

Browse files
committed
Refactor construct server handler
1 parent daf7585 commit 229dad8

File tree

2 files changed

+7
-6
lines changed

2 files changed

+7
-6
lines changed

cmd/server/main.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@ func main() {
3333
repo := repository.NewSQLiteDNSRepository(dbClient.DB)
3434
service := dns.NewService(repo)
3535

36-
dnsServer := server.NewDnsServer(service, cfg.Dns.Addr, cfg.Dns.Net)
36+
dnsHandler := server.NewDnsHandler(service)
37+
dnsServer := server.NewDnsServer(dnsHandler, cfg.Dns.Addr, cfg.Dns.Net)
3738
dnsErrChan := StartDNSServer(dnsServer)
3839

3940
httpHandler := http.NewHandler(service)

internal/adapters/dns/server.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4,20 +4,20 @@ import (
44
"context"
55
server "github.com/miekg/dns"
66
"go-dyndns/internal/adapters/dns/middleware"
7-
"go-dyndns/internal/core/dns"
87
)
98

109
type Server struct {
1110
DnsServer *server.Server
1211
}
1312

14-
func NewDnsServer(service dns.Service, addr, net string) *Server {
15-
handler := NewDnsHandler(service)
16-
dnsServer := &server.Server{Addr: addr, Net: net}
13+
func NewDnsServer(handler *Handler, addr, net string) *Server {
14+
dnsServer := &server.Server{
15+
Addr: addr,
16+
Net: net,
17+
}
1718
server.HandleFunc(".", middleware.LoggingMiddleware(handler.HandleDNSRequest))
1819

1920
return &Server{DnsServer: dnsServer}
20-
2121
}
2222

2323
func (s *Server) Start() error {

0 commit comments

Comments
 (0)