nodejs 서버와 mongojs

저번에 말한 nodeJS로 간단한 Chat Server를 만들었다.

추가적으로 mongojs를 사용하여 Chat Log를 저장하고 최초 접속시 최근 대화기록을 보여주는 기능을 넣었다.

아래는 server.js 이다.

> nodejs와 mongodb 설치는 생략한다.

var io = require('socket.io').listen(443); var db = require('mongojs').connect('member', ['info']); insert = function(data){ db.info.save( data,function(error,result){ if(!error) ;//console.log("insert succeed!"); }); } find = function(callback){ //loading '10' array data from mongodb db.info.find().limit(10).sort({"time":-1}).toArray( function(error,result){ if(!error){ var obj = result; callback(obj); } }); } io.sockets.on('connection', function(socket) { var remoteAddress = socket.client.conn.remoteAddress; console.log("connect IP "+remoteAddress); socket.emit('connection', { type : 'connected' }); socket.on('connection', function(data) { if(data.type == 'join') { socket.join(data.room); addRoom(data.room, data.name, socket.id); find(function(obj){ for(var i=0;i<obj.length;i++) { var _obj = obj[i]; socket.emit('message', { name : _obj.name, message : _obj.message }); } }); socket.emit('system', { type : 0, message : '채팅방 접속하셨습니다.', style : '#ff0000' }); socket.broadcast.to(data.room).emit('system', { type : 1, content : data.name, message : data.name + ' 유저가 접속하셨습니다.', style : '#ff00ff' }); socket.emit('system', { type : 4, content : getData(), message : '정보 업데이트', style : '#ff00ff' }); } }); socket.on('user', function(data) { var log = {time:(new Date()).yyyymmdd(),ip:remoteAddress,name:data.name,message:data.message}; insert(log); var info = searchRoom(socket.id, data.name); if(info.str != "") {socket.broadcast.to(info.room).emit('system', {type : 2, content: {pre:info.pre,aft:info.aft}, message:info.str,style : '#ffff00'}); socket.emit('system', {type : 2, content: {pre:info.pre,aft:info.aft}, message:info.str,style : '#ffff00'});} socket.broadcast.to(info.room).emit('message', data); }); socket.on('disconnect', function() { console.log("out IP "+remoteAddress); var info = leaveRoom(socket.id); socket.leave(info.room); socket.broadcast.to(info.room).emit('system', { type : 3, content : info.name, message : info.name+'가 나가셨습니다.', style : '#ff0000' }); }); function addRoom(room_id, name_id, socket_id) { room.push( { name:room_id, id_name:name_id, id:socket_id }); } function searchRoom(socket_id, name_id) { for(var i=0;i<room.length;i++) if(room[i].id == socket_id){ var info = ""; var preName = ""; if(room[i].id_name != name_id) { preName = room[i].id_name; info = room[i].id_name+"이 "+name_id+"로 이름을 바꾸셧습니다."; room[i].id_name = name_id; } return {pre: preName, aft: name_id, str: info, room: room[i].name}; } } function leaveRoom(socket_id) { var index; var room_num; var name_id; for(var i=0;i<room.length;i++) if(room[i].id == socket_id) { index = i; room_num = room[i].name; name_id = room[i].id_name; } room.splice(index, 1); return {room:room_num,name:name_id}; } function getData() { var arr = new Array(); for(var i=0;i<room.length;i++) arr.push(room[i].id_name); return arr; } }); var room = new Array(); Date.prototype.yyyymmdd = function() { var yyyy = this.getFullYear().toString(); var mm = (this.getMonth() + 1).toString(); var dd = this.getDate().toString(); return yyyy + (mm[1] ? mm : '0'+mm[0]) + (dd[1] ? dd : '0'+dd[0]); }