merona 1.0.0 → 1.0.1

data/lib/channel/channel.rb

@@ -0,0 +1,25 @@
+class Channel
+	def initialize
+		@clients = Array.new
+	end
+	def dispose
+	end
+
+	def subscribe(client)
+		@clients.push name
+	end
+	def unsubscribe(client)
+		@clients.delete name
+		
+		if @clients.size == 0
+			return nil
+		else
+			return true
+		end
+	end
+	def publish(packet)
+		@clients.each do |client|
+			client.send packet
+		end
+	end
+end

data/lib/channel/channel_pool.rb

@@ -0,0 +1,56 @@
+class ChannelPool
+	def initialize
+		@channel = {}
+	end
+	def dispose
+	end
+	
+	def tokenize(name, &block)
+		token = name.split(".")
+		if token.size > 1
+			ch = ""
+			for i in 0..token.size-2
+				ch += token[i] + "."
+			end
+			ch = ch[0..ch.length-2]
+			
+			block.call ch
+		end
+	end
+	
+	def subscribe(name, client)
+		tokenize(name) do |t|
+			subscribe t, client
+		end
+		
+		if @channel[name] == nil
+			@channel[name] = Array.new
+		end
+		
+		@channel[name].push client
+	end
+	def unsubscribe(name, client)
+		tokenize(name) do |t|
+			unsubscribe t, client
+		end
+		
+		return if @channel[name] == nil
+		
+		@channel[name].delete client
+		
+		if @channel[name].size == 0
+			@channel[name] = nil
+		end
+	end
+	def publish(name, packet)
+		tokenize(name) do |t|
+			publish t, packet
+		end
+		
+		return if @channel[name] == nil
+		
+		@channel[name].each do |client|
+			client.send packet
+		end
+	end
+end

data/lib/core/core.rb

@@ -0,0 +1,17 @@
+def polling(&block)
+	EventMachine.run block
+end
+
+def meInitialize
+	
+	Signal.trap("INT")  do
+		Log.output "quit by user"
+		EventMachine.stop
+	end
+  
+	t = Time.new 
+	logfile = "log/#{t.year}.#{t.month}.#{t.day}.log"
+	Log.logfile = File.new(logfile, "a")
+	
+	return true
+end

data/lib/core/version.rb

@@ -0,0 +1,31 @@
+class Version
+    attr_accessor :major, :minor, :fix
+
+    def initialize(major=1,minor=0,fix=0)
+        self.set(major,minor,fix)
+    end
+    def dispose
+    end
+
+    def set(major,minor,fix)
+        @major = major
+        @minof = minor
+        @fix = fix
+    end
+end
+
+def getMajorVersion
+    return 1
+end
+def getMinorVersion
+    return 0
+end
+def getFixVersion
+    return 0
+end
+
+def getVersion
+    ver = Version.new(
+        getMajorVersion,getMinorVersion,getFixVersion)
+    return ver
+end

data/lib/db/database.rb

@@ -0,0 +1,35 @@
+class Database
+	def initialize(addr="localhost", port=9999, id="public", pw="")
+		connect(addr, port , id, pw)
+	end
+	def dispose
+		disconnect
+	end
+	
+	def connect(addr="localhost", port=9999, id="public", pw="")
+	end
+	def disconnect
+	end
+	
+	def prepare(query)
+	end
+	def execute(query)
+	end
+	
+	def select(table, row, condition="", option="")
+		stm = prepare("select #{row} from #{table} #{condition=="" ? "" : "where"} #{condition} #{option};")
+		stm.execute
+	end
+	def update(table, update, condition="", option="")
+		execute("update #{table} set #{update} #{condition=="" ? "" : "where"} #{condition} #{option};")
+	end
+	def insert(table, row, *args)
+		data = ""
+		args.each { |v| data += v.to_s + "," }
+		data = data[0..data.length-2]
+		execute("insert into #{table} (#{row}) values (#{data});")
+	end
+	def delete(table, condition, option="")
+		execute("delete from #{table} #{condition=="" ? "" : "where"} #{condition} #{option};")
+	end
+end

data/lib/db/sqlite.rb

@@ -0,0 +1,24 @@
+require 'sqlite3'
+
+class SQLite < Database
+	def initialize(db)
+		connect db
+	end
+	def dispose
+		disconnect
+	end
+	
+	def connect(addr)
+		@db = SQLite3::Database.new addr
+	end
+	def disconnect
+		@db.close
+	end
+	
+	def prepare(query)
+		@db.prepare query
+	end
+	def execute(query)
+		@db.execute query
+	end
+end

data/lib/debug/log.rb

@@ -0,0 +1,45 @@
+def defLogger(msg)
+    puts msg
+	Log.logfile.puts msg
+end
+
+class Log
+    @@logfunc = method(:defLogger)
+	@@logfile
+	
+    def initialize
+    end
+    def dispose
+    end
+
+	def self.logfile
+		@@logfile
+	end
+	def self.logfile=(fp)
+		@@logfile = fp
+	end
+	
+	def self.logfunc
+		@@logfunc
+	end
+    def self.logfunc=(lf)
+        @@logfunc = lf
+    end
+	
+	private
+	def self.time
+		t = Time.new
+		time = "[#{t.year}:#{t.month}:#{t.day}:#{t.hour}:#{t.min}:#{t.sec}]"
+	end
+	
+	def self.warn(msg)
+		@@logfunc.call time + " [WARN] : " + msg
+	end
+	def self.error(msg)
+		@@logfunc.call time + " [ERROR] : " + msg
+	end
+    def self.output(msg)
+        @@logfunc.call time + " [LOG] : " + msg
+    end
+end
+

data/lib/handler/chat_handler.rb

@@ -0,0 +1,15 @@
+class ChatHandler < Handler
+	def recv(server,connection,packet)
+		case packet.id
+			when Protocol::CHAT
+				puts "[#{packet["sender"]}] : #{packet["msg"]}"
+				
+				reply = Packet.new
+				reply.id = Protocol::CHAT
+				reply["sender"] = packet["sender"]
+				reply["msg"] = packet["msg"]
+				
+				server.pubsub.publish(packet.ch, reply)
+		end
+	end
+end

data/lib/handler/handler.rb

@@ -0,0 +1,13 @@
+class Handler
+	def initialize
+	end
+	def dispose
+	end
+	
+	def recv(server,connection,packet)
+		#case packet.id
+		#	when Protocol::PacketID
+		#		process code here
+		#end
+	end
+end

data/lib/handler/pubsub_handler.rb

@@ -0,0 +1,13 @@
+class PubsubHandler
+	def recv(server,connection,packet)
+		case packet.id
+			when Protocol::SUBSCRIBE
+				server.pubsub.subscribe(packet.ch, connection)
+				puts "sub"
+			when Protocol::UNSUBSCRIBE
+				server.pubsub.unsubscribe(packet.ch, connection)
+			when Protocol::PUBLISH
+			
+		end
+	end
+end

data/lib/handler/sync_handler.rb

@@ -0,0 +1,33 @@
+class SyncHandler < Handler
+	def recv(server,connection,packet)
+		case packet.id
+			when Protocol::SYNC
+				reply = Packet.new
+				reply.id = Protocol::SYNC
+				reply["x"] = packet["x"]
+				reply["y"] = packet["y"]
+				reply["oid"] = packet["oid"]
+				
+				server.pubsub.publish(packet.ch, reply)
+			when Protocol::PUSH_OBJECT
+				reply = Packet.new
+				reply.id = Protocol::OBJECT_ID
+				reply["oid"] = $oid
+				$oid += 1	
+
+				connection.send reply
+				
+				reply.id = Protocol::PUSH_OBJECT
+				reply["x"] = packet["x"]
+				reply["y"] = packet["y"]
+				
+				server.pubsub.publish(packet.ch, reply)
+			when Protocol::DELETE_OBJECT
+				reply = Packet.new
+				reply.id = Protocol::DELETE_OBJECT
+				reply["oid"] = packet["old"]
+				
+				server.pubsub.publish(packet.ch, reply)
+		end
+	end
+end

data/lib/handler/user_handler.rb

@@ -0,0 +1,25 @@
+class UserHandler < Handler
+	def recv(server,connection,packet)
+		case packet.id
+			when Protocol::REGIST
+			when Protocol::UNREGIST
+			when Protocol::LOGIN
+				rs = connection.db.select("account", "*", "id=#{packet["id"]} and pw=#{paacket["pw"]}")
+				row = rs.next
+				
+				reply = Packet.new
+				reply.id = LOGIN_RESULT
+				if row == nil
+					reply["result"] = nil
+				else
+					reply["result"] = true
+					connection.id = packet["id"]
+					connection.mem.push_prefix packet["id"]
+				end
+				
+				rs.close
+				connection.send reply
+			when Protocol::LOGOUT
+		end
+	end
+end

data/lib/merona.rb

@@ -0,0 +1,26 @@
+# Merona
+
+import 'core/*'
+import 'debug/*'
+import 'data/*'
+import 'datastruct/*'
+
+import 'channel/*'
+
+import 'db/database.rb'
+import 'db/*'
+
+import 'sharedmem/shared_memory.rb'
+import 'sharedmem/*'
+
+import 'protocol/*'
+
+import 'packet/packet.rb'
+import 'packet/*'
+
+import 'server/*'
+
+import 'process/*'
+
+import 'handler/handler.rb'
+import 'handler/*'

data/lib/packet/chat_packet.rb

@@ -0,0 +1,26 @@
+class ChatPacket < Packet
+	def initialize
+		super
+		@id = Protocol::CHAT
+	end
+	
+	def msg=(msg)
+		@data["msg"] = msg
+	end
+	def msg
+		@data["msg"]
+	end
+	
+	def sender=(id)
+		@data["sender"] = id
+	end
+	def sender
+		@data["sender"]
+	end
+	def receiver(id)
+		@data["receiver"] = id
+	end
+	def receiver
+		@data["receiver"]
+	end
+end

data/lib/packet/packet.rb

@@ -0,0 +1,51 @@
+class Packet
+	attr_accessor :id
+	attr_accessor :size
+	attr_accessor :data
+	attr_accessor :ch
+	
+	def initialize
+		@data = {}
+		
+		@id = -1
+		@ch = "*"
+	end
+	def dispose
+	end
+	
+	def push(name,data)
+		if name == nil
+			return
+		end
+		
+		@data[name] = data
+	end
+	def get(name)
+		if name == nil
+			return
+		end
+		
+		@data[name]
+	end
+	
+	def []=(name,data)
+		push(name, data)
+	end
+	def [](name)
+		get(name)
+	end
+
+	def ch=(ch)
+		@ch = ch
+	end
+	def ch
+		@ch
+	end
+	
+	def type=(type)
+		@data["type"] = type
+	end
+	def type
+		@data["type"]
+	end
+end

data/lib/process/process_pool.rb

@@ -0,0 +1,46 @@
+class WorkItem
+	attr_accessor :packet
+	attr_accessor :sender
+	
+	def initialize(sender, packet)
+		@packet = packet
+		@sender = sender
+	end
+end
+
+class ProcessPool
+	def initialize(server)
+		@server = server
+		@worker = Array.new
+		
+		@queue = Queue.new
+	end
+	def dispose
+	end
+	
+	def create_worker(n)
+		n.times do
+			thread = Thread.new do
+				while true
+					item = @queue.deq(false)
+
+					@server.handler.each do |handler|
+						handler.recv(@server, item.sender, item.packet)
+					end
+				end
+				
+				@worker.push thread
+			end
+		end
+	end
+	def kill
+		@worker.each do |worker|
+			worker.kill
+		end
+	end
+	
+	def enqueue(sender, item)
+		@queue.push WorkItem.new(sender, item)
+	end
+	
+end

data/lib/protocol/chat_protocol.rb

@@ -0,0 +1,3 @@
+class Protocol
+	CHAT = 1300
+end

data/lib/protocol/protocol.rb

@@ -0,0 +1,5 @@
+class Protocol
+	FIRST = 0
+	
+	LAST = 99999
+end

data/lib/protocol/pubsub_protocol.rb

@@ -0,0 +1,7 @@
+# 9000 ~ 9099
+class Protocol
+	SUBSCRIBE = 9000
+	UNSUBSCRIBE = 9001
+	
+	PUBLISH = 9010
+end

data/lib/protocol/sync_protocol.rb

@@ -0,0 +1,9 @@
+# 1400 ~ 1499
+class Protocol
+	SYNC = 1400
+	
+	PUSH_OBJECT = 1410
+	OBJECT_ID = 1411
+	
+	DELETE_OBJECT = 1420
+end

data/lib/protocol/user_protocol.rb

@@ -0,0 +1,14 @@
+# 1100 ~ 1199
+class Protocol
+	REGIST = 1100
+	REGIST_RESULT = 1101
+	
+	UNREGIST = 1110
+	UNREGIST_RESULT = 1111
+	
+	LOGIN = 1120
+	LOGIN_RESULT = 1121
+	
+	LOGOUT = 1130
+	LOGOUT_RESULT = 1131
+end

data/lib/server/connection.rb

@@ -0,0 +1,43 @@
+require 'eventmachine'
+
+class Connection < EM::Connection
+	include EM::P::ObjectProtocol
+	
+	attr_accessor :id
+	attr_reader :ip, :port
+	
+	def initialize(*args)
+		@port, @ip = Socket.unpack_sockaddr_in(get_peername)
+		@server = args[0]
+		@server.clients.push self
+		
+		@db = DB.new DB_ACCOUNT
+		@mem = SHAREDMEM.new SHAREDMEM_HOST, SHAREDMEM_PORT
+		
+		connect
+		
+		set_sock_opt(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, CONNECTION_NODELAY)
+	end
+	def unbind
+		@server.clients.delete self
+		@db.dispose
+		@mem.dispose
+		
+		disconnect
+	end
+	
+	def send(packet)
+		send_object packet
+	end
+	def receive_object(obj)
+		p obj
+		@server.process.enqueue(self, obj)
+	end
+	
+	def connect
+		Log.output("new connection from " + @ip)
+	end
+	def disconnect
+		Log.output("lost connection " + @ip)
+	end
+end

data/lib/server/server.rb

@@ -0,0 +1,49 @@
+$server = {}
+
+class Server
+	attr_reader :name
+	attr_reader :port
+	attr_reader :handler, :process
+	attr_reader :pubsub
+	
+	attr_accessor :clients
+	
+	def initialize(name,port,klass=Connection)
+		EventMachine.start_server("localhost", port, klass, self)
+
+		@clients = Array.new
+		@handler = Array.new
+		
+		@pubsub = ChannelPool.new
+		@process = ProcessPool.new(self)
+		@process.create_worker SERVER_WORKERS
+		
+		@name = name
+		@port = port
+		
+		$server[@name] = self
+	end
+	def dispose
+		$server.delete @name
+	end
+	
+	def add_handler(handler)
+		@handler.push handler.new
+	end
+	def delete_handler(handler)
+		@handler.each do |value|
+			if value.class == handler
+				@handler.delete value
+			end
+		end
+	end
+	
+	def exclude(connection)
+		@clients.reject { |c| connection == c }
+	end
+	def broadcast(packet, exclusion = nil)
+		(exclusion ? exclude(exclusion) : @clients).each do |client|
+			client.send packet
+		end
+	end
+end

data/lib/sharedmem/redis.rb

@@ -0,0 +1,16 @@
+require 'redis'
+
+class RedisMemory < SharedMemory
+	def initialize(host="localhost", port=6679)
+		@redis = Redis.new(:host => host, :port => port)
+	end
+	def dispose
+	end
+	
+	def get(name)
+		@redis.get name
+	end
+	def set(name, value)
+		@redis.set name, value
+	end
+end

data/lib/sharedmem/shared_memory.rb

@@ -0,0 +1,25 @@
+class SharedMemory
+	attr_accessor :prefix
+	
+	def initialize
+		@prefix = ""
+	end
+	def dispose
+	end
+	
+	def push_prefix(prefix)
+		@prefix += prefix
+	end
+	
+	def get(name)
+	end
+	def set(name, value)
+	end
+	
+	def [](name)
+		get @prefix+name
+	end
+	def []=(name,value)
+		set @prefix+name, value
+	end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: merona
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
   prerelease: 
 platform: ruby
 authors:
@@ -65,10 +65,34 @@ email:
 executables: []
 extensions: []
 extra_rdoc_files: []
-files: []
+files:
+- lib/channel/channel.rb
+- lib/channel/channel_pool.rb
+- lib/core/core.rb
+- lib/core/version.rb
+- lib/db/database.rb
+- lib/db/sqlite.rb
+- lib/debug/log.rb
+- lib/handler/chat_handler.rb
+- lib/handler/handler.rb
+- lib/handler/pubsub_handler.rb
+- lib/handler/sync_handler.rb
+- lib/handler/user_handler.rb
+- lib/merona.rb
+- lib/packet/chat_packet.rb
+- lib/packet/packet.rb
+- lib/process/process_pool.rb
+- lib/protocol/chat_protocol.rb
+- lib/protocol/protocol.rb
+- lib/protocol/pubsub_protocol.rb
+- lib/protocol/sync_protocol.rb
+- lib/protocol/user_protocol.rb
+- lib/server/connection.rb
+- lib/server/server.rb
+- lib/sharedmem/redis.rb
+- lib/sharedmem/shared_memory.rb
 homepage: http://github.com/pjc0247/Merona
-licenses:
-- MIT
+licenses: []
 post_install_message: 
 rdoc_options: []
 require_paths: