From 29d98731c616ffc5fca5286d4c38f6c7a88e3fe9 Mon Sep 17 00:00:00 2001
From: Chen Liang <chen.liang@crenolab.com>
Date: Tue, 25 Nov 2014 14:41:51 -0800
Subject: [PATCH] add sess_ledis select db config

---
 session/ledis/ledis_session.go | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/session/ledis/ledis_session.go b/session/ledis/ledis_session.go
index 3ada47ac..400d7238 100644
--- a/session/ledis/ledis_session.go
+++ b/session/ledis/ledis_session.go
@@ -2,6 +2,8 @@ package session
 
 import (
 	"net/http"
+	"strconv"
+	"strings"
 	"sync"
 
 	"github.com/astaxie/beego/session"
@@ -74,19 +76,29 @@ func (ls *LedisSessionStore) SessionRelease(w http.ResponseWriter) {
 type LedisProvider struct {
 	maxlifetime int64
 	savePath    string
+	db          int
 }
 
 // init ledis session
 // savepath like ledis server saveDataPath,pool size
 // e.g. 127.0.0.1:6379,100,astaxie
 func (lp *LedisProvider) SessionInit(maxlifetime int64, savePath string) error {
+	var err error
 	lp.maxlifetime = maxlifetime
-	lp.savePath = savePath
+	configs := strings.Split(savepath, ",")
+	if len(configs) == 1 {
+		lp.savePath = configs[0]
+	} else if len(configs) == 2 {
+		lp.savePath = configs[0]
+		lp.db, err = strconv.Atoi(configs[1])
+		if err != nil {
+			return err
+		}
+	}
 	cfg := new(config.Config)
 	cfg.DataDir = lp.savePath
-	var err error
 	nowLedis, err := ledis.Open(cfg)
-	c, err = nowLedis.Select(0)
+	c, err = nowLedis.Select(lp.db)
 	if err != nil {
 		println(err)
 		return nil