Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
2a34ce6043 | ||
|
|
e8f94c911c | ||
|
|
f55b3bc609 |
90
README.md
90
README.md
@@ -1,10 +1,10 @@
|
||||
# utility_go:强大的Go语言工具类库
|
||||
|
||||
## 一、项目概述
|
||||
`utility_go` 是一个功能丰富的Go语言工具类库,旨在帮助开发者更高效地开发Go应用程序。它提供了一系列实用的功能模块,涵盖了数据库操作、缓存处理、支付接口集成、排名算法实现等多个方面,可广泛应用于各种类型的Go项目中。
|
||||
`utility_go` 是一个功能丰富的Go语言工具类库,旨在助力开发者更高效地开发Go应用程序。它提供了一系列实用的功能模块,广泛涵盖数据库操作、缓存处理、支付接口集成、排名算法实现等多个方面,适用于各类Go项目。
|
||||
|
||||
## 二、安装方式
|
||||
要使用 `utility_go` 工具类库,你可以使用 `go get` 命令进行安装。请确保你的Go环境已经正确配置,并且可以访问互联网。
|
||||
要使用 `utility_go` 工具类库,可使用 `go get` 命令进行安装。请确保你的Go环境已正确配置,且可访问互联网。
|
||||
|
||||
### 安装命令
|
||||
```sh
|
||||
@@ -12,7 +12,7 @@ go get github.com/ayflying/utility_go
|
||||
```
|
||||
|
||||
### 安装验证
|
||||
安装完成后,你可以在你的Go代码中导入 `utility_go` 相关的包,检查是否能够正常使用。例如:
|
||||
安装完成后,你可以在Go代码中导入 `utility_go` 相关的包,检查是否能够正常使用。示例如下:
|
||||
```go
|
||||
package main
|
||||
|
||||
@@ -29,39 +29,45 @@ func main() {
|
||||
```
|
||||
|
||||
## 三、项目结构
|
||||
`utility_go` 的项目结构设计清晰,各个模块分工明确,方便开发者使用和扩展。以下是项目的主要目录结构:
|
||||
`utility_go` 的项目结构设计清晰,各模块分工明确,便于开发者使用和扩展。以下是项目的主要目录结构及说明:
|
||||
|
||||
### 主要目录说明
|
||||
- **`api/`**:存放与API相关的代码,包含不同模块的API接口定义,如 `admin`、`callback`、`system` 等。这些API接口用于处理各种外部请求,是项目与外部系统交互的重要入口。
|
||||
- **`service/`**:服务层代码,包含各种业务逻辑的实现,如 `game_act.go`、`ip_2_region.go` 等。服务层负责处理具体的业务需求,将数据处理和业务逻辑封装在独立的函数或方法中,提高代码的可维护性和复用性。
|
||||
- **`controller/`**:控制器层代码,主要用于处理HTTP请求和响应。控制器接收客户端的请求,调用相应的服务层方法进行处理,并将处理结果返回给客户端。
|
||||
- **`internal/`**:内部包,包含项目的核心业务逻辑,如 `game`、`logic`、`model` 等模块。其中,`model` 模块定义了项目中使用的数据模型,如数据库表对应的结构体;`logic` 模块实现了各种业务逻辑的处理函数。
|
||||
- **`package/`**:包含各种功能包,提供了丰富的工具和功能,具体如下:
|
||||
- **`aycache`**:缓存相关的功能包,提供了缓存操作的接口和实现,帮助开发者更方便地使用缓存技术,提高应用程序的性能。
|
||||
- **`elasticsearch`**:Elasticsearch相关的功能包,用于与Elasticsearch搜索引擎进行交互,实现数据的存储、检索和分析等功能。
|
||||
- **`excel`**:Excel处理相关的功能包,提供了Excel文件的读写操作接口,方便开发者处理Excel文件中的数据。
|
||||
- **`pay`**:支付相关的功能包,包含了与各种支付平台的接口集成,如 `apple`(苹果支付)、`playstore`(Google Play Store支付)等,支持应用内购买等支付功能。
|
||||
- **`rand`**:随机数相关的功能包,提供了生成各种随机数的函数,可用于测试、加密等场景。
|
||||
- **`rank`**:排名相关的功能包,实现了各种排名算法,如基于Redis的排行榜功能,可用于游戏排名、活动排名等场景。
|
||||
- **`s3`**:S3存储相关的功能包,用于与Amazon S3等云存储服务进行交互,实现文件的上传、下载、删除等操作。
|
||||
- **`tools/`**:工具类代码,包含了一些常用的工具函数,如 `redis.go`(Redis操作相关)、`time.go`(时间处理相关)、`tools.go`(通用工具函数)等,方便开发者在项目中使用。
|
||||
- **api/**:存放与API相关的代码,包含不同模块的API接口定义,如 `admin`、`callback`、`system` 等。这些API接口用于处理各种外部请求,是项目与外部系统交互的重要入口。
|
||||
- **service/**:服务层代码,包含各种业务逻辑的实现,如 `game_act.go`、`ip_2_region.go` 等。服务层负责处理具体的业务需求,将数据处理和业务逻辑封装在独立的函数或方法中,提高代码的可维护性和复用性。
|
||||
- **controller/**:控制器层代码,主要用于处理HTTP请求和响应。控制器接收客户端的请求,调用相应的服务层方法进行处理,并将处理结果返回给客户端。
|
||||
- **internal/**:内部包,包含项目的核心业务逻辑,如 `game`、`logic`、`model` 等模块。其中,`model` 模块定义了项目中使用的数据模型,如数据库表对应的结构体;`logic` 模块实现了各种业务逻辑的处理函数。
|
||||
- **package/**:包含各种功能包,提供了丰富的工具和功能,具体如下:
|
||||
- **aycache**:缓存相关的功能包,提供了缓存操作的接口和实现,帮助开发者更方便地使用缓存技术,提高应用程序的性能。
|
||||
- **elasticsearch**:Elasticsearch相关的功能包,用于与Elasticsearch搜索引擎进行交互,实现数据的存储、检索和分析等功能。
|
||||
- **excel**:Excel处理相关的功能包,提供了Excel文件的读写操作接口,方便开发者处理Excel文件中的数据。
|
||||
- **pay**:支付相关的功能包,包含了与各种支付平台的接口集成,如 `apple`(苹果支付)、`playstore`(Google Play Store支付)等,支持应用内购买等支付功能。
|
||||
- **rand**:随机数相关的功能包,提供了生成各种随机数的函数,可用于测试、加密等场景。
|
||||
- **rank**:排名相关的功能包,实现了各种排名算法,如基于Redis的排行榜功能,可用于游戏排名、活动排名等场景。
|
||||
- **s3**:S3存储相关的功能包,用于与Amazon S3等云存储服务进行交互,实现文件的上传、下载、删除等操作。
|
||||
- **tools/**:工具类代码,包含了一些常用的工具函数,如 `redis.go`(Redis操作相关)、`time.go`(时间处理相关)、`tools.go`(通用工具函数)等,方便开发者在项目中使用。
|
||||
|
||||
## 四、主要模块功能
|
||||
|
||||
### 4.1 `pay/playstore`
|
||||
该模块主要用于与Google Play Store API交互,处理应用内购买相关的操作。它提供了以下主要功能:
|
||||
- **创建客户端**:通过 `New` 函数创建并返回一个包含访问androidpublisher API所需凭证的http客户端,方便开发者与Google Play Store API进行通信。
|
||||
- **使用自定义客户端**:`NewWithClient` 函数允许开发者使用自定义的http客户端创建并返回一个包含访问androidpublisher API所需凭证的http客户端,增加了客户端的灵活性。
|
||||
### 4.1 pay/playstore
|
||||
该模块主要用于与Google Play Store API交互,处理应用内购买相关的操作,提供以下主要功能:
|
||||
- **创建客户端**:通过 `New` 函数创建并返回一个包含访问 `androidpublisher` API所需凭证的HTTP客户端,方便开发者与Google Play Store API进行通信。
|
||||
- **使用自定义客户端**:`NewWithClient` 函数允许开发者使用自定义的HTTP客户端创建并返回一个包含访问 `androidpublisher` API所需凭证的HTTP客户端,增加了客户端的灵活性。
|
||||
- **验证签名**:`VerifySignature` 函数用于验证应用内购买的签名,确保支付信息的安全性和合法性。
|
||||
|
||||
### 4.2 `s3`
|
||||
`s3` 模块主要用于与S3存储服务进行交互,提供了文件存储和管理的功能。其中,`ListBuckets` 函数可以列出S3存储桶的信息,方便开发者管理存储桶中的文件。
|
||||
### 4.2 s3
|
||||
`s3` 模块主要用于与S3存储服务进行交互,提供文件存储和管理的功能。其中,`ListBuckets` 函数可以列出S3存储桶的信息,方便开发者管理存储桶中的文件。
|
||||
|
||||
### 4.3 `model`
|
||||
### 4.3 model
|
||||
`model` 模块定义了项目中使用的数据模型,这些数据模型通常与数据库表相对应,用于数据的存储和操作。例如:
|
||||
- **`GameMailMass`**:表示游戏邮件群发的数据模型,包含邮件的标题、内容、类型等信息。
|
||||
- **`GameBag`**:表示游戏背包的数据模型,包含用户标识、道具数据、图鉴、手势等信息。
|
||||
- **`MemberSave`**:(根据具体代码中的定义)可能表示用户会员信息的数据模型,用于存储用户的会员相关数据。
|
||||
- **GameMailMass**:表示游戏邮件群发的数据模型,包含邮件的标题、内容、类型等信息。
|
||||
- **GameBag**:表示游戏背包的数据模型,包含用户标识、道具数据、图鉴、手势等信息。
|
||||
- **MemberSave**:(根据具体代码中的定义)可能表示用户会员信息的数据模型,用于存储用户的会员相关数据。
|
||||
|
||||
### 4.4 tools
|
||||
`tools` 模块提供了一系列通用工具函数,涵盖时间处理、Redis操作、道具数据处理等功能:
|
||||
- **时间处理**:`time.go` 文件中的函数可以进行时间计算,如获取本周开始时间、计算两个时间间隔天数等。
|
||||
- **Redis操作**:`redis.go` 文件中的函数可以进行Redis的扫描操作,支持批量获取大量数据。
|
||||
- **通用工具**:`tools.go` 文件中的函数提供了字符串处理、切片操作、道具数据合并等功能。
|
||||
|
||||
## 五、使用示例
|
||||
|
||||
@@ -115,6 +121,33 @@ func main() {
|
||||
}
|
||||
```
|
||||
|
||||
### 5.3 工具模块示例
|
||||
以下是一个使用 `tools` 模块处理道具数据的示例代码:
|
||||
```go
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/ayflying/utility_go/tools"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// 字符串转道具类型
|
||||
str := "1|10|2|20"
|
||||
result := tools.Tools.Spilt2Item(str)
|
||||
fmt.Println("Spilt2Item result:", result)
|
||||
|
||||
// 切片换道具类型
|
||||
slice := []int64{1, 10, 2, 20}
|
||||
res := tools.Tools.Slice2Item(slice)
|
||||
fmt.Println("Slice2Item result:", res)
|
||||
|
||||
// 道具格式转map
|
||||
list := tools.Tools.Items2Map(result)
|
||||
fmt.Println("Items2Map result:", list)
|
||||
}
|
||||
```
|
||||
|
||||
## 六、注意事项
|
||||
- **自动生成文件**:项目中有部分代码文件是由GoFrame CLI工具生成并维护的,这些文件通常会标注有 `// Code generated and maintained by GoFrame CLI tool. DO NOT EDIT.` 注释。请不要手动修改这些文件,以免造成不必要的问题。
|
||||
- **版本兼容性**:在使用 `utility_go` 工具类库时,请确保你的Go语言版本与工具类库的要求版本兼容。建议使用最新的Go语言版本,以获得更好的性能和稳定性。
|
||||
@@ -123,5 +156,4 @@ func main() {
|
||||
`utility_go` 工具类库会定期进行更新和维护,以修复已知的问题、添加新的功能和优化性能。你可以关注项目的GitHub仓库(https://github.com/ayflying/utility_go)获取最新的更新信息。如果你在使用过程中遇到任何问题或有任何建议,欢迎提交Issue或Pull Request。
|
||||
|
||||
## 八、许可证信息
|
||||
`utility_go` 工具类库遵循 MIT许可证,你可以在项目的 `LICENSE` 文件中查看详细的许可证条款。请确保在使用该工具类库时遵守相关的许可证规定。
|
||||
```
|
||||
`utility_go` 工具类库遵循MIT许可证,你可以在项目的 `LICENSE` 文件中查看详细的许可证条款。请确保在使用该工具类库时遵守相关的许可证规定。
|
||||
@@ -2,7 +2,6 @@ package aycache
|
||||
|
||||
import (
|
||||
"github.com/ayflying/utility_go/pkg/aycache/drive"
|
||||
drive2 "github.com/ayflying/utility_go/pkg/aycache/drive"
|
||||
"github.com/gogf/gf/v2/os/gcache"
|
||||
)
|
||||
|
||||
@@ -24,13 +23,13 @@ func New(_name ...string) gcache.Adapter {
|
||||
}
|
||||
switch name {
|
||||
case "cache":
|
||||
cacheAdapterObj = drive2.NewAdapterMemory()
|
||||
cacheAdapterObj = drive.NewAdapterMemory()
|
||||
case "redis":
|
||||
cacheAdapterObj = drive2.NewAdapterRedis()
|
||||
cacheAdapterObj = drive.NewAdapterRedis()
|
||||
case "file":
|
||||
cacheAdapterObj = drive2.NewAdapterFile("runtime/cache")
|
||||
case "es":
|
||||
cacheAdapterObj = drive.NewAdapterElasticsearch([]string{"http://127.0.0.1:9200"})
|
||||
cacheAdapterObj = drive.NewAdapterFile("runtime/cache")
|
||||
//case "es":
|
||||
//cacheAdapterObj = drive.NewAdapterElasticsearch("http://127.0.0.1:9200"})
|
||||
}
|
||||
|
||||
//var client = gcache.New()
|
||||
|
||||
@@ -46,7 +46,7 @@ type elastic struct {
|
||||
//}
|
||||
//
|
||||
//// getDocument 获取文档
|
||||
//func (d *Driver) GetDocument(name string, id string) (res json.RawMessage) {
|
||||
//func (d *Driver) Get(name string, id string) (res json.RawMessage) {
|
||||
// resp, err := d.client.Get(name, id).
|
||||
// Do(context.Background())
|
||||
// if err != nil {
|
||||
|
||||
@@ -4,7 +4,6 @@ import (
|
||||
"fmt"
|
||||
"github.com/ayflying/utility_go/internal/model/do"
|
||||
"github.com/ayflying/utility_go/internal/model/entity"
|
||||
"github.com/ayflying/utility_go/pgk"
|
||||
"github.com/ayflying/utility_go/pkg"
|
||||
service2 "github.com/ayflying/utility_go/service"
|
||||
"github.com/ayflying/utility_go/tools"
|
||||
@@ -216,9 +215,28 @@ func (s *sGameAct) Save(actId int) (err error) {
|
||||
// 清空GetRedDot缓存
|
||||
func (s *sGameAct) RefreshGetRedDotCache(uid int64) {
|
||||
cacheKey := fmt.Sprintf("gameAct:GetRedDot:%s:%d", gtime.Now().Format("d"), uid)
|
||||
_, err := pgk.Cache("redis").Remove(gctx.New(), cacheKey)
|
||||
_, err := pkg.Cache("redis").Remove(gctx.New(), cacheKey)
|
||||
if err != nil {
|
||||
g.Log().Error(ctx, err)
|
||||
g.Dump(err)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *sGameAct) Del(uid int64, actId int) {
|
||||
if uid == 0 || actId == 0 {
|
||||
g.Log().Error(ctx, "当前参数为空")
|
||||
return
|
||||
}
|
||||
// 构造缓存键名
|
||||
keyCache := fmt.Sprintf("act:%v:%v", actId, uid)
|
||||
|
||||
//删除活动缓存
|
||||
g.Redis().Del(ctx, keyCache)
|
||||
|
||||
//删除当前活动储存
|
||||
g.Model(Name).Where(do.GameAct{
|
||||
Uid: uid,
|
||||
ActId: actId,
|
||||
}).Delete()
|
||||
|
||||
}
|
||||
|
||||
@@ -29,7 +29,7 @@ func New(_name ...string) gcache.Adapter {
|
||||
case "file":
|
||||
cacheAdapterObj = drive2.NewAdapterFile("runtime/cache")
|
||||
case "es":
|
||||
cacheAdapterObj = drive.NewAdapterElasticsearch([]string{"http://127.0.0.1:9200"})
|
||||
cacheAdapterObj = drive.NewAdapterElasticsearch(_name[1])
|
||||
}
|
||||
|
||||
//var client = gcache.New()
|
||||
|
||||
@@ -2,23 +2,51 @@ package drive
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"github.com/elastic/go-elasticsearch/v8"
|
||||
"github.com/gogf/gf/v2/container/gvar"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/os/gcache"
|
||||
"github.com/gogf/gf/v2/os/gctx"
|
||||
"github.com/gogf/gf/v2/util/gconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
var (
|
||||
adapterElasticsearchClient gcache.Adapter
|
||||
)
|
||||
|
||||
type AdapterElasticsearch struct {
|
||||
//FilePath string
|
||||
Addresses []string
|
||||
client *elasticsearch.TypedClient
|
||||
name string
|
||||
}
|
||||
|
||||
func (a AdapterElasticsearch) Set(ctx context.Context, key interface{}, value interface{}, duration time.Duration) error {
|
||||
//TODO implement me
|
||||
panic("implement me")
|
||||
func (a AdapterElasticsearch) Set(ctx context.Context, _key interface{}, value interface{}, duration time.Duration) (err error) {
|
||||
key := gconv.String(_key)
|
||||
data := gconv.Map(value)
|
||||
if duration > 0 {
|
||||
data["delete_time"] = time.Now().Add(duration)
|
||||
}
|
||||
_, err = a.client.Index(a.name).Id(key).
|
||||
Document(data).Do(ctx)
|
||||
if err != nil {
|
||||
fmt.Printf("indexing document failed, err:%v\n", err)
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (a AdapterElasticsearch) SetMap(ctx context.Context, data map[interface{}]interface{}, duration time.Duration) error {
|
||||
//TODO implement me
|
||||
for k, v := range data {
|
||||
save := gconv.Map(v)
|
||||
if duration > 0 {
|
||||
save["delete_time"] = time.Now().Add(duration)
|
||||
}
|
||||
key := gconv.String(k)
|
||||
a.client.Index(a.name).Id(key).
|
||||
Document(save).Do(ctx)
|
||||
}
|
||||
|
||||
panic("implement me")
|
||||
}
|
||||
|
||||
@@ -37,9 +65,17 @@ func (a AdapterElasticsearch) SetIfNotExistFuncLock(ctx context.Context, key int
|
||||
panic("implement me")
|
||||
}
|
||||
|
||||
func (a AdapterElasticsearch) Get(ctx context.Context, key interface{}) (*gvar.Var, error) {
|
||||
//TODO implement me
|
||||
panic("implement me")
|
||||
func (a AdapterElasticsearch) Get(ctx context.Context, key interface{}) (res *gvar.Var, err error) {
|
||||
_key := gconv.String(key)
|
||||
resp, err := a.client.Get(a.name, _key).
|
||||
Do(context.Background())
|
||||
if err != nil {
|
||||
fmt.Printf("get document by id failed, err:%v\n", err)
|
||||
return
|
||||
}
|
||||
fmt.Printf("fileds:%s\n", resp.Source_)
|
||||
res = gvar.New(resp.Source_)
|
||||
return
|
||||
}
|
||||
|
||||
func (a AdapterElasticsearch) GetOrSet(ctx context.Context, key interface{}, value interface{}, duration time.Duration) (result *gvar.Var, err error) {
|
||||
@@ -82,9 +118,26 @@ func (a AdapterElasticsearch) Values(ctx context.Context) (values []interface{},
|
||||
panic("implement me")
|
||||
}
|
||||
|
||||
func (a AdapterElasticsearch) Update(ctx context.Context, key interface{}, value interface{}) (oldValue *gvar.Var, exist bool, err error) {
|
||||
//TODO implement me
|
||||
panic("implement me")
|
||||
func (a AdapterElasticsearch) Update(ctx context.Context, _key interface{}, value interface{}) (oldValue *gvar.Var, exist bool, err error) {
|
||||
key := gconv.String(_key)
|
||||
data := gconv.Map(value)
|
||||
oldValue, err = a.Get(ctx, key)
|
||||
if err != nil {
|
||||
exist = false
|
||||
} else {
|
||||
for k, v := range oldValue.Map() {
|
||||
if _, ok := data[k]; !ok {
|
||||
data[k] = v
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
_, err = a.client.Update(a.name, key).
|
||||
Doc(data).Do(context.Background())
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (a AdapterElasticsearch) UpdateExpire(ctx context.Context, key interface{}, duration time.Duration) (oldDuration time.Duration, err error) {
|
||||
@@ -98,8 +151,15 @@ func (a AdapterElasticsearch) GetExpire(ctx context.Context, key interface{}) (t
|
||||
}
|
||||
|
||||
func (a AdapterElasticsearch) Remove(ctx context.Context, keys ...interface{}) (lastValue *gvar.Var, err error) {
|
||||
//TODO implement me
|
||||
panic("implement me")
|
||||
//获取keys最后一个
|
||||
lastKey := keys[len(keys)-1]
|
||||
lastValue, _ = a.Get(ctx, lastKey)
|
||||
|
||||
for k := range keys {
|
||||
key := gconv.String(k)
|
||||
a.client.Delete(a.name, key).Do(ctx)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (a AdapterElasticsearch) Clear(ctx context.Context) error {
|
||||
@@ -112,8 +172,17 @@ func (a AdapterElasticsearch) Close(ctx context.Context) error {
|
||||
panic("implement me")
|
||||
}
|
||||
|
||||
func NewAdapterElasticsearch(addresses []string) gcache.Adapter {
|
||||
return &AdapterElasticsearch{
|
||||
Addresses: addresses,
|
||||
func NewAdapterElasticsearch(name string) gcache.Adapter {
|
||||
|
||||
if adapterElasticsearchClient == nil {
|
||||
_cfg, _ := g.Cfg().Get(gctx.New(), "elasticsearch")
|
||||
var cfg elasticsearch.Config
|
||||
_cfg.Scan(&cfg)
|
||||
es, _ := elasticsearch.NewTypedClient(cfg)
|
||||
adapterElasticsearchClient = &AdapterElasticsearch{
|
||||
client: es,
|
||||
name: name,
|
||||
}
|
||||
}
|
||||
return adapterElasticsearchClient
|
||||
}
|
||||
|
||||
@@ -16,17 +16,6 @@ type AdapterFile struct {
|
||||
}
|
||||
|
||||
func (a AdapterFile) Set(ctx context.Context, key interface{}, value interface{}, duration time.Duration) error {
|
||||
//defer a.handleLruKey(ctx, key)
|
||||
//expireTime := a.getInternalExpire(duration)
|
||||
//a.data.Set(key, memoryDataItem{
|
||||
// a: value,
|
||||
// a: expireTime,
|
||||
//})
|
||||
//c.eventList.PushBack(&adapterMemoryEvent{
|
||||
// k: key,
|
||||
// e: expireTime,
|
||||
//})
|
||||
|
||||
arr := strings.Split(":", gconv.String(key))
|
||||
fileName := path.Join(arr...)
|
||||
return gfile.PutBytes(fileName, gconv.Bytes(value))
|
||||
|
||||
@@ -19,7 +19,6 @@ func NewAdapterRedis() gcache.Adapter {
|
||||
redisObj, _ := gredis.New(cfg)
|
||||
//adapterRedisClient = gcache.NewAdapterRedis(g.Redis("default"))
|
||||
adapterRedisClient = gcache.NewAdapterRedis(redisObj)
|
||||
|
||||
adapterRedisCache.SetAdapter(adapterRedisClient)
|
||||
}
|
||||
return adapterRedisCache
|
||||
|
||||
@@ -5,96 +5,102 @@ import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"github.com/elastic/go-elasticsearch/v8"
|
||||
"github.com/elastic/go-elasticsearch/v8/typedapi/core/bulk"
|
||||
"github.com/elastic/go-elasticsearch/v8/typedapi/core/delete"
|
||||
"github.com/elastic/go-elasticsearch/v8/typedapi/core/search"
|
||||
"github.com/elastic/go-elasticsearch/v8/typedapi/core/update"
|
||||
"github.com/elastic/go-elasticsearch/v8/typedapi/types"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/os/gctx"
|
||||
)
|
||||
|
||||
var (
|
||||
es *elasticsearch.TypedClient
|
||||
)
|
||||
|
||||
type elastic struct {
|
||||
type Elastic struct {
|
||||
client *elasticsearch.TypedClient
|
||||
name string
|
||||
}
|
||||
|
||||
func New(name ...string) *elastic {
|
||||
// ES 配置
|
||||
cfg := elasticsearch.Config{
|
||||
Addresses: []string{
|
||||
"http://ay.cname.com:9200",
|
||||
},
|
||||
}
|
||||
func NewV1(name string) *Elastic {
|
||||
var cfg elasticsearch.Config
|
||||
_cfg := g.Cfg().MustGetWithEnv(gctx.New(), "elasticsearch")
|
||||
_cfg.Scan(&cfg)
|
||||
if es == nil {
|
||||
var err error
|
||||
es, err = elasticsearch.NewTypedClient(cfg)
|
||||
if err != nil {
|
||||
fmt.Printf("elasticsearch.NewTypedClient failed, err:%v\n", err)
|
||||
return &elastic{}
|
||||
return &Elastic{}
|
||||
}
|
||||
}
|
||||
return &elastic{
|
||||
return &Elastic{
|
||||
client: es,
|
||||
name: name,
|
||||
}
|
||||
}
|
||||
|
||||
// createIndex 创建索引
|
||||
func (s *elastic) CreateIndex(name string) {
|
||||
resp, err := s.client.Indices.
|
||||
Create(name).
|
||||
Do(context.Background())
|
||||
if err != nil {
|
||||
fmt.Printf("create index failed, err:%v\n", err)
|
||||
return
|
||||
}
|
||||
fmt.Printf("index:%#v\n", resp.Index)
|
||||
}
|
||||
|
||||
// indexDocument 索引文档
|
||||
func (s *elastic) IndexDocument(name string, key string, data interface{}) {
|
||||
//// Create 创建索引
|
||||
//func (s *Elastic) Create(ctx context.Context) {
|
||||
// resp, err := s.client.Indices.
|
||||
// Create(s.name).Do(ctx)
|
||||
// if err != nil {
|
||||
// fmt.Printf("create index failed, err:%v\n", err)
|
||||
// return
|
||||
// }
|
||||
// fmt.Printf("index:%#v\n", resp.Index)
|
||||
//}
|
||||
|
||||
// Set 索引文档
|
||||
func (s *Elastic) Set(ctx context.Context, key string, data interface{}) (err error) {
|
||||
// 添加文档
|
||||
resp, err := s.client.Index(name).
|
||||
Id(key).
|
||||
Document(data).
|
||||
Do(context.Background())
|
||||
if err != nil {
|
||||
fmt.Printf("indexing document failed, err:%v\n", err)
|
||||
return
|
||||
}
|
||||
fmt.Printf("result:%#v\n", resp.Result)
|
||||
}
|
||||
|
||||
// getDocument 获取文档
|
||||
func (s *elastic) GetDocument(name string, id string) (res json.RawMessage) {
|
||||
resp, err := s.client.Get(name, id).
|
||||
Do(context.Background())
|
||||
if err != nil {
|
||||
fmt.Printf("get document by id failed, err:%v\n", err)
|
||||
return
|
||||
}
|
||||
fmt.Printf("fileds:%s\n", resp.Source_)
|
||||
res = resp.Source_
|
||||
_, err = s.client.Index(s.name).Id(key).Document(data).Do(ctx)
|
||||
return
|
||||
}
|
||||
|
||||
// updateDocument 更新文档
|
||||
func (s *elastic) UpdateDocument(name string, key string, data interface{}) {
|
||||
|
||||
resp, err := s.client.Update(name, key).
|
||||
Doc(data). // 使用结构体变量更新
|
||||
Do(context.Background())
|
||||
if err != nil {
|
||||
fmt.Printf("update document failed, err:%v\n", err)
|
||||
return
|
||||
// SetBulk 批量添加文档
|
||||
func (s *Elastic) SetBulk(ctx context.Context, data []any) (err error) {
|
||||
var save *bulk.Request
|
||||
save = &bulk.Request{
|
||||
data,
|
||||
}
|
||||
fmt.Printf("result:%v\n", resp.Result)
|
||||
s.client.Bulk().Index(s.name).Request(save).Do(ctx)
|
||||
return
|
||||
}
|
||||
|
||||
// deleteDocument 删除 document
|
||||
func (s *elastic) DeleteDocument(name string, key string) {
|
||||
resp, err := s.client.Delete(name, key).
|
||||
Do(context.Background())
|
||||
// Get 获取文档
|
||||
func (s *Elastic) Get(ctx context.Context, id string) (res json.RawMessage, err error) {
|
||||
get, err := s.client.Get(s.name, id).Do(ctx)
|
||||
if err != nil {
|
||||
fmt.Printf("delete document failed, err:%v\n", err)
|
||||
return
|
||||
}
|
||||
fmt.Printf("result:%v\n", resp.Result)
|
||||
res = get.Source_
|
||||
return
|
||||
}
|
||||
|
||||
// Update 更新文档
|
||||
func (s *Elastic) Update(ctx context.Context, key string, data interface{}) (res *update.Response, err error) {
|
||||
res, err = s.client.Update(s.name, key).Doc(data).Do(ctx)
|
||||
return
|
||||
}
|
||||
|
||||
// Delete 删除 document
|
||||
func (s *Elastic) Delete(ctx context.Context, key string) (res *delete.Response, err error) {
|
||||
res, err = s.client.Delete(s.name, key).Do(ctx)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// Select 查询
|
||||
func (s *Elastic) Select(ctx context.Context, query *types.MatchAllQuery) (res *search.Response, err error) {
|
||||
res, err = s.client.Search(). //Index("my_index").
|
||||
Request(&search.Request{
|
||||
Query: &types.Query{
|
||||
MatchAll: &types.MatchAllQuery{},
|
||||
},
|
||||
}).Do(ctx)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -4,6 +4,7 @@ import (
|
||||
v1 "github.com/ayflying/utility_go/api/pkg/v1"
|
||||
"github.com/ayflying/utility_go/pkg/aycache"
|
||||
"github.com/ayflying/utility_go/pkg/config"
|
||||
"github.com/ayflying/utility_go/pkg/elasticsearch"
|
||||
"github.com/ayflying/utility_go/pkg/notice"
|
||||
"github.com/ayflying/utility_go/pkg/rank"
|
||||
"github.com/ayflying/utility_go/pkg/s3"
|
||||
@@ -38,3 +39,7 @@ func Websocket() *websocket.SocketV1 {
|
||||
func Config() *config.Cfg {
|
||||
return config.NewV1()
|
||||
}
|
||||
|
||||
func Elastic(name string) *elasticsearch.Elastic {
|
||||
return elasticsearch.NewV1(name)
|
||||
}
|
||||
|
||||
@@ -33,6 +33,7 @@ type (
|
||||
Save(actId int) (err error)
|
||||
// 清空GetRedDot缓存
|
||||
RefreshGetRedDotCache(uid int64)
|
||||
Del(uid int64, actId int)
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user