Example #1
0
func NewMsgRepository(db *sql.DB) *MsgRepository {
	mr := &MsgRepository{}
	mr.db = db
	mr.urIDs = make(map[types.ID]types.ID, 10240)
	gsql.MustExec(mr.db, `create table if not exists read_msgs(
	user_id bigint primary key,
	msg_id bigint not null default 0
	)`)

	for i := 0; i < userMsgTableCount; i++ {
		mr.CreateUserMsgTable(types.ID(i))
	}

	for i := 0; i < groupMsgTableCount; i++ {
		mr.CreateGroupMsgTable(types.ID(i))
	}

	var rows *sql.Rows
	rows, err := mr.db.Query("select user_id, msg_id from read_msgs")
	if err != nil {
		log.Panic(err)
	}
	defer rows.Close()
	var uid, msgID types.ID
	for rows.Next() {
		err = rows.Scan(&uid, &msgID)
		if err != nil {
			panic(err)
		}
		mr.urIDs[uid] = msgID
	}
	return mr
}
Example #2
0
func (rs *AccountsService) AutoLogin(c context.Context, req *pb.AutoLoginRequest) (*pb.LoginResponse, error) {
	return rs.handleLoginResult(services.AutoLogin(c, types.ID(req.AccountId), req.LoginToken))
}