update

core/store.go

@@ -1,6 +1,9 @@
 package core
 
 import (
+	"fmt"
+	"strconv"
+
 	"github.com/boltdb/bolt"
 )
 
@@ -25,13 +28,13 @@ func init() {
 	}
 }
 
-func (bucket Bucket) Set(key, value string) {
+func (bucket Bucket) Set(key string, value interface{}) {
 	db.Update(func(tx *bolt.Tx) error {
 		b := tx.Bucket([]byte(bucket))
 		if b == nil {
 			b, _ = tx.CreateBucket([]byte(bucket))
 		}
-		b.Put([]byte(key), []byte(value))
+		b.Put([]byte(key), []byte(fmt.Sprint(value)))
 		return nil
 	})
 }
@@ -56,6 +59,22 @@ func (bucket Bucket) Get(kv ...string) string {
 	return value
 }
 
+func (bucket Bucket) GetInt(key string, vs ...int) int {
+	var value int
+	if len(vs) != 0 {
+		value = vs[0]
+	}
+	db.View(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte(bucket))
+		if b == nil {
+			return nil
+		}
+		value = Int(string(b.Get([]byte(key))))
+		return nil
+	})
+	return value
+}
+
 func (bucket Bucket) Foreach(f func(k, v []byte) error) {
 	db.View(func(tx *bolt.Tx) error {
 		b := tx.Bucket([]byte(bucket))
@@ -63,3 +82,8 @@ func (bucket Bucket) Foreach(f func(k, v []byte) error) {
 		return nil
 	})
 }
+
+var Int = func(s string) int {
+	i, _ := strconv.Atoi(s)
+	return i
+}

core/test.go

@@ -9,6 +9,7 @@ import (
 func init() {
 	AddCommand("", []Function{
 		{
+			Admin: true,
 			Rules: []string{"set ? ? ?"},
 			Handle: func(s im.Sender) interface{} {
 				b := Bucket(s.Get(0))
@@ -20,6 +21,7 @@ func init() {
 			},
 		},
 		{
+			Admin: true,
 			Rules: []string{"delete ? ?"},
 			Handle: func(s im.Sender) interface{} {
 				b := Bucket(s.Get(0))
@@ -31,6 +33,7 @@ func init() {
 			},
 		},
 		{
+			Admin: true,
 			Rules: []string{"get ? ?"},
 			Handle: func(s im.Sender) interface{} {
 				b := Bucket(s.Get(0))