package GoLog

import (
	"fmt"
	"gopkg.in/mgo.v2"
)

// 	+---------------------------------------------------------+
// 	+                                                         +
//  +                        MongoLog 					      +
// 	+                                                         +
// 	+---------------------------------------------------------+
// 使用mongo
type MongoLog struct {
	log     *BufferLog
	session *mgo.Session
}

func newSession(host string, port int, authdb, user, pwd string, auth bool) (*mgo.Session, error) {
	var err error
	var url string
	if auth {
		url = fmt.Sprintf("mongodb://%s:%s@%s:%d/%s", user, pwd, host, port, authdb)

	} else {
		url = fmt.Sprintf("%s:%d", host, port)
	}
	session, err := mgo.Dial(url)
	if err != nil {
		return nil, err
	}
	session.SetMode(mgo.Eventual, true)
	return session, nil
}

func NewMongoLog(
	host string,
	port int,
	dbname string,
	mongodbAuthDB string,
	mongodbUser string,
	mongodbPwd string,
	mongodbAuth bool,
	mongodbCollection string,
) (*MongoLog, error) {

	session, err := newSession(host, port, mongodbAuthDB, mongodbUser, mongodbPwd, mongodbAuth)
	if err != nil {
		return nil, err
	}

	c := session.DB(dbname).C(mongodbCollection)

	writer := func(msg IMsg) {
		r := msg.ToMap()
		_ = c.Insert(r)
	}

	log := NewBufferLog(50, writer)
	return &MongoLog{log: log, session: session}, nil
}

func (c *MongoLog) Write(msg IMsg) {
	c.log.Write(msg)
}

func (c *MongoLog) Stop() {
	c.log.Stop()
	c.session.Close()
}