added golang

backend/database/mongo.go

@@ -0,0 +1,45 @@
+package database
+
+import (
+	"github.com/globalsign/mgo"
+	"github.com/spf13/viper"
+)
+
+var Session *mgo.Session
+
+func GetSession() *mgo.Session {
+	return Session.Copy()
+}
+
+func GetDb() (*mgo.Session, *mgo.Database) {
+	s := GetSession()
+	return s, s.DB(viper.GetString("mongo.db"))
+}
+
+func GetCol(collectionName string) (*mgo.Session, *mgo.Collection) {
+	s := GetSession()
+	db := s.DB(viper.GetString("mongo.db"))
+	col := db.C(collectionName)
+	return s, col
+}
+
+func GetGridFs(prefix string) (*mgo.Session, *mgo.GridFS) {
+	s, db := GetDb()
+	gf := db.GridFS(prefix)
+	return s, gf
+}
+
+func InitMongo() error {
+	var mongoHost = viper.GetString("mongo.host")
+	var mongoPort = viper.GetString("mongo.port")
+	var mongoDb = viper.GetString("mongo.db")
+
+	if Session == nil {
+		sess, err := mgo.Dial("mongodb://" + mongoHost + ":" + mongoPort + "/" + mongoDb)
+		if err != nil {
+			return err
+		}
+		Session = sess
+	}
+	return nil
+}

backend/database/pubsub.go

@@ -0,0 +1,72 @@
+package database
+
+import (
+	"fmt"
+	"github.com/apex/log"
+	"github.com/gomodule/redigo/redis"
+	"unsafe"
+)
+
+type SubscribeCallback func(channel, message string)
+
+type Subscriber struct {
+	client redis.PubSubConn
+	cbMap  map[string]SubscribeCallback
+}
+
+func (c *Subscriber) Connect() {
+	conn, err := GetRedisConn()
+	if err != nil {
+		log.Fatalf("redis dial failed.")
+	}
+
+	c.client = redis.PubSubConn{Conn: conn}
+	c.cbMap = make(map[string]SubscribeCallback)
+
+	go func() {
+		for {
+			log.Debug("wait...")
+			switch res := c.client.Receive().(type) {
+			case redis.Message:
+				channel := (*string)(unsafe.Pointer(&res.Channel))
+				message := (*string)(unsafe.Pointer(&res.Data))
+				c.cbMap[*channel](*channel, *message)
+			case redis.Subscription:
+				fmt.Printf("%s: %s %d\n", res.Channel, res.Kind, res.Count)
+			case error:
+				log.Error("error handle...")
+				continue
+			}
+		}
+	}()
+
+}
+
+func (c *Subscriber) Close() {
+	err := c.client.Close()
+	if err != nil {
+		log.Errorf("redis close error.")
+	}
+}
+
+func (c *Subscriber) Subscribe(channel interface{}, cb SubscribeCallback) {
+	err := c.client.Subscribe(channel)
+	if err != nil {
+		log.Fatalf("redis Subscribe error.")
+	}
+
+	c.cbMap[channel.(string)] = cb
+}
+
+func Publish(channel string, msg string) error {
+	c, err := GetRedisConn()
+	if err != nil {
+		return err
+	}
+
+	if _, err := c.Do("PUBLISH", channel, msg); err != nil {
+		return err
+	}
+
+	return nil
+}

backend/database/redis.go

@@ -0,0 +1,119 @@
+package database
+
+import (
+	"github.com/gomodule/redigo/redis"
+	"github.com/spf13/viper"
+	"runtime/debug"
+)
+
+var RedisClient = Redis{}
+
+type ConsumeFunc func(channel string, message []byte) error
+
+type Redis struct {
+}
+
+func (r *Redis) RPush(collection string, value interface{}) error {
+	c, err := GetRedisConn()
+	if err != nil {
+		debug.PrintStack()
+		return err
+	}
+	defer c.Close()
+
+	if _, err := c.Do("RPUSH", collection, value); err != nil {
+		debug.PrintStack()
+		return err
+	}
+	return nil
+}
+
+func (r *Redis) LPop(collection string) (string, error) {
+	c, err := GetRedisConn()
+	if err != nil {
+		debug.PrintStack()
+		return "", err
+	}
+	defer c.Close()
+
+	value, err2 := redis.String(c.Do("LPOP", collection))
+	if err2 != nil {
+		return value, err2
+	}
+	return value, nil
+}
+
+func (r *Redis) HSet(collection string, key string, value string) error {
+	c, err := GetRedisConn()
+	if err != nil {
+		debug.PrintStack()
+		return err
+	}
+	defer c.Close()
+
+	if _, err := c.Do("HSET", collection, key, value); err != nil {
+		debug.PrintStack()
+		return err
+	}
+	return nil
+}
+
+func (r *Redis) HGet(collection string, key string) (string, error) {
+	c, err := GetRedisConn()
+	if err != nil {
+		debug.PrintStack()
+		return "", err
+	}
+	defer c.Close()
+
+	value, err2 := redis.String(c.Do("HGET", collection, key))
+	if err2 != nil {
+		return value, err2
+	}
+	return value, nil
+}
+
+func (r *Redis) HDel(collection string, key string) error {
+	c, err := GetRedisConn()
+	if err != nil {
+		debug.PrintStack()
+		return err
+	}
+	defer c.Close()
+
+	if _, err := c.Do("HDEL", collection, key); err != nil {
+		return err
+	}
+	return nil
+}
+
+func (r *Redis) HKeys(collection string) ([]string, error) {
+	c, err := GetRedisConn()
+	if err != nil {
+		debug.PrintStack()
+		return []string{}, err
+	}
+	defer c.Close()
+
+	value, err2 := redis.Strings(c.Do("HKeys", collection))
+	if err2 != nil {
+		return []string{}, err2
+	}
+	return value, nil
+}
+
+func GetRedisConn() (redis.Conn, error) {
+	c, err := redis.Dial(
+		viper.GetString("redis.network"),
+		viper.GetString("redis.address"),
+	)
+	if err != nil {
+		debug.PrintStack()
+		return c, err
+	}
+	return c, nil
+}
+
+func InitRedis() error {
+	return nil
+}