Fix retry mechanism for APNS

.github/workflows/unit-tests.yaml

@@ -48,7 +48,7 @@ jobs:
       - name: Build
         run: docker run -v $PWD:/go/src/github.com/topfreegames/pusher tfgco/pusher:ci-test go build -v -o bin/pusher main.go
       - name: Test
-        run: docker run -v $PWD:/go/src/github.com/topfreegames/pusher tfgco/pusher:ci-test ginkgo -v -r --randomizeAllSpecs --randomizeSuites --cover --focus="\[Unit\].*" .
+        run: docker run -v $PWD:/go/src/github.com/topfreegames/pusher tfgco/pusher:ci-test ginkgo -v -r --randomizeAllSpecs --randomizeSuites --cover --focus="\[Unit\].*" --skipPackage=e2e .
       - name: Lint
         continue-on-error: true
         run: docker run -v $PWD:/go/src/github.com/topfreegames/pusher tfgco/pusher:ci-test golangci-lint run

Makefile

@@ -105,8 +105,8 @@ test-unit:
 	@echo "-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-="
 	@echo
 	@export $ACK_GINKGO_RC=true
-	@$(GINKGO) --race -trace -r --randomizeAllSpecs --randomizeSuites --cover --focus="\[Unit\].*" .
-	@$(MAKE) test-coverage-func
+	@$(GINKGO) -trace -r --randomizeAllSpecs --randomizeSuites --cover --focus="\[Unit\].*" --skipPackage=e2e .
+	@#$(MAKE) test-coverage-func
 	@echo
 	@echo "-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-="
 	@echo "=                  Unit tests finished.                  ="
@@ -120,7 +120,7 @@ run-integration-test:
 	@echo "-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-="
 	@echo
 	@export $ACK_GINKGO_RC=true
-	@$(GINKGO) --race -trace -r -tags=integration --randomizeAllSpecs --randomizeSuites --focus="\[Integration\].*" .
+	@$(GINKGO) -trace -r -tags=integration --randomizeAllSpecs --randomizeSuites --focus="\[Integration\].*" .
 	@echo
 	@echo "-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-="
 	@echo "=               Integration tests finished.              ="
@@ -173,4 +173,7 @@ integration-test-container-dev: build-image-dev start-deps-container-dev test-db
 
 .PHONY: mocks
 mocks:
-	$(MOCKGENERATE) -source=interfaces/client.go -destination=mocks/firebase/client.go
+	$(MOCKGENERATE) -source=interfaces/client.go -destination=mocks/firebase/client.go
+	$(MOCKGENERATE) -source=interfaces/apns.go -destination=mocks/interfaces/apns.go
+	$(MOCKGENERATE) -source=interfaces/statsd.go -destination=mocks/interfaces/statsd.go
+	$(MOCKGENERATE) -source=interfaces/feedback_reporter.go -destination=mocks/interfaces/feedback_reporter.go

cmd/apns.go

@@ -28,14 +28,16 @@ import (
 	"github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
+	"github.com/topfreegames/pusher/config"
 	"github.com/topfreegames/pusher/interfaces"
 	"github.com/topfreegames/pusher/pusher"
 	"github.com/topfreegames/pusher/util"
 )
 
 func startApns(
 	debug, json, production bool,
-	config *viper.Viper,
+	vConfig *viper.Viper,
+	config *config.Config,
 	statsdClientOrNil interfaces.StatsDClient,
 	dbOrNil interfaces.DB,
 	queueOrNil interfaces.APNSPushQueue,
@@ -49,7 +51,7 @@ func startApns(
 	} else {
 		log.Level = logrus.InfoLevel
 	}
-	return pusher.NewAPNSPusher(production, config, log, statsdClientOrNil, dbOrNil, queueOrNil)
+	return pusher.NewAPNSPusher(production, vConfig, config, log, statsdClientOrNil, dbOrNil, queueOrNil)
 }
 
 // apnsCmd represents the apns command
@@ -58,19 +60,19 @@ var apnsCmd = &cobra.Command{
 	Short: "starts pusher in apns mode",
 	Long:  `starts pusher in apns mode`,
 	Run: func(cmd *cobra.Command, args []string) {
-		config, err := util.NewViperWithConfigFile(cfgFile)
+		config, vConfig, err := config.NewConfigAndViper(cfgFile)
 		if err != nil {
 			panic(err)
 		}
 
-		sentryURL := config.GetString("sentry.url")
+		sentryURL := vConfig.GetString("sentry.url")
 		if sentryURL != "" {
 			raven.SetDSN(sentryURL)
 		}
 
 		ctx := context.Background()
 
-		apnsPusher, err := startApns(debug, json, production, config, nil, nil, nil)
+		apnsPusher, err := startApns(debug, json, production, vConfig, config, nil, nil, nil)
 		if err != nil {
 			raven.CaptureErrorAndWait(err, map[string]string{
 				"version": util.Version,

cmd/apns_test.go

@@ -24,30 +24,31 @@ package cmd
 
 import (
 	"fmt"
+	"github.com/topfreegames/pusher/config"
 	"os"
 
 	. "github.com/onsi/ginkgo"
 	. "github.com/onsi/gomega"
 	"github.com/sirupsen/logrus"
 	"github.com/spf13/viper"
 	"github.com/topfreegames/pusher/mocks"
-	"github.com/topfreegames/pusher/util"
 )
 
 var _ = Describe("APNS", func() {
-	cfg := os.Getenv("CONFIG_FILE")
-	if cfg == "" {
-		cfg = "../config/test.yaml"
+	configFile := os.Getenv("CONFIG_FILE")
+	if configFile == "" {
+		configFile = "../config/test.yaml"
 	}
 
-	var config *viper.Viper
+	var vConfig *viper.Viper
+	var cfg *config.Config
 	var mockPushQueue *mocks.APNSPushQueueMock
 	var mockDB *mocks.PGMock
 	var mockStatsDClient *mocks.StatsDClientMock
 
 	BeforeEach(func() {
 		var err error
-		config, err = util.NewViperWithConfigFile(cfg)
+		cfg, vConfig, err = config.NewConfigAndViper(configFile)
 		Expect(err).NotTo(HaveOccurred())
 		mockDB = mocks.NewPGMock(0, 1)
 		mockPushQueue = mocks.NewAPNSPushQueueMock()
@@ -56,7 +57,7 @@ var _ = Describe("APNS", func() {
 
 	Describe("[Unit]", func() {
 		It("Should return apnsPusher without errors", func() {
-			apnsPusher, err := startApns(false, false, false, config, mockStatsDClient, mockDB, mockPushQueue)
+			apnsPusher, err := startApns(false, false, false, vConfig, cfg, mockStatsDClient, mockDB, mockPushQueue)
 			Expect(err).NotTo(HaveOccurred())
 			Expect(apnsPusher).NotTo(BeNil())
 			Expect(apnsPusher.ViperConfig).NotTo(BeNil())
@@ -66,21 +67,21 @@ var _ = Describe("APNS", func() {
 		})
 
 		It("Should set log to json format", func() {
-			apnsPusher, err := startApns(false, true, false, config, mockStatsDClient, mockDB, mockPushQueue)
+			apnsPusher, err := startApns(false, true, false, vConfig, cfg, mockStatsDClient, mockDB, mockPushQueue)
 			Expect(err).NotTo(HaveOccurred())
 			Expect(apnsPusher).NotTo(BeNil())
 			Expect(fmt.Sprintf("%T", apnsPusher.Logger.Formatter)).To(Equal(fmt.Sprintf("%T", &logrus.JSONFormatter{})))
 		})
 
 		It("Should set log to debug", func() {
-			apnsPusher, err := startApns(true, false, false, config, mockStatsDClient, mockDB, mockPushQueue)
+			apnsPusher, err := startApns(true, false, false, vConfig, cfg, mockStatsDClient, mockDB, mockPushQueue)
 			Expect(err).NotTo(HaveOccurred())
 			Expect(apnsPusher).NotTo(BeNil())
 			Expect(apnsPusher.Logger.Level).To(Equal(logrus.DebugLevel))
 		})
 
 		It("Should set log to production", func() {
-			apnsPusher, err := startApns(false, false, true, config, mockStatsDClient, mockDB, mockPushQueue)
+			apnsPusher, err := startApns(false, false, true, vConfig, cfg, mockStatsDClient, mockDB, mockPushQueue)
 			Expect(err).NotTo(HaveOccurred())
 			Expect(apnsPusher).NotTo(BeNil())
 			Expect(apnsPusher.IsProduction).To(BeTrue())

config/config.go

@@ -14,16 +14,34 @@ type (
 	// Config is the struct that holds all the configuration for the Pusher.
 	Config struct {
 		GCM                     GCM
+		Apns                    Apns
+		Queue                   Kafka
 		GracefulShutdownTimeout int
 	}
 
+	Kafka struct {
+		Brokers string
+	}
+
 	GCM struct {
 		Apps               string
 		PingInterval       int
 		PingTimeout        int
 		MaxPendingMessages int
 		LogStatsInterval   int
 	}
+
+	Apns struct {
+		Apps  string
+		Certs map[string]Cert
+	}
+
+	Cert struct {
+		AuthKeyPath string
+		KeyID       string
+		TeamID      string
+		Topic       string
+	}
 )
 
 // NewConfigAndViper returns a new Config object and the corresponding viper instance.
@@ -45,7 +63,7 @@ func NewConfigAndViper(configFile string) (*Config, *viper.Viper, error) {
 	return config, v, nil
 }
 
-func (c *Config) GetAppsArray() []string {
+func (c *Config) GetGcmAppsArray() []string {
 	arr := strings.Split(c.GCM.Apps, ",")
 	res := make([]string, 0, len(arr))
 	for _, a := range arr {
@@ -55,6 +73,16 @@ func (c *Config) GetAppsArray() []string {
 	return res
 }
 
+func (c *Config) GetApnsAppsArray() []string {
+	arr := strings.Split(c.Apns.Apps, ",")
+	res := make([]string, 0, len(arr))
+	for _, a := range arr {
+		res = append(res, strings.TrimSpace(a))
+	}
+
+	return res
+}
+
 func decodeHookFunc() viper.DecoderConfigOption {
 	hooks := mapstructure.ComposeDecodeHookFunc(
 		StringToMapStringHookFunc(),

config/docker_test.yaml

@@ -38,7 +38,7 @@ feedback:
     topics: "com.games.test.feedbacks"
     brokers: "kafka:9092"
   cache:
-    requestTimeout: 100
+    requestTimeout: 3000
     cleaningInterval: 20
 stats:
   reporters:

config/test.yaml

@@ -38,7 +38,7 @@ feedback:
     topics: "com.games.test.feedbacks"
     brokers: "localhost:9941"
   cache:
-    requestTimeout: 100
+    requestTimeout: 3000
     cleaningInterval: 20
 stats:
   reporters: