33import itertools
44import logging
55
6- from .. import internal_types
6+ from .. import collectd_types
77from .. import protocol
88from .. import stream
99
1212
1313class Receiver (object ):
1414 def __init__ (
15- self , host , port , log , plugin = internal_types .COLLECTD_PLUGIN_NAME
15+ self , host , port , log , plugin = collectd_types .COLLECTD_PLUGIN_NAME
1616 ):
1717 self .log = log
1818 self .plugin = plugin
19- self .internal_types = [
20- internal_types .pool_internal ,
21- internal_types .totals_internal ,
22- internal_types .process_internal ,
19+ self .collectd_types = [
20+ collectd_types .pool_internal ,
21+ collectd_types .totals_internal ,
22+ collectd_types .process_internal ,
2323 ]
2424 self .network_receiver = protocol .NetworkReceiver (
25- protocol .ServerConnection (host , port , log ), self .internal_types
25+ protocol .ServerConnection (host , port , log ), self .collectd_types
2626 )
27- self .translator = stream .StreamTranslator (* self .internal_types )
28- self .bucket_names = [t .name for t in self .internal_types ]
27+ self .translator = stream .StreamTranslator (* self .collectd_types )
28+ self .bucket_names = [t .name for t in self .collectd_types ]
2929 self .buckets = {
3030 name : stream .TimeBucket () for name in self .bucket_names
3131 }
@@ -35,7 +35,7 @@ def receive(self):
3535 self ._set_stats (values_obj )
3636
3737 def summarize (self , collectd , timestamp ):
38- for type_ in self .internal_types :
38+ for type_ in self .collectd_types :
3939 for values_obj in self .get_stats_by_progname (
4040 type_ .name , timestamp
4141 ):
@@ -70,14 +70,14 @@ def _set_stats(self, values):
7070 # process_token (which is roughly the pid plus a unique key
7171 # generated by the client plugin). we also use a larger interval
7272 # for this value so that the process count changes more slowly
73- process_bucket = self .buckets [internal_types .process_internal .name ]
73+ process_bucket = self .buckets [collectd_types .process_internal .name ]
7474 process_records = process_bucket .get_data (
7575 timestamp , interval = interval * 5
7676 )
7777 process_records [
7878 (hostname , progname , process_token )
7979 ] = values .build (
80- type = internal_types .process_internal .name , values = [1 ]
80+ type = collectd_types .process_internal .name , values = [1 ]
8181 )
8282
8383 def get_stats_by_progname (self , bucket_name , timestamp , agg_func = sum ):
0 commit comments