个人中心
gpt4 book ai didi

Go语言Elasticsearch数据清理工具思路详解

转载 作者：qq735679552 更新时间：2022-09-27 22:32:09 25 4
gpt4 key购买 nike

CFSDN坚持开源创造价值，我们致力于搭建一个资源共享平台，让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章Go语言Elasticsearch数据清理工具思路详解由作者收集整理，如果你对这篇文章有兴趣，记得点赞哟.

Go语言Elasticsearch数据清理工具思路详解

微服务架构中收集通常大家都采用ELK进行日志收集，同时我们还采用了SkyWalking进行链路跟踪，而SkyWalking数据存储也用到了ES，SkyWalking每天产生大量的索引数据，如下:

Go语言Elasticsearch数据清理工具思路详解

WX20211008-104751@2x 。

这里一天大概产生了700左右个索引数据。对历史的链路数据我们不做过多的保留.

这里我整理了个小工具，可以定期清理es数据.

1、清理思路

可以看到索引数据都是以日期结尾，我们可以根据日期去匹配索引数据，并对索引进行删除。这里需要考虑一点，有的Es服务开启了索引保护机制，不能通过*index去删除，只能通过索引的全名称去删除。所以我们整体流程如下:

1、获取es服务中全部索引数据.

2、根据当前时间-保留天数，获取要删除的日期.

3、通过字符串匹配，判断索引中是否包含要删除的日期，如果包含则进行删除.

4、工具友好性，我们可以通过配置文件配置ES服务地址、日期格式化类型、保留天数等信息.

2、代码实现

2.1、获取ES服务中全部索引数据

要获取Es服务中全部索引数据，我们首先连接Es服务器，这里我们使用github.com/olivere/elastic/v7库操作Es.

连接ES

func GetEsClient(data Data) *elastic.Client {    Init()    file := "./eslog.log"    logFile, _ := os.OpenFile(file, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0766)     client, err := elastic.NewClient(        elastic.SetURL(data.Host),        elastic.SetSniff(false),        elastic.SetInfoLog(log.New(logFile, "ES-INFO: ", 0)),        elastic.SetTraceLog(log.New(logFile, "ES-TRACE: ", 0)),        elastic.SetErrorLog(log.New(logFile, "ES-ERROR: ", 0)),    )    if err != nil {        return nil    }    return client}

我们通过GetEsClient方法，连接ES,并返回client，供后续方法使用。这里的Data是包含了ES服务地址等信息，我们后面会给出Data的数据结构.

获取全部索引数据 。

func getIndex(data Data) map[string]interface{} {    client := GetEsClient(data)    mapping := client.GetMapping()    service := mapping.Index("*")    result, err := service.Do(context.Background())    if err != nil {        fmt.Printf("create index failed, err: %v", err)        return nil    }    return result}

通过client.GetMapping().Index("*")API获取es服务中全部的索引数据，并返回，数据格式如下:

Go语言Elasticsearch数据清理工具思路详解

WX20211008-110537@2x 。

这次我们获取全部索引完成.

2.2、根据当前时间-保留天数，获取要删除的日期

我们根据当前时间-保留天数，获取当前需要删除的日期数据。我们通过GoLang内置的函数库time完成该功能的实现.

currentTime := time.Now()//获取当前时间oldTime := currentTime.AddDate(0, 0, data.Day)//通过配置文件获取保留天数format := oldTime.Format(data.IndexFmt)//通过配置文件获取序列化日期格式

2.3、通过字符串匹配，判断索引中是否包含要删除的日期，如果包含则进行删除

这里通过字符串匹配进行判断是否需要删除索引数据.

func delIndex(data Data) {    currentTime := time.Now()    oldTime := currentTime.AddDate(0, 0, data.Day)    format := oldTime.Format(data.IndexFmt)    index := getIndex(data)//获取全部索引    for k := range index {//遍历索引数据        fmt.Println("key:", k, "format:", format)        if find := strings.Contains(k, format); find { //判断索引中是否包含要删除的日期格式，            DelIndex(data, k)//如果包含则调用DelIndex方法删除        }    }}
// DelIndex 删除 indexfunc DelIndex(data Data, index ...string) bool {    client := GetEsClient(data)    response, err := client.DeleteIndex(index...).Do(context.Background())    if err != nil {        fmt.Printf("delete index failed, err: %v", err)        return false    }    return response.Acknowledged}

通过DeleteIndexAPI删除指定的数据.

2.4、通过配置文件灵活配置数据

这里我们定义了Config和Data对象，对象结构如下:

type Config struct {    Data []Data `json:"data"`} type Data struct {    Host     string `json:"host"`    IndexFmt string `json:"index_fmt"`    Day      int    `json:"day"`}

配置文件内容如下:

{  "data": [    {      "host": "http://ip1:9200",//服务IP      "index_fmt": "20060102",//日期格式化      "day": -1 //保留天数 保留1天    },    {      "host": "http://ip2:9200/",      "index_fmt": "20060102",      "day": -1    },    {      "host": "http://ip3:32093",      "index_fmt": "2006.01.02",      "day": -7  //保留天数 保留7天    }  ]}

我们通过Init方法加载配置文件到Config,

var config Config func Init() {    JsonParse := NewJsonStruct()    //下面使用的是相对路径，config.json文件和main.go文件处于同一目录下    JsonParse.Load("config/config.json", &config)} type JsonStruct struct {} func NewJsonStruct() *JsonStruct {    return &JsonStruct{}} func (jst *JsonStruct) Load(filename string, v interface{}) {    //ReadFile函数会读取文件的全部内容，并将结果以[]byte类型返回    data, err := ioutil.ReadFile(filename)    if err != nil {        return    }     //读取的数据为json格式，需要进行解码    err = json.Unmarshal(data, v)    if err != nil {        return    }}

编写Main方法运行程序:

func main() {    Init()    for i, datum := range config.Data {        fmt.Printf("config data Host is [%s], fmt is [%s]", datum.Host, datum.IndexFmt)        println(i)        delIndex(datum)    }}

这里我们依然遍历配置文件中的多个服务配置。可以同时管理多个Es服务.

3、完整代码

package main import (    "encoding/json"    "fmt"    "io/ioutil"    "strings"    "time") type Config struct {    Data []Data `json:"data"`} type Data struct {    Host     string `json:"host"`    IndexFmt string `json:"index_fmt"`    Day      int    `json:"day"`} var config Config func Init() {    JsonParse := NewJsonStruct()    //下面使用的是相对路径，config.json文件和main.go文件处于同一目录下    JsonParse.Load("config/config.json", &config)} type JsonStruct struct {} func NewJsonStruct() *JsonStruct {    return &JsonStruct{}} func (jst *JsonStruct) Load(filename string, v interface{}) {    //ReadFile函数会读取文件的全部内容，并将结果以[]byte类型返回    data, err := ioutil.ReadFile(filename)    if err != nil {        return    }     //读取的数据为json格式，需要进行解码    err = json.Unmarshal(data, v)    if err != nil {        return    }} func delIndex(data Data) {    currentTime := time.Now()    oldTime := currentTime.AddDate(0, 0, data.Day)    format := oldTime.Format(data.IndexFmt)    index := getIndex(data)    for k := range index {        fmt.Println("key:", k, "format:", format)        if find := strings.Contains(k, format); find {            DelIndex(data, k)        }    }} func main() {    Init()    for i, datum := range config.Data {        fmt.Printf("config data Host is [%s], fmt is [%s]", datum.Host, datum.IndexFmt)        println(i)        delIndex(datum)    }}
package main import (    "context"    "fmt"    "github.com/olivere/elastic/v7"    "log"    "os"    "time") // GetEsClient 初始化客户端func GetEsClient(data Data) *elastic.Client {    Init()    file := "./eslog.log"    logFile, _ := os.OpenFile(file, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0766) // 应该判断error，此处简略    client, err := elastic.NewClient(        elastic.SetURL(data.Host),        elastic.SetSniff(false),        elastic.SetInfoLog(log.New(logFile, "ES-INFO: ", 0)),        elastic.SetTraceLog(log.New(logFile, "ES-TRACE: ", 0)),        elastic.SetErrorLog(log.New(logFile, "ES-ERROR: ", 0)),    )    if err != nil {        return nil    }    return client} // IsDocExists 判断索引是否存储func IsDocExists(data Data, id string, index string) bool {    client := GetEsClient(data)    defer client.Stop()    exist, _ := client.Exists().Index(index).Id(id).Do(context.Background())    if !exist {        log.Println("ID may be incorrect! ", id)        return false    }    return true} // PingNode 是否联通func PingNode(data Data) {    start := time.Now()    client := GetEsClient(data)    info, code, err := client.Ping(data.Host).Do(context.Background())    if err != nil {        fmt.Printf("ping es failed, err: %v", err)    }    duration := time.Since(start)    fmt.Printf("cost time: %v", duration)    fmt.Printf("Elasticsearch returned with code %d and version %s", code, info.Version.Number)} // GetDoc 获取文档func GetDoc(data Data, id string, index string) (*elastic.GetResult, error) {    client := GetEsClient(data)    defer client.Stop()    if !IsDocExists(data, id, index) {        return nil, fmt.Errorf("id不存在")    }    esResponse, err := client.Get().Index(index).Id(id).Do(context.Background())    if err != nil {        return nil, err    }    return esResponse, nil} // CreateIndex 创建 indexfunc CreateIndex(data Data, index, mapping string) bool {    client := GetEsClient(data)    result, err := client.CreateIndex(index).BodyString(mapping).Do(context.Background())    if err != nil {        fmt.Printf("create index failed, err: %v", err)        return false    }    return result.Acknowledged} // DelIndex 删除 indexfunc DelIndex(data Data, index ...string) bool {    client := GetEsClient(data)    response, err := client.DeleteIndex(index...).Do(context.Background())    if err != nil {        fmt.Printf("delete index failed, err: %v", err)        return false    }    return response.Acknowledged} func getIndex(data Data) map[string]interface{} {    client := GetEsClient(data)    mapping := client.GetMapping()    service := mapping.Index("*")    result, err := service.Do(context.Background())    if err != nil {        fmt.Printf("create index failed, err: %v", err)        return nil    }    return result}

代码已经上传github需要的可自行下载.

到此这篇关于Go语言Elasticsearch数据清理工具的文章就介绍到这了,更多相关Go Elasticsearch数据清理工具内容请搜索我以前的文章或继续浏览下面的相关文章希望大家以后多多支持我！ 。

原文链接：https://blog.csdn.net/July_whj/article/details/120648785 。

最后此篇关于Go语言Elasticsearch数据清理工具思路详解的文章就讲到这里了,如果你想了解更多关于Go语言Elasticsearch数据清理工具思路详解的内容请搜索CFSDN的文章或继续浏览相关文章，希望大家以后支持我的博客！ 。

25 4 0
文章推荐： Go select 死锁的一个细节
文章推荐： Python面向对象程序设计中类的定义、实例化、封装及私有变量/方法详解
文章推荐： Go语言--切片(Slice)详解
文章推荐： Python类的继承、多态及获取对象信息操作详解
qq735679552
个人简介

我是一名优秀的程序员,十分优秀！

滴滴打车优惠券免费领取
滴滴打车优惠券
全站热门文章
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com