away
4 years ago
parent
75d308de37
commit
1e4a1dffdd
@ -1,72 +1,60 @@
|
|||||||
package db
|
package db
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"Open_IM/src/common/config"
|
||||||
|
"github.com/garyburd/redigo/redis"
|
||||||
"gopkg.in/mgo.v2"
|
"gopkg.in/mgo.v2"
|
||||||
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
var DB DataBases
|
var DB DataBases
|
||||||
|
|
||||||
type DataBases struct {
|
type DataBases struct {
|
||||||
MgoDB mongoDB
|
MysqlDB mysqlDB
|
||||||
RedisDB redisDB
|
mgoSession *mgo.Session
|
||||||
MysqlDB mysqlDB
|
redisPool *redis.Pool
|
||||||
}
|
}
|
||||||
|
|
||||||
func key(dbAddress, dbName string) string {
|
func key(dbAddress, dbName string) string {
|
||||||
return dbAddress + "_" + dbName
|
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() {
|
func init() {
|
||||||
DB.RedisDB.newPool()
|
//mysql init
|
||||||
}
|
|
||||||
func (d *DataBases) session(dbName string) *mgo.Session {
|
// mongo init
|
||||||
return d.MgoDB.mgoSession(dbName)
|
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