* Remove useless subscriber
* Fix typo in the config file
* Update Helper accordingly
This commit is contained in:
Raphaël Vinot 2014-08-26 17:36:57 +02:00
parent d7a7da4814
commit 623e876f3b
22 changed files with 10 additions and 45 deletions

View file

@ -31,6 +31,7 @@ class Redis_Queues(object):
self.subscriber_name = subscriber_name self.subscriber_name = subscriber_name
self.sub_channel = self.config.get(conf_section, conf_channel) self.sub_channel = self.config.get(conf_section, conf_channel)
self.sub_address = self.config.get(conf_section, 'address')
self.redis_channel = self.sub_channel + self.subscriber_name self.redis_channel = self.sub_channel + self.subscriber_name
# Redis Queue # Redis Queue
@ -40,11 +41,10 @@ class Redis_Queues(object):
port=self.config.getint(config_section, "port"), port=self.config.getint(config_section, "port"),
db=self.config.getint(config_section, "db")) db=self.config.getint(config_section, "db"))
def zmq_sub(self, conf_section): def zmq_sub(self):
sub_address = self.config.get(conf_section, 'adress')
context = zmq.Context() context = zmq.Context()
self.sub_socket = context.socket(zmq.SUB) self.sub_socket = context.socket(zmq.SUB)
self.sub_socket.connect(sub_address) self.sub_socket.connect(self.sub_address)
self.sub_socket.setsockopt(zmq.SUBSCRIBE, self.sub_channel) self.sub_socket.setsockopt(zmq.SUBSCRIBE, self.sub_channel)
def zmq_pub(self, config_section, config_channel): def zmq_pub(self, config_section, config_channel):
@ -74,6 +74,7 @@ class Redis_Queues(object):
return self.r_queues.srem('SHUTDOWN_FLAGS', flag) return self.r_queues.srem('SHUTDOWN_FLAGS', flag)
def redis_queue_subscribe(self, publisher): def redis_queue_subscribe(self, publisher):
self.zmq_sub()
publisher.info("Suscribed to channel {}".format(self.sub_channel)) publisher.info("Suscribed to channel {}".format(self.sub_channel))
while True: while True:
msg = self.sub_socket.recv() msg = self.sub_socket.recv()

View file

@ -34,5 +34,4 @@ if __name__ == "__main__":
subscriber_name = 'feed' subscriber_name = 'feed'
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
h.zmq_sub(config_section)
h.redis_queue_subscribe(publisher) h.redis_queue_subscribe(publisher)

View file

@ -32,5 +32,4 @@ if __name__ == "__main__":
subscriber_name = 'categ' subscriber_name = 'categ'
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
h.zmq_sub(config_section)
h.redis_queue_subscribe(publisher) h.redis_queue_subscribe(publisher)

View file

@ -48,9 +48,6 @@ if __name__ == "__main__":
pub_config_section = 'PubSub_Longlines' pub_config_section = 'PubSub_Longlines'
h.zmq_pub(pub_config_section, None) h.zmq_pub(pub_config_section, None)
# Subscriber
h.zmq_sub(config_section)
# SCRIPT PARSER # # SCRIPT PARSER #
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
description='''This script is a part of the Analysis Information \ description='''This script is a part of the Analysis Information \

View file

@ -31,5 +31,4 @@ if __name__ == "__main__":
subscriber_name = 'line' subscriber_name = 'line'
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
h.zmq_sub(config_section)
h.redis_queue_subscribe(publisher) h.redis_queue_subscribe(publisher)

View file

@ -32,5 +32,4 @@ if __name__ == "__main__":
subscriber_name = 'tokenize' subscriber_name = 'tokenize'
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
h.zmq_sub(config_section)
h.redis_queue_subscribe(publisher) h.redis_queue_subscribe(publisher)

View file

@ -42,9 +42,6 @@ if __name__ == "__main__":
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
# Subscriber
h.zmq_sub(config_section)
# FUNCTIONS # # FUNCTIONS #
publisher.info("""ZMQ Attribute is Running""") publisher.info("""ZMQ Attribute is Running""")

View file

@ -32,5 +32,4 @@ if __name__ == "__main__":
subscriber_name = 'attributes' subscriber_name = 'attributes'
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
h.zmq_sub(config_section)
h.redis_queue_subscribe(publisher) h.redis_queue_subscribe(publisher)

View file

@ -18,9 +18,6 @@ if __name__ == "__main__":
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
# Subscriber
h.zmq_sub(config_section)
# FUNCTIONS # # FUNCTIONS #
publisher.info("Creditcard script subscribed to channel creditcard_categ") publisher.info("Creditcard script subscribed to channel creditcard_categ")

View file

@ -15,5 +15,4 @@ if __name__ == "__main__":
subscriber_name = 'cards' subscriber_name = 'cards'
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
h.zmq_sub(config_section)
h.redis_queue_subscribe(publisher) h.redis_queue_subscribe(publisher)

View file

@ -41,9 +41,6 @@ if __name__ == "__main__":
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
# Subscriber
h.zmq_sub(config_section)
# REDIS # # REDIS #
r_serv1 = redis.StrictRedis( r_serv1 = redis.StrictRedis(
host=h.config.get("Redis_Level_DB", "host"), host=h.config.get("Redis_Level_DB", "host"),

View file

@ -32,5 +32,4 @@ if __name__ == "__main__":
subscriber_name = 'curve' subscriber_name = 'curve'
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
h.zmq_sub(config_section)
h.redis_queue_subscribe(publisher) h.redis_queue_subscribe(publisher)

View file

@ -31,9 +31,6 @@ if __name__ == "__main__":
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
# Subscriber
h.zmq_sub(config_section)
# REDIS # # REDIS #
# DB OBJECT & HASHS ( DISK ) # DB OBJECT & HASHS ( DISK )
# FIXME increase flexibility # FIXME increase flexibility

View file

@ -14,5 +14,4 @@ if __name__ == "__main__":
subscriber_name = 'duplicate' subscriber_name = 'duplicate'
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
h.zmq_sub(config_section)
h.redis_queue_subscribe(publisher) h.redis_queue_subscribe(publisher)

View file

@ -26,5 +26,4 @@ if __name__ == "__main__":
subscriber_name = 'indexer' subscriber_name = 'indexer'
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
h.zmq_sub(config_section)
h.redis_queue_subscribe(publisher) h.redis_queue_subscribe(publisher)

View file

@ -21,9 +21,6 @@ if __name__ == "__main__":
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
# Subscriber
h.zmq_sub(config_section)
# REDIS # # REDIS #
r_serv2 = redis.StrictRedis( r_serv2 = redis.StrictRedis(
host=h.config.get("Redis_Cache", "host"), host=h.config.get("Redis_Cache", "host"),

View file

@ -14,5 +14,4 @@ if __name__ == "__main__":
subscriber_name = 'emails' subscriber_name = 'emails'
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
h.zmq_sub(config_section)
h.redis_queue_subscribe(publisher) h.redis_queue_subscribe(publisher)

View file

@ -39,9 +39,6 @@ if __name__ == "__main__":
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
# Subscriber
h.zmq_sub(config_section)
# FUNCTIONS # # FUNCTIONS #
publisher.info("Script subscribed to channel onion_categ") publisher.info("Script subscribed to channel onion_categ")

View file

@ -31,5 +31,4 @@ if __name__ == "__main__":
subscriber_name = 'tor' subscriber_name = 'tor'
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
h.zmq_sub(config_section)
h.redis_queue_subscribe(publisher) h.redis_queue_subscribe(publisher)

View file

@ -31,9 +31,6 @@ if __name__ == "__main__":
pub_config_channel = 'channel' pub_config_channel = 'channel'
h.zmq_pub(pub_config_section, pub_config_channel) h.zmq_pub(pub_config_section, pub_config_channel)
# Subscriber
h.zmq_sub(config_section)
# REDIS # # REDIS #
r_serv2 = redis.StrictRedis( r_serv2 = redis.StrictRedis(
host=h.config.get("Redis_Cache", "host"), host=h.config.get("Redis_Cache", "host"),

View file

@ -15,5 +15,4 @@ if __name__ == "__main__":
subscriber_name = 'urls' subscriber_name = 'urls'
h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) h = Helper.Redis_Queues(config_section, config_channel, subscriber_name)
h.zmq_sub(config_section)
h.redis_queue_subscribe(publisher) h.redis_queue_subscribe(publisher)

View file

@ -38,31 +38,31 @@ db = 1
# PUB / SUB : ZMQ # PUB / SUB : ZMQ
[Feed] [Feed]
adress = tcp://crf.circl.lu:5556 address = tcp://crf.circl.lu:5556
topicfilter = 102 topicfilter = 102
[PubSub_Global] [PubSub_Global]
adress = tcp://127.0.0.1:5000 address = tcp://127.0.0.1:5000
channel = filelist channel = filelist
[PubSub_Longlines] [PubSub_Longlines]
adress = tcp://127.0.0.1:5001 address = tcp://127.0.0.1:5001
channel_0 = Longlines channel_0 = Longlines
channel_1 = Shortlines channel_1 = Shortlines
[PubSub_Words] [PubSub_Words]
adress = tcp://127.0.0.1:5002 address = tcp://127.0.0.1:5002
channel_0 = words channel_0 = words
[PubSub_Categ] [PubSub_Categ]
adress = tcp://127.0.0.1:5003 address = tcp://127.0.0.1:5003
channel_0 = creditcard_categ channel_0 = creditcard_categ
channel_1 = mails_categ channel_1 = mails_categ
channel_2 = onion_categ channel_2 = onion_categ
channel_3 = web_categ channel_3 = web_categ
[PubSub_Url] [PubSub_Url]
adress = tcp://127.0.0.1:5004 address = tcp://127.0.0.1:5004
channel = urls channel = urls
# country code logged as critical # country code logged as critical
cc_critical = DE cc_critical = DE