feat: Add badgerDB support. (#353)

See https://github.com/dgraph-io/badger
Slide: https://github.com/gopherchina/conference/blob/master/2018/1.5%20Badger_%20Fast%20Key-Value%20DB%20in%20Go.pdf

storage/badger/badger.go

@@ -0,0 +1,183 @@
+package badger
+
+import (
+	"fmt"
+	"log"
+	"os"
+	"strconv"
+
+	"github.com/appleboy/gorush/config"
+	"github.com/appleboy/gorush/storage"
+
+	"github.com/appleboy/com/convert"
+	"github.com/dgraph-io/badger"
+)
+
+// New func implements the storage interface for gorush (https://github.com/appleboy/gorush)
+func New(config config.ConfYaml) *Storage {
+	return &Storage{
+		config: config,
+	}
+}
+
+// Storage is interface structure
+type Storage struct {
+	config config.ConfYaml
+	opts   badger.Options
+	name   string
+}
+
+// Init client storage.
+func (s *Storage) Init() error {
+	s.name = "bager"
+	s.opts = badger.DefaultOptions
+	s.opts.Dir = os.TempDir() + "badger"
+	s.opts.ValueDir = os.TempDir() + "badger"
+	fmt.Println(s.opts.Dir)
+	fmt.Println(s.opts.ValueDir)
+	return nil
+}
+
+// Reset Client storage.
+func (s *Storage) Reset() {
+	s.setBadger(storage.TotalCountKey, 0)
+	s.setBadger(storage.IosSuccessKey, 0)
+	s.setBadger(storage.IosErrorKey, 0)
+	s.setBadger(storage.AndroidSuccessKey, 0)
+	s.setBadger(storage.AndroidErrorKey, 0)
+}
+
+func (s *Storage) setBadger(key string, count int64) {
+	db, err := badger.Open(s.opts)
+
+	if err != nil {
+		log.Println(s.name, "open error:", err.Error())
+		return
+	}
+
+	defer func() {
+		err := db.Close()
+		if err != nil {
+			log.Println(s.name, "close error:", err.Error())
+		}
+	}()
+
+	err = db.Update(func(txn *badger.Txn) error {
+		value := convert.ToString(count).(string)
+		return txn.Set([]byte(key), []byte(value))
+	})
+
+	if err != nil {
+		log.Println(s.name, "update error:", err.Error())
+	}
+}
+
+func (s *Storage) getBager(key string, count *int64) {
+	db, err := badger.Open(s.opts)
+
+	if err != nil {
+		log.Println(s.name, "open error:", err.Error())
+		return
+	}
+
+	defer func() {
+		err := db.Close()
+		if err != nil {
+			log.Println(s.name, "close error:", err.Error())
+		}
+	}()
+
+	err = db.View(func(txn *badger.Txn) error {
+		item, err := txn.Get([]byte(key))
+		if err != nil {
+			return err
+		}
+		val, err := item.Value()
+		if err != nil {
+			return err
+		}
+
+		i, err := strconv.ParseInt(fmt.Sprintf("%s", val), 10, 64)
+		if err != nil {
+			return err
+		}
+
+		*count = i
+
+		return nil
+	})
+
+	if err != nil {
+		log.Println(s.name, "get error:", err.Error())
+	}
+}
+
+// AddTotalCount record push notification count.
+func (s *Storage) AddTotalCount(count int64) {
+	total := s.GetTotalCount() + count
+	s.setBadger(storage.TotalCountKey, total)
+}
+
+// AddIosSuccess record counts of success iOS push notification.
+func (s *Storage) AddIosSuccess(count int64) {
+	total := s.GetIosSuccess() + count
+	s.setBadger(storage.IosSuccessKey, total)
+}
+
+// AddIosError record counts of error iOS push notification.
+func (s *Storage) AddIosError(count int64) {
+	total := s.GetIosError() + count
+	s.setBadger(storage.IosErrorKey, total)
+}
+
+// AddAndroidSuccess record counts of success Android push notification.
+func (s *Storage) AddAndroidSuccess(count int64) {
+	total := s.GetAndroidSuccess() + count
+	s.setBadger(storage.AndroidSuccessKey, total)
+}
+
+// AddAndroidError record counts of error Android push notification.
+func (s *Storage) AddAndroidError(count int64) {
+	total := s.GetAndroidError() + count
+	s.setBadger(storage.AndroidErrorKey, total)
+}
+
+// GetTotalCount show counts of all notification.
+func (s *Storage) GetTotalCount() int64 {
+	var count int64
+	s.getBager(storage.TotalCountKey, &count)
+
+	return count
+}
+
+// GetIosSuccess show success counts of iOS notification.
+func (s *Storage) GetIosSuccess() int64 {
+	var count int64
+	s.getBager(storage.IosSuccessKey, &count)
+
+	return count
+}
+
+// GetIosError show error counts of iOS notification.
+func (s *Storage) GetIosError() int64 {
+	var count int64
+	s.getBager(storage.IosErrorKey, &count)
+
+	return count
+}
+
+// GetAndroidSuccess show success counts of Android notification.
+func (s *Storage) GetAndroidSuccess() int64 {
+	var count int64
+	s.getBager(storage.AndroidSuccessKey, &count)
+
+	return count
+}
+
+// GetAndroidError show error counts of Android notification.
+func (s *Storage) GetAndroidError() int64 {
+	var count int64
+	s.getBager(storage.AndroidErrorKey, &count)
+
+	return count
+}

storage/badger/badger_test.go

@@ -0,0 +1,47 @@
+package badger
+
+import (
+	"testing"
+
+	c "github.com/appleboy/gorush/config"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestBadgerEngine(t *testing.T) {
+	var val int64
+
+	config, _ := c.LoadConf("")
+
+	badger := New(config)
+	err := badger.Init()
+	assert.Nil(t, err)
+	badger.Reset()
+
+	badger.AddTotalCount(10)
+	val = badger.GetTotalCount()
+	assert.Equal(t, int64(10), val)
+	badger.AddTotalCount(10)
+	val = badger.GetTotalCount()
+	assert.Equal(t, int64(20), val)
+
+	badger.AddIosSuccess(20)
+	val = badger.GetIosSuccess()
+	assert.Equal(t, int64(20), val)
+
+	badger.AddIosError(30)
+	val = badger.GetIosError()
+	assert.Equal(t, int64(30), val)
+
+	badger.AddAndroidSuccess(40)
+	val = badger.GetAndroidSuccess()
+	assert.Equal(t, int64(40), val)
+
+	badger.AddAndroidError(50)
+	val = badger.GetAndroidError()
+	assert.Equal(t, int64(50), val)
+
+	// test reset db
+	badger.Reset()
+	val = badger.GetAndroidError()
+	assert.Equal(t, int64(0), val)
+}