diff --git a/.gitignore b/.gitignore index 56bf979..91a8b23 100644 --- a/.gitignore +++ b/.gitignore @@ -32,3 +32,4 @@ coverage.out gorush/log/*.log build.tar.gz gorush.tar.gz +gorush.db diff --git a/README.md b/README.md index af72368..340f579 100644 --- a/README.md +++ b/README.md @@ -60,11 +60,14 @@ log: error_level: "error" stat: - engine: "memory" # support memory or redis + engine: "memory" # support memory, redis or boltdb redis: addr: "localhost:6379" password: "" db: 0 + boltdb: + path: "gorush.db" + bucket: "gorush" ``` ## Basic Usage diff --git a/config/config.yml b/config/config.yml index dd96f17..5fb8558 100644 --- a/config/config.yml +++ b/config/config.yml @@ -37,3 +37,6 @@ stat: addr: "localhost:6379" password: "" db: 0 + boltdb: + path: "gorush.db" + bucket: "gorush" diff --git a/gorush/config.go b/gorush/config.go index 29a8899..dc6f85a 100644 --- a/gorush/config.go +++ b/gorush/config.go @@ -63,6 +63,7 @@ type SectionLog struct { type SectionStat struct { Engine string `yaml:"service"` Redis SectionRedis `yaml:"redis"` + BoltDB SectionBoltDB `yaml:"boltdb"` } // SectionRedis is sub seciont of config. @@ -72,6 +73,12 @@ type SectionRedis struct { DB int64 `yaml:"db"` } +// SectionBoltDB is sub seciont of config. +type SectionBoltDB struct { + Path string `yaml:"path"` + Bucket string `yaml:"bucket"` +} + // BuildDefaultPushConf is default config setting. func BuildDefaultPushConf() ConfYaml { var conf ConfYaml @@ -114,6 +121,9 @@ func BuildDefaultPushConf() ConfYaml { conf.Stat.Redis.Password = "" conf.Stat.Redis.DB = 0 + conf.Stat.BoltDB.Path = "gorush.db" + conf.Stat.BoltDB.Bucket = "gorush" + return conf } diff --git a/gorush/status.go b/gorush/status.go index d75122f..27fd686 100644 --- a/gorush/status.go +++ b/gorush/status.go @@ -6,6 +6,7 @@ import ( "net/http" "strconv" "sync/atomic" + "github.com/asdine/storm" ) // StatusApp is app status structure @@ -69,6 +70,14 @@ func initRedis() error { return nil } +func initBoltDB() { + RushStatus.TotalCount = getTotalCount() + RushStatus.Ios.PushSuccess = getIosSuccess() + RushStatus.Ios.PushError = getIosError() + RushStatus.Android.PushSuccess = getAndroidSuccess() + RushStatus.Android.PushError = getAndroidError() +} + // InitAppStatus for initialize app status func InitAppStatus() { switch PushConf.Stat.Engine { @@ -76,18 +85,34 @@ func InitAppStatus() { initApp() case "redis": initRedis() + case "boltdb": + initBoltDB() default: initApp() } } +func boltdbSet(key string, count int64) { + db, _ := storm.Open(PushConf.Stat.BoltDB.Path) + db.Set(PushConf.Stat.BoltDB.Bucket, key, count) + defer db.Close() +} + +func boltdbGet(key string, count *int64) { + db, _ := storm.Open(PushConf.Stat.BoltDB.Path) + db.Get(PushConf.Stat.BoltDB.Bucket, key, count) + defer db.Close() +} + func addTotalCount(count int64) { switch PushConf.Stat.Engine { case "memory": atomic.AddInt64(&RushStatus.TotalCount, count) case "redis": RedisClient.Set(gorushTotalCount, strconv.Itoa(int(count)), 0) + case "boltdb": + boltdbSet(gorushTotalCount, count) default: atomic.AddInt64(&RushStatus.TotalCount, count) } @@ -99,6 +124,8 @@ func addIosSuccess(count int64) { atomic.AddInt64(&RushStatus.Ios.PushSuccess, count) case "redis": RedisClient.Set(gorushIosSuccess, strconv.Itoa(int(count)), 0) + case "boltdb": + boltdbSet(gorushIosSuccess, count) default: atomic.AddInt64(&RushStatus.Ios.PushSuccess, count) } @@ -110,6 +137,8 @@ func addIosError(count int64) { atomic.AddInt64(&RushStatus.Ios.PushError, count) case "redis": RedisClient.Set(gorushIosError, strconv.Itoa(int(count)), 0) + case "boltdb": + boltdbSet(gorushIosError, count) default: atomic.AddInt64(&RushStatus.Ios.PushError, count) } @@ -120,8 +149,9 @@ func addAndroidSuccess(count int64) { case "memory": atomic.AddInt64(&RushStatus.Android.PushSuccess, count) case "redis": - RedisClient.Set(gorushAndroidSuccess, strconv.Itoa(int(count)), 0) + case "boltdb": + boltdbSet(gorushAndroidSuccess, count) default: atomic.AddInt64(&RushStatus.Android.PushSuccess, count) } @@ -133,6 +163,8 @@ func addAndroidError(count int64) { atomic.AddInt64(&RushStatus.Android.PushError, count) case "redis": RedisClient.Set(gorushAndroidError, strconv.Itoa(int(count)), 0) + case "boltdb": + boltdbSet(gorushAndroidError, count) default: atomic.AddInt64(&RushStatus.Android.PushError, count) } @@ -144,7 +176,9 @@ func getTotalCount() int64 { case "memory": count = atomic.LoadInt64(&RushStatus.TotalCount) case "redis": - count = getRedisInt64Result(gorushAndroidError) + count = getRedisInt64Result(gorushTotalCount) + case "boltdb": + boltdbGet(gorushTotalCount, &count) default: count = atomic.LoadInt64(&RushStatus.TotalCount) } @@ -158,7 +192,9 @@ func getIosSuccess() int64 { case "memory": count = atomic.LoadInt64(&RushStatus.Ios.PushSuccess) case "redis": - count = getRedisInt64Result(gorushAndroidError) + count = getRedisInt64Result(gorushIosSuccess) + case "boltdb": + boltdbGet(gorushIosSuccess, &count) default: count = atomic.LoadInt64(&RushStatus.Ios.PushSuccess) } @@ -172,7 +208,9 @@ func getIosError() int64 { case "memory": count = atomic.LoadInt64(&RushStatus.Ios.PushError) case "redis": - count = getRedisInt64Result(gorushAndroidError) + count = getRedisInt64Result(gorushIosError) + case "boltdb": + boltdbGet(gorushIosError, &count) default: count = atomic.LoadInt64(&RushStatus.Ios.PushError) } @@ -186,7 +224,9 @@ func getAndroidSuccess() int64 { case "memory": count = atomic.LoadInt64(&RushStatus.Android.PushSuccess) case "redis": - count = getRedisInt64Result(gorushAndroidError) + count = getRedisInt64Result(gorushAndroidSuccess) + case "boltdb": + boltdbGet(gorushAndroidSuccess, &count) default: count = atomic.LoadInt64(&RushStatus.Android.PushSuccess) } @@ -201,6 +241,8 @@ func getAndroidError() int64 { count = atomic.LoadInt64(&RushStatus.Android.PushError) case "redis": count = getRedisInt64Result(gorushAndroidError) + case "boltdb": + boltdbGet(gorushAndroidError, &count) default: count = atomic.LoadInt64(&RushStatus.Android.PushError) } diff --git a/gorush/status_test.go b/gorush/status_test.go index c80c1f4..044d4a1 100644 --- a/gorush/status_test.go +++ b/gorush/status_test.go @@ -73,22 +73,22 @@ func TestStatForRedisEngine(t *testing.T) { PushConf.Stat.Redis.Addr = "localhost:6379" InitAppStatus() - addTotalCount(1000) - addIosSuccess(1000) - addIosError(1000) - addAndroidSuccess(1000) - addAndroidError(1000) + addTotalCount(10) + addIosSuccess(20) + addIosError(30) + addAndroidSuccess(40) + addAndroidError(50) val = getTotalCount() - assert.Equal(t, int64(1000), val) + assert.Equal(t, int64(10), val) val = getIosSuccess() - assert.Equal(t, int64(1000), val) + assert.Equal(t, int64(20), val) val = getIosError() - assert.Equal(t, int64(1000), val) + assert.Equal(t, int64(30), val) val = getAndroidSuccess() - assert.Equal(t, int64(1000), val) + assert.Equal(t, int64(40), val) val = getAndroidError() - assert.Equal(t, int64(1000), val) + assert.Equal(t, int64(50), val) } func TestDefaultEngine(t *testing.T) { @@ -96,20 +96,43 @@ func TestDefaultEngine(t *testing.T) { PushConf.Stat.Engine = "test" InitAppStatus() - addTotalCount(1000) - addIosSuccess(1000) - addIosError(1000) - addAndroidSuccess(1000) - addAndroidError(1000) + addTotalCount(1) + addIosSuccess(2) + addIosError(3) + addAndroidSuccess(4) + addAndroidError(5) val = getTotalCount() - assert.Equal(t, int64(1000), val) + assert.Equal(t, int64(1), val) val = getIosSuccess() - assert.Equal(t, int64(1000), val) + assert.Equal(t, int64(2), val) val = getIosError() - assert.Equal(t, int64(1000), val) + assert.Equal(t, int64(3), val) val = getAndroidSuccess() - assert.Equal(t, int64(1000), val) + assert.Equal(t, int64(4), val) val = getAndroidError() - assert.Equal(t, int64(1000), val) + assert.Equal(t, int64(5), val) +} + +func TestStatForBoltDBEngine(t *testing.T) { + var val int64 + PushConf.Stat.Engine = "boltdb" + InitAppStatus() + + addTotalCount(100) + addIosSuccess(200) + addIosError(300) + addAndroidSuccess(400) + addAndroidError(500) + + val = getTotalCount() + assert.Equal(t, int64(100), val) + val = getIosSuccess() + assert.Equal(t, int64(200), val) + val = getIosError() + assert.Equal(t, int64(300), val) + val = getAndroidSuccess() + assert.Equal(t, int64(400), val) + val = getAndroidError() + assert.Equal(t, int64(500), val) }