From 077f836d1c10b2fadf02657bf85bf43962462d35 Mon Sep 17 00:00:00 2001
From: Bo-Yi Wu <appleboy.tw@gmail.com>
Date: Mon, 2 May 2016 11:04:19 +0800
Subject: [PATCH] Fixed #66 Support memory interface.

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
---
 gorush/storage.go             | 15 ++++++++
 storage/memory/memory.go      | 67 +++++++++++++++++++++++++++++++++++
 storage/memory/memory_test.go | 33 +++++++++++++++++
 3 files changed, 115 insertions(+)
 create mode 100644 gorush/storage.go
 create mode 100644 storage/memory/memory.go
 create mode 100644 storage/memory/memory_test.go

diff --git a/gorush/storage.go b/gorush/storage.go
new file mode 100644
index 0000000..a17e64c
--- /dev/null
+++ b/gorush/storage.go
@@ -0,0 +1,15 @@
+package gorush
+
+// Storage interface
+type Storage interface {
+	addTotalCount(int64)
+	addIosSuccess(int64)
+	addIosError(int64)
+	addAndroidSuccess(int64)
+	addAndroidError(int64)
+	getTotalCount() int64
+	getIosSuccess() int64
+	getIosError() int64
+	getAndroidSuccess() int64
+	getAndroidError() int64
+}
diff --git a/storage/memory/memory.go b/storage/memory/memory.go
new file mode 100644
index 0000000..d9cdec0
--- /dev/null
+++ b/storage/memory/memory.go
@@ -0,0 +1,67 @@
+package memory
+
+import (
+	"github.com/appleboy/gorush/gorush"
+	"sync/atomic"
+)
+
+// Storage implements the storage interface for gorush (https://github.com/appleboy/gorush)
+func New(stat gorush.StatusApp) *Storage {
+	return &Storage{
+		stat: stat,
+	}
+}
+
+type Storage struct {
+	stat gorush.StatusApp
+}
+
+func (s *Storage) addTotalCount(count int64) {
+	atomic.AddInt64(&s.stat.TotalCount, count)
+}
+
+func (s *Storage) addIosSuccess(count int64) {
+	atomic.AddInt64(&s.stat.Ios.PushSuccess, count)
+}
+
+func (s *Storage) addIosError(count int64) {
+	atomic.AddInt64(&s.stat.Ios.PushError, count)
+}
+
+func (s *Storage) addAndroidSuccess(count int64) {
+	atomic.AddInt64(&s.stat.Android.PushSuccess, count)
+}
+
+func (s *Storage) addAndroidError(count int64) {
+	atomic.AddInt64(&s.stat.Android.PushError, count)
+}
+
+func (s *Storage) getTotalCount() int64 {
+	count := atomic.LoadInt64(&s.stat.TotalCount)
+
+	return count
+}
+
+func (s *Storage) getIosSuccess() int64 {
+	count := atomic.LoadInt64(&s.stat.Ios.PushSuccess)
+
+	return count
+}
+
+func (s *Storage) getIosError() int64 {
+	count := atomic.LoadInt64(&s.stat.Ios.PushError)
+
+	return count
+}
+
+func (s *Storage) getAndroidSuccess() int64 {
+	count := atomic.LoadInt64(&s.stat.Android.PushSuccess)
+
+	return count
+}
+
+func (s *Storage) getAndroidError() int64 {
+	count := atomic.LoadInt64(&s.stat.Android.PushError)
+
+	return count
+}
diff --git a/storage/memory/memory_test.go b/storage/memory/memory_test.go
new file mode 100644
index 0000000..54d670c
--- /dev/null
+++ b/storage/memory/memory_test.go
@@ -0,0 +1,33 @@
+package memory
+
+import (
+	"github.com/appleboy/gorush/gorush"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestMemoryEngine(t *testing.T) {
+	var val int64
+
+	memory := New(gorush.StatusApp{})
+
+	memory.addTotalCount(1)
+	val = memory.getTotalCount()
+	assert.Equal(t, int64(1), val)
+
+	memory.addIosSuccess(2)
+	val = memory.getIosSuccess()
+	assert.Equal(t, int64(2), val)
+
+	memory.addIosError(3)
+	val = memory.getIosError()
+	assert.Equal(t, int64(3), val)
+
+	memory.addAndroidSuccess(4)
+	val = memory.getAndroidSuccess()
+	assert.Equal(t, int64(4), val)
+
+	memory.addAndroidError(5)
+	val = memory.getAndroidError()
+	assert.Equal(t, int64(5), val)
+}