away
4 years ago
parent
75d308de37
commit
1e4a1dffdd
@ -1,72 +1,60 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"Open_IM/src/common/config"
|
||||
"github.com/garyburd/redigo/redis"
|
||||
"gopkg.in/mgo.v2"
|
||||
"time"
|
||||
)
|
||||
|
||||
var DB DataBases
|
||||
|
||||
type DataBases struct {
|
||||
MgoDB mongoDB
|
||||
RedisDB redisDB
|
||||
MysqlDB mysqlDB
|
||||
MysqlDB mysqlDB
|
||||
mgoSession *mgo.Session
|
||||
redisPool *redis.Pool
|
||||
}
|
||||
|
||||
func key(dbAddress, dbName string) string {
|
||||
return dbAddress + "_" + dbName
|
||||
}
|
||||
|
||||
//type Config struct {
|
||||
// Mongo struct {
|
||||
// DBAddress []string `yaml:"dbAddress"`
|
||||
// DBDirect bool `yaml:"dbDirect"`
|
||||
// DBTimeout int `yaml:"dbTimeout"`
|
||||
// DBDatabase []string `yaml:"dbDatabase"`
|
||||
// DBSource string `yaml:"dbSource"`
|
||||
// DBUserName string `yaml:"dbUserName"`
|
||||
// DBPassword string `yaml:"dbPassword"`
|
||||
// DBMaxPoolSize int `yaml:"dbMaxPoolSize"`
|
||||
// }
|
||||
// Mysql struct {
|
||||
// DBAddress []string `yaml:"dbAddress"`
|
||||
// DBPort int `yaml:"dbPort"`
|
||||
// DBUserName string `yaml:"dbUserName"`
|
||||
// DBPassword string `yaml:"dbPassword"`
|
||||
// DBDatabaseName string `yaml:"dbChatName"` // 默认使用DBAddress[0]
|
||||
// DBTableName string `yaml:"dbMsgName"`
|
||||
// DBMsgTableNum int `yaml:"dbMsgTableNum"`
|
||||
// DBCharset string `yaml:"dbCharset"`
|
||||
// DBMaxOpenConns int `yaml:"dbMaxOpenConns"`
|
||||
// DBMaxIdleConns int `yaml:"dbMaxIdleConns"`
|
||||
// DBMaxLifeTime int `yaml:"dbMaxLifeTime"`
|
||||
// }
|
||||
// Redis struct {
|
||||
// DBAddress string `yaml:"dbAddress"`
|
||||
// DBPort int `yaml:"dbPort"`
|
||||
// DBMaxIdle int `yaml:"dbMaxIdle"`
|
||||
// DBMaxActive int `yaml:"dbMaxActive"`
|
||||
// DBIdleTimeout int `yaml:"dbIdleTimeout"`
|
||||
// }
|
||||
//}
|
||||
|
||||
//func init() {
|
||||
// bytes, err := ioutil.ReadFile("config/db.yaml")
|
||||
// if err != nil {
|
||||
// log.Error("", "", "read db.yaml config fail! err = %s", err.Error())
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// if err = yaml.Unmarshal(bytes, &DB.Config); err != nil {
|
||||
// log.Error("", "", "unmarshal db.yaml config fail! err = %s", err.Error())
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// DB.RedisDB.newPool(DB.Config)
|
||||
// //DB.MysqlDB.sqlxDB(DB.Config.Mysql.DBName[0], DB.Config)
|
||||
//}
|
||||
func init() {
|
||||
DB.RedisDB.newPool()
|
||||
}
|
||||
func (d *DataBases) session(dbName string) *mgo.Session {
|
||||
return d.MgoDB.mgoSession(dbName)
|
||||
//mysql init
|
||||
|
||||
// mongo init
|
||||
mgoDailInfo := &mgo.DialInfo{
|
||||
Addrs: config.Config.Mongo.DBAddress,
|
||||
Direct: config.Config.Mongo.DBDirect,
|
||||
Timeout: time.Second * time.Duration(config.Config.Mongo.DBTimeout),
|
||||
Database: config.Config.Mongo.DBDatabase,
|
||||
Source: config.Config.Mongo.DBSource,
|
||||
Username: config.Config.Mongo.DBUserName,
|
||||
Password: config.Config.Mongo.DBPassword,
|
||||
PoolLimit: config.Config.Mongo.DBMaxPoolSize,
|
||||
}
|
||||
mgoSession, err := mgo.DialWithInfo(mgoDailInfo)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
DB.mgoSession = mgoSession
|
||||
DB.mgoSession.SetMode(mgo.Monotonic, true)
|
||||
|
||||
// redis pool init
|
||||
DB.redisPool = &redis.Pool{
|
||||
MaxIdle: config.Config.Redis.DBMaxIdle,
|
||||
MaxActive: config.Config.Redis.DBMaxActive,
|
||||
IdleTimeout: time.Duration(config.Config.Redis.DBIdleTimeout) * time.Second,
|
||||
Dial: func() (redis.Conn, error) {
|
||||
return redis.Dial(
|
||||
"tcp",
|
||||
config.Config.Redis.DBAddress,
|
||||
redis.DialReadTimeout(time.Duration(1000)*time.Millisecond),
|
||||
redis.DialWriteTimeout(time.Duration(1000)*time.Millisecond),
|
||||
redis.DialConnectTimeout(time.Duration(1000)*time.Millisecond),
|
||||
redis.DialDatabase(0),
|
||||
redis.DialPassword(config.Config.Redis.DBPassWord),
|
||||
)
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
@ -0,0 +1,62 @@
|
||||
package im_mysql_model
|
||||
|
||||
import (
|
||||
"Open_IM/src/common/db"
|
||||
"time"
|
||||
)
|
||||
|
||||
func InsertIntoGroupRequest(groupId, fromUserId, toUserId, reqMsg, fromUserNickName, fromUserFaceUrl string) error {
|
||||
dbConn, err := db.DB.MysqlDB.DefaultGormDB()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
toInsertInfo := GroupRequest{GroupID: groupId, FromUserID: fromUserId, ToUserID: toUserId, ReqMsg: reqMsg, FromUserNickname: fromUserNickName, FromUserFaceUrl: fromUserFaceUrl, CreateTime: time.Now()}
|
||||
err = dbConn.Table("group_request").Create(toInsertInfo).Error
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func FindGroupRequestUserInfoByUidAndGroupID(groupId, uid string) (*GroupRequest, error) {
|
||||
dbConn, err := db.DB.MysqlDB.DefaultGormDB()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var requestUserInfo GroupRequest
|
||||
err = dbConn.Table("group_request").Where("from_user_id=? and group_id=?", uid, groupId).Find(&requestUserInfo).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &requestUserInfo, nil
|
||||
}
|
||||
|
||||
func DelGroupRequest(groupId, fromUserId, toUserId string) error {
|
||||
dbConn, err := db.DB.MysqlDB.DefaultGormDB()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
err = dbConn.Exec("delete from group_request where group_id=? and from_user_id=? and to_user_id=?", groupId, fromUserId, toUserId).Error
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func FindGroupBeInvitedRequestInfoByUidAndGroupID(groupId, uid string) (*GroupRequest, error) {
|
||||
dbConn, err := db.DB.MysqlDB.DefaultGormDB()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var beInvitedRequestUserInfo GroupRequest
|
||||
err = dbConn.Table("group_request").Where("to_user_id=? and group_id=?", uid, groupId).Find(&beInvitedRequestUserInfo).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &beInvitedRequestUserInfo, nil
|
||||
|
||||
}
|
||||
|
||||
func InsertGroupRequest(groupId, fromUser, fromUserNickName, fromUserFaceUrl, toUser, requestMsg, handledMsg string, handleStatus int) error {
|
||||
return nil
|
||||
}
|
||||
Loading…
Reference in new issue