博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
Python Elasticsearch api
阅读量:5162 次
发布时间:2019-06-13

本文共 23083 字,大约阅读时间需要 76 分钟。

一、介绍

ElasticSearch是一个基于Lucene的搜索服务器。它提供了一个分布式多用户能力的全文搜索引擎,基于RESTful web接口。下面介绍了利用Python API接口进行数据查询,方便其他系统的调用。

安装API

pip3 install elasticsearch

 

建立es连接

无用户名密码状态

from elasticsearch import Elasticsearches = Elasticsearch([{'host':'10.10.13.12','port':9200}])

 

默认的超时时间是10秒,如果数据量很大,时间设置更长一些。如果端口是9200,直接写IP即可。代码如下:

es = Elasticsearch(['10.10.13.12'], timeout=3600)

 

用户名密码状态

如果Elasticsearch开启了验证,需要用户名和密码

es = Elasticsearch(['10.10.13.12'], http_auth=('xiao', '123456'), timeout=3600)

 

数据检索功能

es.search(index='logstash-2015.08.20', q='http_status_code:5* AND server_name:"web1"', from_='124119')

 

常用参数
  • index - 索引名
  • q - 查询指定匹配 使用Lucene查询语法
  • from_ - 查询起始点  默认0
  • doc_type - 文档类型
  • size - 指定查询条数 默认10
  • field - 指定字段 逗号分隔
  • sort - 排序  字段:asc/desc
  • body - 使用Query DSL
  • scroll - 滚动查询

 

统计查询功能

语法同search大致一样,但只输出统计值

es.count(index='logstash-2015.08.21', q='http_status_code:500')

输出:

{
'_shards':{
'failed':0, 'successful':5, 'total':5}, 'count':17042}

 

17042 就是统计值!

 

知识扩展

滚动demo

# Initialize the scrollpage = es.search(    index ='yourIndex',    doc_type ='yourType',    scroll ='2m',    search_type ='scan',    size =1000,    body ={    # Your query's body}) sid = page['_scroll_id']scroll_size = page['hits']['total'] # Start scrollingwhile(scroll_size >0):    print "Scrolling..."    page = es.scroll(scroll_id = sid, scroll ='2m')    # Update the scroll ID    sid = page['_scroll_id']    # Get the number of results that we returned in the last scroll    scroll_size = len(page['hits']['hits'])    print "scroll size: "+ str(scroll_size)    # Do something with the obtained page
View Code

 

以上demo实现了一次取若干数据,数据取完之后结束,不会获取到最新更新的数据。我们滚动完之后想获取最新数据怎么办?滚动的时候会有一个统计值,如total: 5。跳出循环之后,我们可以用_from参数定位到5开始滚动之后的数据。

 

但是我用的不是这个,用的是以下方法,链接如下:

 

在下面的内容中,我会详细介绍此代码如何使用!

 

二、Query DSL

range过滤器查询范围

gt: > 大于lt: < 小于gte: >= 大于或等于lte: <= 小于或等于

 

示例代码1

"range":{    "money":{        "gt":20,        "lt":40    }}

 

时间范围

最近时间段

比如我要查询最近1分钟的

"range": {    '@timestamp': {
'gt': 'now-1m'}}

 

最新1小时

"range": {    '@timestamp': {
'gt': 'now-1h'}}

 

最新1天的

"range": {    '@timestamp': {
'gt': 'now-1d'}}

 

指定时间段

那么问题来了,它是根据当前时间来计算最近的时间。但是有些情况下,我需要制定时间范围,精确到分钟

假设需要查询早上8点到9点的数据,可以这样

"range": {    '@timestamp': {        "gt" : "{}T{}:00:00".format("2018-12-17","08"),        "lt": "{}T{}:59:59".format("2018-12-17","09"),        "time_zone": "Asia/Shanghai"    }}

 

注意:日期和小时之间,有一个字母T来间隔。不能用空格!

time_zone 表示时区,如果默认的时区不会,可能会影响查询结果!

 

bool组合过滤器

must:所有分句都必须匹配,与 AND 相同。must_not:所有分句都必须不匹配,与 NOT 相同。should:至少有一个分句匹配,与 OR 相同。

 

示例代码

{    "bool":{      "must":[],      "should":[],      "must_not":[],    }}

 

term过滤器

term单过滤

{    "terms":{      "money":20    }}

 

表示money包含20的记录

 

terms复数版本

允许多个匹配条件

{    "terms":{      "money": [20,30]    }}

 

表示money包含20或者30的记录

 

结合bool+term来举一个实际的例子:

查询path字段中包含applogs最近1分钟的记录

"bool": {    "must": [        {            "terms": {                "path": [                    "applogs",                ]            }        },        {            "range": {                '@timestamp': {
'gt': 'now-1m'} } } ]}
View Code

 

这里使用了terms复数版本,可以随时添加多个条件!

正则查询 

{    "regexp": {        "http_status_code": "5.*"    }}

 

match查询

match 精确匹配

{    "match":{      "email":"123456@qq.com"    }}

 

multi_match 多字段搜索

{    "multi_match":{      "query":"11",      "fields":["Tr","Tq"]    }}

 

demo

获取最近一小时的数据

{
'query': {
'filtered': {
'filter': {
'range': {
'@timestamp':{
'gt':'now-1h'}} } } }}
View Code

条件过滤查询

{    "query":{        "filtered":{            "query":{
"match":{
"http_status_code":500}}, "filter":{
"term":{
"server_name":"vip03"}} } }}
View Code

 

Terms Facet 单字段统计

{
'facets': {
'stat': {
'terms': {
'field':'http_status_code', 'order':'count', 'size':50} } }, 'size':0}
View Code

 

一次统计多个字段

{
'facets': {
'cip': {
'terms': {
'fields':['client_ip']}}, 'status_facets':{
'terms':{
'fields':['http_status_code'], 'order':'term', 'size':50}}}, 'query':{
'query_string':{
'query':'*'}}, 'size':0}
View Code

 

多个字段一起统计

{
'facets': {
'tag': {
'terms': {
'fields':['http_status_code','client_ip'], 'size':10 } } }, 'query': {
'match_all':{}}, 'size':0}
View Code

 

数据组装

以下是kibana首页的demo,用来统计一段时间内的日志数量

{  "facets": {    "0": {      "date_histogram": {        "field": "@timestamp",        "interval": "5m"      },      "facet_filter": {        "fquery": {          "query": {            "filtered": {              "query": {                "query_string": {                  "query": "*"                }              },              "filter": {                "bool": {                  "must": [                    {                      "range": {                        "@timestamp": {                          'gt': 'now-1h'                        }                      }                    },                    {                      "exists": {                        "field": "http_status_code.raw"                      }                    },                    # --------------- -------                    # 此处加匹配条件                  ]                }              }            }          }        }      }    }  },  "size": 0}
View Code

 

如果想添加匹配条件,在以上代码标识部分加上过滤条件,按照以下代码格式即可

{"query": {    "query_string": {
"query": "backend_name:baidu.com"} }},
View Code

 

先介绍到这里,后续会有Query DSL API介绍。

 

三、需求分析

需求

下面是kibana展示的日志

需要统计某一天的日志,统计每一个小时用户数,要求用户id不能重复。一个用户id就是一个用户,也称之为一个PV。

看一段message字段信息

2018-12-17 12:00:00,533 l=INFO [r=9538381535][s=2] [t=http-xxx-543] [APP=user] [Class=o.s.u.c.AccountController:1189] [Method=findCustomerByLoid]- Operation=find customer by loid,Params=loid:001,Content=start

 

其中有一个[r=9538381535],这个9538381535就是用户id。那么用户登录手机APP操作,都会带着这个id,产生一条日志。

比如user项目,那么最终要的数据格式如下:

"user":{    "00":1,    "01":0,    ...    "22":3245,    "23":765}

 

这里使用24小时制来表示每一个时间段,有多个个用户访问了。注意:已经去重了用户id,统计用户数!

 

四、相关技术点

在放出最终代码之前,先来介绍相关技术点,便于理解代码。按照代码从上到下原则,分别来介绍!

项目列表

project_list = ['user',...]

实际的项目是user,但是存储到elasticsearch中,是userlogs,加了一个logs后缀。这个是java后端代码定义的,便于识别!

 

判断日期是否合法

def isVaildDate(self, date):    try:        if ":" in date:            time.strptime(date, "%Y-%m-%d %H:%M:%S")        else:            time.strptime(date, "%Y-%m-%d")        return True    except:        return False
View Code

 

因为需要统计一周的数据,所以脚本执行时,需要传一个日期参数。那么日期参数,传给程序是否合法呢?需要有一个函数来判断!

 

记录日志

def write_log(self, content):        """        写入日志文件        :param path:        :param content:        :return:        """        path = "print.log"        with open(path, mode='a+', encoding='utf-8') as f:            content = time.strftime('%Y-%m-%d %H:%M:%S') + ' ' + content + "\n"            print(content)            f.write(content)
View Code

 

为啥不用Python的日志模块呢?因为测试发现,它写入一些,我不想要的信息,太占用磁盘空间了。所以,我单独写了一个记录日志方法。

 

获取elasticsearch数据

def Get_Data_By_Body(self, project, fixed_date, hour):    """    获取数据    :param project: 项目名    :param fixed_date: 指定日期    :param hour: 24小时制中的某一个小时    :return: object    """    # 查询条件,查询项目最近1小时的数据。    doc = {        "query": {            "bool": {                "must": [                    {                        "terms": {                            "path": [                                project + "logs",                            ]                        }                    },                    {                        "range": {                            '@timestamp': {                                "gt": "{}T{}:00:00".format(fixed_date, hour),                                "lt": "{}T{}:59:59".format(fixed_date, hour),                                "time_zone": "Asia/Shanghai"                            }                        }                    }                ]            }        }    }
View Code

 

由于线上数据量过大,因此直接查询一天的数据,会卡死。所以是切分为每一个小时查询!

上面的query表示查询语句,大概就是查询指定项目(项目名+logs),1小时范围内的数据

 

scroll获取数据

由于1小时内的数据量,也很大。不能直接返回!默认不指定size,是返回10条数据!

size = 1000  # 指定返回1000条queryData = self.es.search(index=self.index_name, body=doc, size=size, scroll='1m', )

 

参数解释:

size 指定返回的条数,默认返回10条

index 指定索引名

body 查询语句

scroll 告诉 Elasticsearch 把搜索上下文再保持一分钟。1m表示1分钟

 

返回结果

mdata = queryData.get("hits").get("hits")  # 返回数据,它是一个列表类型if not mdata:    self.write_log('%s mdata is empty!' % project)

 

queryData 返回一个字典,那么真正的查询结果在queryData['hits']['hits']中,如果这个值没有,表示没有查询到数据!

注意:它并不是返回所有的结果,而是一页的数据,是一个列表类型。因为我们使用了scroll获取数据,只返回一页!

 

分页数据

上面只是返回了1页,我要所有数据,怎么办?需要使用分页,先来看一下分页公式

divmod(总条数, 每页大小)

 

注意:divmod返回一个元祖,第一个元素,就是要分页数

 

总条数,使用

total = queryData['hits']['total']  # 返回数据的总条数

 

每页大小,就是上面指定的size

size = 1000  # 指定返回1000条

 

那么遍历每一页数据,需要这样

scroll_id = queryData['_scroll_id']  # 获取scrollIDtotal = queryData['hits']['total']  # 返回数据的总条数# 使用divmod设置分页查询# divmod(total,1000)[0]+1 表示总条数除以1000,结果取整数加1for i in range(divmod(total, size)[0] + 1):    res = self.es.scroll(scroll_id=scroll_id, scroll='1m')  # scroll参数必须指定否则会报错    mdata += res["hits"]["hits"]  # 扩展列表

 

scroll_id给es.scroll获取数据使用,这个参数必须要有。

由于Python中的range是顾头不顾尾,所以需要加1。使用for循环,就可以遍历每一个分页数

es.scroll(scroll_id=scroll_id, scroll='1m') 才是真正查询每一页的数据,必须要指定这2个参数。它的返回结果,就是查询结果!返回一个列表

上面的mdata是一个列表,res也是列表。因此使用+=就可以扩展列表,得到所有数据!

 

创建年月日目录

def create_folder(self, fixed_date):    """    创建年/月/日 文件夹    :return: path    """    # 系统当前时间年份    # year = time.strftime('%Y', time.localtime(time.time()))    # # 月份    # month = time.strftime('%m', time.localtime(time.time()))    # # 日期    # day = time.strftime('%d', time.localtime(time.time()))    # 年月日    year, month, day = fixed_date.split("-")    # 具体时间 小时分钟毫秒    # mdhms = time.strftime('%m%d%H%M%S', time.localtime(time.time()))    # 判断基础目录是否存在    if not os.path.exists(os.path.join(self.BASE_DIR, 'data_files')):        os.mkdir(os.path.join(self.BASE_DIR, 'data_files'))    # 年月日    fileYear = os.path.join(self.BASE_DIR, 'data_files', year)    fileMonth = os.path.join(fileYear, month)    fileDay = os.path.join(fileMonth, day)    # 判断目录是否存在,否则创建    try:        if not os.path.exists(fileYear):            os.mkdir(fileYear)            os.mkdir(fileMonth)            os.mkdir(fileDay)        else:            if not os.path.exists(fileMonth):                os.mkdir(fileMonth)                os.mkdir(fileDay)            else:                if not os.path.exists(fileDay):                    os.mkdir(fileDay)        return fileDay    except Exception as e:        print(e)        return False
View Code

 

统计结果是最终写入到一个txt里面,那么如何存储呢?使用年月日目录在区分,可以知道这个txt文件,是属于哪一天的。到了一定时间后,可以定期清理,非常方便!

这里使用的传参方式,传入一个日期。所以使用"-"就可以切割出年月日

# 年月日year, month, day = fixed_date.split("-")

 

输出24小时

使用以下代码就可以实现

hour_list = ['{num:02d}'.format(num=i) for i in range(24)]

输出:

['00', '01', '02', '03', '04', '05', '06', '07', '08', '09', '10', '11', '12', '13', '14', '15', '16', '17', '18', '19', '20', '21', '22', '23']

 

项目统计字典

需要统计每一个项目的每一个小时的用户id,用户id必须去重。既然要去重,我们首先会想到用集合。

但是还有一个办法,使用字典,也可以去重。因为字典的key是唯一的。

构造24小时字典

先来构造项目user的数据,格式如下:

"basebusiness": {    "00": {},    "01": {},    "02": {},    "03": {},    "04": {},    "05": {},    "06": {},    "07": {},    "08": {},    "09": {},    "10": {},    "11": {},    "12": {},    "13": {},    "14": {},    "15": {},    "16": {},    "17": {},    "18": {},    "19": {},    "20": {},    "21": {},    "22": {},    "23": {},}
View Code

 

这只是一个项目,实际有很多项目。所以每一个字典,都有这样的24小时数据。相关代码如下:

project_dic = {}  # 项目统计字典# 24小时hour_list = ['{num:02d}'.format(num=i) for i in range(24)]for hour in hour_list:  # 遍历24小时    # print("查询{}点的数据###############".format(hour))    self.write_log("查询{}点的数据###############".format(hour))    for project in project_list:  # 遍历项目列表        if not project_dic.get(project):            project_dic[project] = {}  # 初始化项目字典        if not project_dic[project].get(hour):            project_dic[project][hour] = {}  # 初始化项目小时字典
View Code

 

这里的每一个小时,都是空字典。还没有添加数据,需要添加用户id,下面会讲到!

 

正则匹配用户id

看这一点字符串

2018-12-17 12:00:00,533 l=INFO [r=9538381535][s=2] [t=http-xxx-543] [APP=user]

 

需要提取出9538381535,思路就是:匹配中括号内容-->提取以r=开头的内容-->使用等号切割,获取用户id

匹配中括号内容

p1 = re.compile(r'[[](.*?)[]]', re.S)  # 最小匹配,匹配中括号的内容

 

注意:这里要使用最小匹配,不能使用贪婪匹配。这一段正则,我是用网上找的,测试ok

 

提取和切割,就比较简单了。使用startswith和split方法,就可以了!

使用字典去重

接下来,需要将用户id写入到字典中,需要去重,否则字典添加时,会报错!

那么如何使用字典去重呢?只需要遵循一个原则即可! 有则忽略,无则添加

# 判断字典中rid不存在时,避免字典键值重复if not project_dic[project][hour].get(rid):    project_dic[project][hour][rid] = True  # 添加值

 

生成器

这里主要在2个方法中,使用了生成器。生成器的优点,就是节省内容。

一处在是Get_Data_By_Body方法中,它需要返回所有查询的数据。数据量非常大,因此必须要生成器,否则服务器内容就溢出!

还有一处,就main方法。它是返回项目的统计结果。注意,它不是最终结果。它里面保存了每一个项目,每一个小时中的用户id,是已经去重的用户id。

数据量也是比较大,当然,没有Get_Data_By_Body方法返回的结果大。

 

统计每一个小时用户数

main方法,返回的字典格式如下:

"user":{    "00":{        "242412":True,    }    "01":{        "":True,    },    ...    "22":{        "457577":True,        "546583":True,    },    "23":{        "457577":True,        "546583":True,        "765743":True,    }}

 

我需要知道,每一个小时的用户数。怎么统计呢?用2个方法

1. 遍历字典的每一个小时,使用计数器

2. 使用len方法(推荐)

 

最简单的方法,就是使用len方法,就可以知道每一个小时有多少个key

for i in dic:  # 遍历数据    if not final_dic.get(i):        final_dic[i] = {}  # 初始化字典    for h in sorted(dic[i]):  # 遍历项目的每一个小时        # 统计字典的长度        final_dic[i][h] = len(dic[i][h])

 

有序字典

看下面的数据

 

可以发现,24小时,排序是乱的。这样给领导看时,不太美观。所以需要对24小时进行排序!

在Python 3.6之前,字典的key是无序的。因此,需要定义一个有序字典,在写入之前,要对字典的key做一次排序。

这样顺序写入到有序字典之后,之后再次调用,依然是有序的!

order_dic = OrderedDict()  # 实例化一个有序字典    final_dic = {}  # 最终统计结果    for dic in data:  # 遍历生成器        for i in dic:  # 遍历数据            if not final_dic.get(i):                final_dic[i] = order_dic  # 初始化字典            # 有序字典必须先对普通字典key做排序            for h in sorted(dic[i]):  # 遍历项目的每一个小时                # 统计字典的长度                final_dic[i][h] = len(dic[i][h])

 

完整代码

#!/usr/bin/env python3# coding: utf-8import reimport osimport sysimport jsonimport timefrom collections import OrderedDictfrom elasticsearch import Elasticsearch# 项目列表project_list = ['usercenter', ['login']]# yesterday = (datetime.datetime.now() + datetime.timedelta(days=-1)).strftime("%Y-%m-%d")# today = datetime.datetime.now().strftime("%Y-%m-%d")class ElasticObj:    def __init__(self, index_name, ip, fixed_date, timeout=3600):        '''        :param index_name: 索引名称        :param ip: elasticsearch地址        :param timeout: 设置超时间,默认是10秒的,如果数据量很大,时间要设置更长一些        '''        self.index_name = index_name        self.ip = ip        self.timeout = timeout        # 无用户名密码状态        # self.es = Elasticsearch([self.ip], timeout=self.timeout)        # 用户名密码状态        # self.es = Elasticsearch([ip],http_auth=('elastic', 'password'),port=9200)        self.es = Elasticsearch([self.ip], http_auth=('elastic', '123456'), timeout=self.timeout)        self.fixed_date = fixed_date  # 指定日期        # 当前py文件所在的文件夹        self.BASE_DIR = os.path.dirname(os.path.abspath(__file__))        self.fileDay = self.create_folder()  # 创建日志和数据目录    @staticmethod    def isVaildDate(date):        """        判断日期是否合法        :param date: 日期,比如: 2018-03-30        :return: bool        """        try:            if ":" in date:                time.strptime(date, "%Y-%m-%d %H:%M:%S")            else:                time.strptime(date, "%Y-%m-%d")            return True        except:            return False    def write_log(self, content):        """        写入日志文件        :param content: 写入内容        :return:        """        path = os.path.join(self.fileDay,"output_%s.log" %self.fixed_date)        # path = "output_{}.log".format(self.fixed_date)        with open(path, mode='a+', encoding='utf-8') as f:            content = time.strftime('%Y-%m-%d %H:%M:%S') + ' ' + content + "\n"            print(content)            f.write(content)    def Get_Data_By_Body(self, project, hour):        """        获取数据        :param project: 项目名        :param hour: 24小时制中的某一个小时        :return: 生成器        """        # doc = {'query': {'match_all': {}}}        # 查询条件,查询项目最近1小时的数据。now-1h表示最近1小时        # print(type(fixed_date))        # print("{date}T00:00:00".format(date=fixed_date))        # 24小时        doc = {            "query": {                "bool": {                    "must": [                        {                            "terms": {                                "path": [                                    project + "logs",                                ]                            }                        },                        {                            # "range": {
# '@timestamp': {'gt': 'now-1m'} # } "range": { '@timestamp': { "gt": "{}T{}:00:00".format(self.fixed_date, hour), "lt": "{}T{}:59:59".format(self.fixed_date, hour), "time_zone": "Asia/Shanghai" } } } ] } } } # queryData = self.es.search(index=self.index_name, body=doc) # scroll 参数告诉 Elasticsearch 把搜索上下文再保持一分钟,1m表示1分钟 # size 参数允许我们配置没匹配结果返回的最大命中数。每次调用 scroll API 都会返回下一批结果,直到不再有可以返回的结果,即命中数组为空。 size = 1000 # 指定返回1000条 queryData = self.es.search(index=self.index_name, body=doc, size=size, scroll='1m', ) # print(queryData['hits']['total']) mdata = queryData.get("hits").get("hits") # 返回查询的数据,不是所有数据,而是一页的数据,它是一个列表类型 if not mdata: self.write_log('%s mdata is empty!' % project) # scroll_id 的值就是上一个请求中返回的 _scroll_id 的值 scroll_id = queryData['_scroll_id'] # 获取scrollID total = queryData['hits']['total'] # 返回数据的总条数 # print("查询项目{} {}点的数据,总共有{}条".format(project,hour,total)) self.write_log("查询项目{} {}点的数据,总共有{}条".format(project, hour, total)) # 使用divmod设置分页查询 # divmod(total,1000)[0]+1 表示总条数除以1000,结果取整数加1 for i in range(divmod(total, size)[0] + 1): res = self.es.scroll(scroll_id=scroll_id, scroll='1m') # scroll参数必须指定否则会报错 mdata += res["hits"]["hits"] # 扩展列表 # yield mdata # print(mdata) # return mdata yield mdata def create_folder(self): """ 创建年/月/日 文件夹 :return: path """ # 系统当前时间年份 # year = time.strftime('%Y', time.localtime(time.time())) # # 月份 # month = time.strftime('%m', time.localtime(time.time())) # # 日期 # day = time.strftime('%d', time.localtime(time.time())) # 年月日 year, month, day = self.fixed_date.split("-") # 具体时间 小时分钟毫秒 # mdhms = time.strftime('%m%d%H%M%S', time.localtime(time.time())) # 判断基础目录是否存在 if not os.path.exists(os.path.join(self.BASE_DIR, 'data_files')): os.mkdir(os.path.join(self.BASE_DIR, 'data_files')) # 年月日 fileYear = os.path.join(self.BASE_DIR, 'data_files', year) fileMonth = os.path.join(fileYear, month) fileDay = os.path.join(fileMonth, day) # 判断目录是否存在,否则创建 try: if not os.path.exists(fileYear): os.mkdir(fileYear) os.mkdir(fileMonth) os.mkdir(fileDay) else: if not os.path.exists(fileMonth): os.mkdir(fileMonth) os.mkdir(fileDay) else: if not os.path.exists(fileDay): os.mkdir(fileDay) return fileDay except Exception as e: print(e) return False def main(self): """ 主要处理逻辑 :return: 生成器 """ project_dic = {} # 项目统计字典 # fixed_date = datetime.datetime.strptime(fixed_date, "%Y-%m-%d") # strftime("%Y-%m-%d") # conv_date = fixed_date.strftime("%Y-%m-%d") # print(conv_date, type(conv_date)) # exit() # now_hour = fixed_date.strftime('%H') # 当前时间的小时 # print(now_hour) # 24小时 hour_list = ['{num:02d}'.format(num=i) for i in range(24)] # hour_list = ['{num:02d}'.format(num=i) for i in range(2)] # project="usercenter" # project_dic[project] = {now_hour: {}} # 初始化字典 for hour in hour_list: # 遍历24小时 # print("查询{}点的数据###############".format(hour)) self.write_log("查询{}点的数据###############".format(hour)) for project in project_list: # 遍历项目列表 if not project_dic.get(project): project_dic[project] = {} # 初始化项目字典 if not project_dic[project].get(hour): project_dic[project][hour] = {} # 初始化项目小时字典 mdata = self.Get_Data_By_Body(project, hour) # 获取数据 for item in mdata: # 遍历生成器 for hit in item: # 遍历返回数据 # hit是一个字典 str1 = hit['_source']['message'] # 查询指定字典 p1 = re.compile(r'[[](.*?)[]]', re.S) # 最小匹配,匹配中括号的内容 for i in re.findall(p1, str1): # 遍历结果 if i.startswith('r='): # 判断以r=开头的 rid = i.split("=")[1] # 获取rid # print("rid",rid) # 判断字典中rid不存在时,避免字典键值重复 if not project_dic[project][hour].get(rid): project_dic[project][hour][rid] = True # 添加值 time.sleep(1) # 休眠1秒钟 # return project_dic yield project_dicif __name__ == '__main__': # fixed_date = "2018-12-16" fixed_date = sys.argv[1] # 日期参数 if not ElasticObj.isVaildDate(fixed_date): print("日期不合法!") exit() startime = time.time() # 开始时间 index_name = "common-*" es_server = "192.168.92.131" obj = ElasticObj(index_name, es_server, fixed_date) # 连接elasticsearch print("正在查询日期%s这一天的数据" % fixed_date) obj.write_log("###########################################") obj.write_log("正在查询日期%s这一天的数据" % fixed_date) data = obj.main() # print("初步结果",data) # fileDay = obj.create_folder() # 创建目录 # if not fileDay: # # print("创建目录失败!") # obj.write_log("创建目录失败!") # exit() order_dic = OrderedDict() # 实例化一个有序字典 final_dic = {} # 最终统计结果 for dic in data: # 遍历生成器 for i in dic: # 遍历数据 if not final_dic.get(i): final_dic[i] = order_dic # 初始化字典 # 有序字典必须先对普通字典key做排序 for h in sorted(dic[i]): # 遍历项目的每一个小时 # 统计字典的长度 final_dic[i][h] = len(dic[i][h]) # print("最终结果",final_dic) # 统计结果 obj.write_log("最终结果执行完毕!") # 写入文件 with open(os.path.join(obj.fileDay, 'access_data.txt'), encoding='utf-8', mode='a') as f: f.write(json.dumps(final_dic) + "\n") endtime = time.time() take_time = endtime - startime if take_time < 1: # 判断不足1秒时 take_time = 1 # 设置为1秒 # 计算花费时间 m, s = divmod(take_time, 60) h, m = divmod(m, 60) # print("本次花费时间 %02d:%02d:%02d" % (h, m, s)) obj.write_log("统计日期%s这一天的数据完成!请查阅data_files目录的日志和数据文件" % fixed_date) obj.write_log("本次花费时间 %02d:%02d:%02d" % (h, m, s))
View Code

 

 

日志文件和数据文件,都在年月日目录里面!

 

本文参考链接:

 

转载于:https://www.cnblogs.com/xiao987334176/p/10130712.html

你可能感兴趣的文章
postgresql学习文档
查看>>
Struts2返回JSON数据的具体应用范例
查看>>
js深度克隆对象、数组
查看>>
socket阻塞与非阻塞,同步与异步
查看>>
团队工作第二天
查看>>
System类
查看>>
tableView
查看>>
Happy Great BG-卡精度
查看>>
Xamarin Visual Studio不识别JDK路径
查看>>
菜鸟“抄程序”之道
查看>>
Ubuntu下关闭防火墙
查看>>
TCP/IP 邮件的原理
查看>>
原型设计工具
查看>>
windows下的C++ socket服务器(4)
查看>>
css3 2d转换3d转换以及动画的知识点汇总
查看>>
【Java】使用Eclipse进行远程调试,Linux下开启远程调试
查看>>
对Vue为什么不支持IE8的解释之一
查看>>
计算机改名导致数据库链接的诡异问题
查看>>
Java8内存模型—永久代(PermGen)和元空间(Metaspace)(转)
查看>>
ObjectiveC基础教程(第2版)
查看>>