GO es数据准备DB copy生成泛型解决重复代码

原创已于 2024-11-11 17:12:20 修改 · 380 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#elasticsearch #大数据 #搜索引擎

于 2024-11-11 17:11:24 首次发布

go实战全家桶同时被 3 个专栏收录

258 篇文章

订阅专栏

228 篇文章

订阅专栏

elastic

13 篇文章

订阅专栏

每个es的代码

//bulkRequest := es.GetElastic1().Bulk()
var qq=facade.Default[*esentity.NotesEs]()
for _, note := range notes {
    noteEs := &esentity.NotesEs{}
    noteEs.EntityFillEs(note)
    //index1Req := elastic.NewBulkIndexRequest().Index(esentity.NotesEs{}.AliasName()).Id(utils.FormatInt(noteEs.Id)).Doc(noteEs)
    //bulkRequest = bulkRequest.Add(index1Req)
    qq.AddDocObject(noteEs)
}
//bulkResponse, err := bulkRequest.Do(context.TODO())
var ret= qq.BulkIndex()
if ret.IsFailed() {
    golog.Error(ret)
}
if ret.Data==nil {
}

泛型接口通用写法

facade.EsBulkIndex[entity.Notes,*esentity.NotesEs](notes)

方法通用实现

func EsBulkIndex[E schema.Tabler, T esiface.EsObjectStru[E]](entities []*E) bool {
    var qq = Default[T]()
    for _, entity := range entities {

       var esObj = esiface.NewOfPtrEsObject[T]() //gconv.Struct(entity, esObj)
       esObj.EntityFillEs(entity)
       qq.AddDocObject(esObj)
    }
    ret := qq.BulkIndex()
    if ret.IsFailed() {
       golog.Error(ret)
    }
    if ret.Data == nil {
       golog.Error("expected bulkResponse to be != nil; got nil")
       return true
    }
    return false
}

type EsObjectStru[E any] interface {
    EsObject
    EntityFillEs(entity *E)
}
type EsObject interface {
    Unmarshal([]byte) error

    IndexID() string
    TableName() string
    IndexName() string
    AliasName() string
    GetMapping() string

    Mapping() string
    InitDoc() EsObject
}