Refactor CoapConfig struct in main.go

The CoapConfig struct in main.go has been refactored to improve code organization and readability. The struct now includes separate fields for host, port, DTLS, targetHost, and targetPort. This change allows for better configuration of CoAP settings. The code has been updated to reflect these changes.

Signed-off-by: SammyOina <sammyoina@gmail.com>

Author: SammyOina

cmd/main.go

@@ -104,16 +104,19 @@ type config struct {
 	httpConfig HTTPConfig
 	mqttConfig MQTTConfig
 	wsConfig   WSConfig
-
-	coapHost       string
-	coapPort       string
-	coapDTLS       bool
-	coapTargetHost string
-	coapTargetPort string
+	coapConfig CoapConfig
 
 	logLevel string
 }
 
+type CoapConfig struct {
+	host       string
+	port       string
+	DTLS       bool
+	targetHost string
+	targetPort string
+}
+
 type WSConfig struct {
 	host         string
 	port         string
@@ -185,7 +188,7 @@ func main() {
 	}
 
 	switch {
-	case cfg.coapDTLS:
+	case cfg.coapConfig.DTLS:
 		tlsCfg, err := mptls.LoadTLSCfg(cfg.caCerts, cfg.serverCert, cfg.serverKey)
 		if err != nil {
 			errs <- err
@@ -194,15 +197,15 @@ func main() {
 			Certificates: tlsCfg.Certificates,
 			ClientCAs:    tlsCfg.ClientCAs,
 		}
-		go proxyCoapDTLS(cfg, dtlsCfg, logger, h, errs)
+		go proxyCoapDTLS(cfg.coapConfig, dtlsCfg, logger, h, errs)
 	case cfg.clientTLS:
 		tlsCfg, err := mptls.LoadTLSCfg(cfg.caCerts, cfg.serverCert, cfg.serverKey)
 		if err != nil {
 			errs <- err
 		}
-		go proxyCoapTLS(cfg, tlsCfg, logger, h, errs)
+		go proxyCoapTLS(cfg.coapConfig, tlsCfg, logger, h, errs)
 	default:
-		go proxyCoap(cfg, logger, h, errs)
+		go proxyCoap(cfg.coapConfig, logger, h, errs)
 	}
 
 	go func() {
@@ -271,11 +274,13 @@ func loadConfig() config {
 		},
 
 		// CoAP
-		coapHost:       env(envCoAPHost, defCoAPHost),
-		coapPort:       env(envCoAPPort, defCoAPPort),
-		coapDTLS:       dtls,
-		coapTargetHost: env(envCoAPTargetHost, defCoAPTargetHost),
-		coapTargetPort: env(envCoAPTargetPort, defCoAPTargetPort),
+		coapConfig: CoapConfig{
+			host:       env(envCoAPHost, defCoAPHost),
+			port:       env(envCoAPPort, defCoAPPort),
+			DTLS:       dtls,
+			targetHost: env(envCoAPTargetHost, defCoAPTargetHost),
+			targetPort: env(envCoAPTargetPort, defCoAPTargetPort),
+		},
 
 		// Log
 		logLevel: env(envLogLevel, defLogLevel),
@@ -336,9 +341,9 @@ func proxyHTTPS(ctx context.Context, cfg HTTPConfig, logger mflog.Logger, handle
 	http.HandleFunc("/", hp.Handler)
 	errs <- hp.ListenTLS(cfg.serverCert, cfg.serverKey)
 }
-func proxyCoapTLS(cfg config, tlsCfg *tls.Config, logger mflog.Logger, errs chan error) {
-	address := fmt.Sprintf("%s:%s", cfg.coapHost, cfg.coapPort)
-	target := fmt.Sprintf("%s:%s", cfg.coapTargetHost, cfg.coapTargetPort)
+func proxyCoapTLS(cfg CoapConfig, tlsCfg *tls.Config, logger mflog.Logger, handler session.Handler, errs chan error) {
+	address := fmt.Sprintf("%s:%s", cfg.host, cfg.port)
+	target := fmt.Sprintf("%s:%s", cfg.targetHost, cfg.targetPort)
 	cp, err := coap.NewProxy(address, target, logger, handler)
 	if err != nil {
 		errs <- err
@@ -347,9 +352,9 @@ func proxyCoapTLS(cfg config, tlsCfg *tls.Config, logger mflog.Logger, errs chan
 	errs <- cp.ListenTLS(tlsCfg)
 }
 
-func proxyCoapDTLS(cfg config, dtlsCfg *dtls.Config, logger mflog.Logger, handler session.Handler, errs chan error) {
-	address := fmt.Sprintf("%s:%s", cfg.coapHost, cfg.coapPort)
-	target := fmt.Sprintf("%s:%s", cfg.coapTargetHost, cfg.coapTargetPort)
+func proxyCoapDTLS(cfg CoapConfig, dtlsCfg *dtls.Config, logger mflog.Logger, handler session.Handler, errs chan error) {
+	address := fmt.Sprintf("%s:%s", cfg.host, cfg.port)
+	target := fmt.Sprintf("%s:%s", cfg.targetHost, cfg.targetPort)
 	cp, err := coap.NewProxy(address, target, logger, handler)
 	if err != nil {
 		errs <- err
@@ -358,9 +363,9 @@ func proxyCoapDTLS(cfg config, dtlsCfg *dtls.Config, logger mflog.Logger, handle
 	errs <- cp.ListenDLS(dtlsCfg)
 }
 
-func proxyCoap(cfg config, logger mflog.Logger, handler session.Handler, errs chan error) {
-	address := fmt.Sprintf("%s:%s", cfg.coapHost, cfg.coapPort)
-	target := fmt.Sprintf("%s:%s", cfg.coapTargetHost, cfg.coapTargetPort)
+func proxyCoap(cfg CoapConfig, logger mflog.Logger, handler session.Handler, errs chan error) {
+	address := fmt.Sprintf("%s:%s", cfg.host, cfg.port)
+	target := fmt.Sprintf("%s:%s", cfg.targetHost, cfg.targetPort)
 	cp, err := coap.NewProxy(address, target, logger, handler)
 	if err != nil {
 		errs <- err

go.mod

@@ -7,13 +7,13 @@ require (
 	github.com/google/uuid v1.3.0
 	github.com/gorilla/websocket v1.5.0
 	github.com/mainflux/mainflux v0.12.0
+	github.com/pion/dtls/v2 v2.2.8-0.20230905141523-2b584af66577
 )
 
 require (
 	github.com/dsnet/golib/memfile v1.0.0 // indirect
 	github.com/hashicorp/errwrap v1.1.0 // indirect
 	github.com/hashicorp/go-multierror v1.1.1 // indirect
-	github.com/pion/dtls/v2 v2.2.8-0.20230905141523-2b584af66577 // indirect
 	github.com/pion/logging v0.2.2 // indirect
 	github.com/pion/transport/v3 v3.0.1 // indirect
 	go.uber.org/atomic v1.11.0 // indirect