飞书
This commit is contained in:
hahwu
parent
e9f619d8f8
commit
283c975d10
15
feishu/.vscode/launch.json
vendored
Normal file
15
feishu/.vscode/launch.json
vendored
Normal file
@ -0,0 +1,15 @@
|
|||||||
|
{
|
||||||
|
// 使用 IntelliSense 了解相关属性。
|
||||||
|
// 悬停以查看现有属性的描述。
|
||||||
|
// 欲了解更多信息，请访问: https://go.microsoft.com/fwlink/?linkid=830387
|
||||||
|
"version": "0.2.0",
|
||||||
|
"configurations": [
|
||||||
|
{
|
||||||
|
"name": "Launch file",
|
||||||
|
"type": "go",
|
||||||
|
"request": "launch",
|
||||||
|
"mode": "debug",
|
||||||
|
"program": "${file}"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
31
feishu/Handler/P2BotMenuV6.go
Normal file
31
feishu/Handler/P2BotMenuV6.go
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"feishu/client"
|
||||||
|
"log"
|
||||||
|
|
||||||
|
larkapplication "github.com/larksuite/oapi-sdk-go/v3/service/application/v6"
|
||||||
|
)
|
||||||
|
|
||||||
|
var funcMap map[string]func(ctx context.Context, event *larkapplication.P2BotMenuV6) error
|
||||||
|
|
||||||
|
func P2Handle(ctx context.Context, event *larkapplication.P2BotMenuV6) error {
|
||||||
|
if f, ok := funcMap[*event.Event.EventKey]; ok {
|
||||||
|
f(ctx, event)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
log.Printf("no handler for event: %s", *event.Event.EventKey)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
funcMap = make(map[string]func(ctx context.Context, event *larkapplication.P2BotMenuV6) error)
|
||||||
|
funcMap["operation"] = operation
|
||||||
|
}
|
||||||
|
|
||||||
|
func operation(ctx context.Context, event *larkapplication.P2BotMenuV6) error {
|
||||||
|
openId := *event.Event.Operator.OperatorId.OpenId
|
||||||
|
client.C.SendMsg(openId, "operation")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
61
feishu/client/c.go
Normal file
61
feishu/client/c.go
Normal file
@ -0,0 +1,61 @@
|
|||||||
|
package client
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
lark "github.com/larksuite/oapi-sdk-go/v3"
|
||||||
|
larkcore "github.com/larksuite/oapi-sdk-go/v3/core"
|
||||||
|
larkim "github.com/larksuite/oapi-sdk-go/v3/service/im/v1"
|
||||||
|
)
|
||||||
|
|
||||||
|
type client struct {
|
||||||
|
*lark.Client
|
||||||
|
}
|
||||||
|
|
||||||
|
var C *client
|
||||||
|
|
||||||
|
const (
|
||||||
|
AppId = "cli_a726aa9056f1d00b"
|
||||||
|
AppSecret = "KWteGycoqAYdcqbTeCcUNbteVmpcTOCU"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
C = &client{lark.NewClient(AppId, AppSecret)}
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetClient() *client {
|
||||||
|
return C
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *client) SendMsg(open_id, content string) error {
|
||||||
|
// 创建 Client
|
||||||
|
b, _ := json.Marshal(map[string]string{"text": content})
|
||||||
|
uuidVal := uuid.New().String()
|
||||||
|
req := larkim.NewCreateMessageReqBuilder().
|
||||||
|
ReceiveIdType(`open_id`).
|
||||||
|
Body(larkim.NewCreateMessageReqBodyBuilder().
|
||||||
|
ReceiveId(open_id).
|
||||||
|
MsgType(`text`).
|
||||||
|
Content(string(b)).
|
||||||
|
Uuid(uuidVal).
|
||||||
|
Build()).
|
||||||
|
Build()
|
||||||
|
|
||||||
|
// 发起请求
|
||||||
|
resp, err := c.Im.V1.Message.Create(context.Background(), req)
|
||||||
|
|
||||||
|
// 处理错误
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
// 服务端错误处理
|
||||||
|
if !resp.Success() {
|
||||||
|
fmt.Printf("logId: %s, error response: \n%s", resp.RequestId(), larkcore.Prettify(resp.CodeError))
|
||||||
|
return fmt.Errorf("logId: %s, error response: \n%s", resp.RequestId(), larkcore.Prettify(resp.CodeError))
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
81
feishu/client/h.go
Normal file
81
feishu/client/h.go
Normal file
@ -0,0 +1,81 @@
|
|||||||
|
package client
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
|
"feishu/data"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func SendMsg(data []*data.Operation) error {
|
||||||
|
retainStr := ""
|
||||||
|
for _, v := range data {
|
||||||
|
retainStr += fmt.Sprintf("| %s | %d | %d | %.2f | %.2f%% | %.2f%% | %d | %d |\n", v.Date.Format("2006-01-02"), v.Register, v.Login, v.Recharge, float64(v.SecondRemain)/float64(v.Recharge), float64(v.ThirdRemain)/float64(v.Recharge), v.SeventhRemain, v.ThirtiethRemain)
|
||||||
|
}
|
||||||
|
str := fmt.Sprintf(`
|
||||||
|
# 日期 %s 😁
|
||||||
|
-----------------------------
|
||||||
|
## 昨日数据
|
||||||
|
|
||||||
|
- 注册：%d
|
||||||
|
- 充值：%.2f
|
||||||
|
- 登录：%d
|
||||||
|
----------------------
|
||||||
|
|
||||||
|
## 留存数据
|
||||||
|
|
||||||
|
| Date | Reg | Login | Pay | Retain2 | Retain3 | Retain7 | Retain30 |
|
||||||
|
| -------- | -------- | -------- | -------- | -------- | -------- | -------- | -------- |
|
||||||
|
%s
|
||||||
|
|
||||||
|
## 玩家数据
|
||||||
|
|
||||||
|
| Uid | Level | Exp | Diamond | Star | Energy |
|
||||||
|
| -------- | -------- | -------- | -------- | -------- | -------- |
|
||||||
|
| 1 | 2 | 3 | 4 | 5 | 6 |
|
||||||
|
| 1 | 2 | 3 | 4 | 5 | 6 |
|
||||||
|
`, time.Now().In(time.UTC).Format("2006-01-02"), data[0].Register, data[0].Recharge, data[0].Login, retainStr)
|
||||||
|
// 创建请求体
|
||||||
|
payload := map[string]interface{}{
|
||||||
|
"msg_type": "interactive",
|
||||||
|
"card": map[string]interface{}{
|
||||||
|
"type": "template",
|
||||||
|
"data": map[string]interface{}{
|
||||||
|
"template_id": "AAqBcfmUwQya1",
|
||||||
|
"template_version_name": "1.0.1",
|
||||||
|
"template_variable": map[string]interface{}{
|
||||||
|
"msg": str,
|
||||||
|
"appName": "merge_pet_london",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
payloadBytes, err := json.Marshal(payload)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 创建HTTP请求
|
||||||
|
req, err := http.NewRequest("POST", "https://open.feishu.cn/open-apis/bot/v2/hook/64bad1f3-3a41-4dca-9037-399067ffb252", bytes.NewBuffer(payloadBytes))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
req.Header.Set("Content-Type", "application/json")
|
||||||
|
|
||||||
|
// 发送请求
|
||||||
|
client := &http.Client{}
|
||||||
|
resp, err := client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
// 检查响应状态码
|
||||||
|
if resp.StatusCode != http.StatusOK {
|
||||||
|
return fmt.Errorf("failed to send message, status code: %d", resp.StatusCode)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
47
feishu/data/operation.go
Normal file
47
feishu/data/operation.go
Normal file
@ -0,0 +1,47 @@
|
|||||||
|
package data
|
||||||
|
|
||||||
|
import (
|
||||||
|
"feishu/util"
|
||||||
|
"log"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Operation struct {
|
||||||
|
Date time.Time `db:"Date"`
|
||||||
|
Register int `db:"Register"`
|
||||||
|
SecondRemain int `db:"SecondRemain"`
|
||||||
|
ThirdRemain int `db:"ThirdRemain"`
|
||||||
|
SeventhRemain int `db:"SeventhRemain"`
|
||||||
|
ThirtiethRemain int `db:"ThirtiethRemain"`
|
||||||
|
Recharge float64 `db:"Recharge"`
|
||||||
|
Login int `db:"Login"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type User struct {
|
||||||
|
Uid int `db:"Uid"`
|
||||||
|
Level int `db:"Level"`
|
||||||
|
Exp int `db:"Exp"`
|
||||||
|
Diamond int `db:"Diamond"`
|
||||||
|
Star int `db:"Star"`
|
||||||
|
Energy int `db:"Energy"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetOperation() ([]*Operation, error) {
|
||||||
|
Db, err := util.MySQL()
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("GetOperation MySQL error: %v", err)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer Db.Close()
|
||||||
|
Result := []*Operation{}
|
||||||
|
ZeroTimestamp := util.ZeroTimestampByTz("Europe/London") - 86400
|
||||||
|
ZeroTime := time.Unix(ZeroTimestamp, 0).In(time.UTC)
|
||||||
|
StartDate := ZeroTime.AddDate(0, 0, -7).Format("2006-01-02")
|
||||||
|
EndDate := ZeroTime.Format("2006-01-02")
|
||||||
|
err = Db.Select(&Result, "SELECT `Date`, `Register`, `SecondRemain`, `ThirdRemain`, `SeventhRemain`, `ThirtiethRemain`, `Recharge`, `Login` FROM remain where `Date` >= ? and `Date` <= ? order by `Date` desc", StartDate, EndDate)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("GetOperation Select error: %v", err)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return Result, nil
|
||||||
|
}
|
||||||
16
feishu/go.mod
Normal file
16
feishu/go.mod
Normal file
@ -0,0 +1,16 @@
|
|||||||
|
module feishu
|
||||||
|
|
||||||
|
go 1.23.1
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/go-sql-driver/mysql v1.8.1
|
||||||
|
github.com/google/uuid v1.6.0
|
||||||
|
github.com/jmoiron/sqlx v1.4.0
|
||||||
|
github.com/larksuite/oapi-sdk-go/v3 v3.4.9
|
||||||
|
)
|
||||||
|
|
||||||
|
require (
|
||||||
|
filippo.io/edwards25519 v1.1.0 // indirect
|
||||||
|
github.com/gogo/protobuf v1.3.2 // indirect
|
||||||
|
github.com/gorilla/websocket v1.5.0 // indirect
|
||||||
|
)
|
||||||
47
feishu/go.sum
Normal file
47
feishu/go.sum
Normal file
@ -0,0 +1,47 @@
|
|||||||
|
filippo.io/edwards25519 v1.1.0 h1:FNf4tywRC1HmFuKW5xopWpigGjJKiJSV0Cqo0cJWDaA=
|
||||||
|
filippo.io/edwards25519 v1.1.0/go.mod h1:BxyFTGdWcka3PhytdK4V28tE5sGfRvvvRV7EaN4VDT4=
|
||||||
|
github.com/go-sql-driver/mysql v1.8.1 h1:LedoTUt/eveggdHS9qUFC1EFSa8bU2+1pZjSRpvNJ1Y=
|
||||||
|
github.com/go-sql-driver/mysql v1.8.1/go.mod h1:wEBSXgmK//2ZFJyE+qWnIsVGmvmEKlqwuVSjsCm7DZg=
|
||||||
|
github.com/gogo/protobuf v1.3.2 h1:Ov1cvc58UF3b5XjBnZv7+opcTcQFZebYjWzi34vdm4Q=
|
||||||
|
github.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q=
|
||||||
|
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||||
|
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||||
|
github.com/gorilla/websocket v1.5.0 h1:PPwGk2jz7EePpoHN/+ClbZu8SPxiqlu12wZP/3sWmnc=
|
||||||
|
github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
|
||||||
|
github.com/jmoiron/sqlx v1.4.0 h1:1PLqN7S1UYp5t4SrVVnt4nUVNemrDAtxlulVe+Qgm3o=
|
||||||
|
github.com/jmoiron/sqlx v1.4.0/go.mod h1:ZrZ7UsYB/weZdl2Bxg6jCRO9c3YHl8r3ahlKmRT4JLY=
|
||||||
|
github.com/kisielk/errcheck v1.5.0/go.mod h1:pFxgyoBC7bSaBwPgfKdkLd5X25qrDl4LWUI2bnpBCr8=
|
||||||
|
github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck=
|
||||||
|
github.com/larksuite/oapi-sdk-go/v3 v3.4.9 h1:ZzsPtdF2tsLbocQdKLNFwFCzWfwmKtD4kbyx7to++M0=
|
||||||
|
github.com/larksuite/oapi-sdk-go/v3 v3.4.9/go.mod h1:ZEplY+kwuIrj/nqw5uSCINNATcH3KdxSN7y+UxYY5fI=
|
||||||
|
github.com/lib/pq v1.10.9 h1:YXG7RB+JIjhP29X+OtkiDnYaXQwpS4JEWq7dtCCRUEw=
|
||||||
|
github.com/lib/pq v1.10.9/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=
|
||||||
|
github.com/mattn/go-sqlite3 v1.14.22 h1:2gZY6PC6kBnID23Tichd1K+Z0oS6nE/XwU+Vz/5o4kU=
|
||||||
|
github.com/mattn/go-sqlite3 v1.14.22/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
|
||||||
|
github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
|
||||||
|
github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
|
||||||
|
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||||
|
golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
|
||||||
|
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
|
||||||
|
golang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
|
||||||
|
golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
|
||||||
|
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||||
|
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||||
|
golang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||||
|
golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
|
||||||
|
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
|
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||||
|
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||||
|
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
|
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||||
|
golang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=
|
||||||
|
golang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=
|
||||||
|
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
|
golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
|
golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
154
feishu/main.go
Normal file
154
feishu/main.go
Normal file
@ -0,0 +1,154 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
handler "feishu/Handler"
|
||||||
|
"feishu/client"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
larkcore "github.com/larksuite/oapi-sdk-go/v3/core"
|
||||||
|
"github.com/larksuite/oapi-sdk-go/v3/event/dispatcher"
|
||||||
|
larkapplication "github.com/larksuite/oapi-sdk-go/v3/service/application/v6"
|
||||||
|
larkim "github.com/larksuite/oapi-sdk-go/v3/service/im/v1"
|
||||||
|
larkws "github.com/larksuite/oapi-sdk-go/v3/ws"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// 注册事件回调，OnP2MessageReceiveV1 为接收消息 v2.0；OnCustomizedEvent 内的 message 为接收消息 v1.0。
|
||||||
|
eventHandler := dispatcher.NewEventDispatcher("", "").
|
||||||
|
OnP2MessageReceiveV1(func(ctx context.Context, event *larkim.P2MessageReceiveV1) error {
|
||||||
|
// fmt.Printf("[ OnP2MessageReceiveV1 access ], data: %s\n", larkcore.Prettify(event))
|
||||||
|
Content := *event.Event.Message.Content
|
||||||
|
OpenId := *event.Event.Sender.SenderId.OpenId
|
||||||
|
r := map[string]string{}
|
||||||
|
if Content != "" {
|
||||||
|
json.Unmarshal([]byte(Content), &r)
|
||||||
|
go SendMsg(OpenId, r["text"])
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}).
|
||||||
|
OnP2BotMenuV6(func(ctx context.Context, event *larkapplication.P2BotMenuV6) error {
|
||||||
|
return handler.P2Handle(ctx, event)
|
||||||
|
})
|
||||||
|
|
||||||
|
// 创建Client
|
||||||
|
cli := larkws.NewClient(client.AppId, client.AppSecret,
|
||||||
|
larkws.WithEventHandler(eventHandler),
|
||||||
|
larkws.WithLogLevel(larkcore.LogLevelDebug),
|
||||||
|
)
|
||||||
|
// 启动客户端
|
||||||
|
err := cli.Start(context.Background())
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func SendMsg(open_id, content string) {
|
||||||
|
// 创建 Client
|
||||||
|
|
||||||
|
uuidVal := uuid.New().String()
|
||||||
|
Answer := AI(content)
|
||||||
|
b, _ := json.Marshal(map[string]string{"text": Answer})
|
||||||
|
// 创建请求对象
|
||||||
|
req := larkim.NewCreateMessageReqBuilder().
|
||||||
|
ReceiveIdType(`open_id`).
|
||||||
|
Body(larkim.NewCreateMessageReqBodyBuilder().
|
||||||
|
ReceiveId(open_id).
|
||||||
|
MsgType(`text`).
|
||||||
|
Content(string(b)).
|
||||||
|
Uuid(uuidVal).
|
||||||
|
Build()).
|
||||||
|
Build()
|
||||||
|
|
||||||
|
// 发起请求
|
||||||
|
resp, err := client.C.Im.V1.Message.Create(context.Background(), req)
|
||||||
|
|
||||||
|
// 处理错误
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 服务端错误处理
|
||||||
|
if !resp.Success() {
|
||||||
|
fmt.Printf("logId: %s, error response: \n%s", resp.RequestId(), larkcore.Prettify(resp.CodeError))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type AiMessage struct {
|
||||||
|
Event string `json:"event"`
|
||||||
|
Conversation_id string `json:"conversation_id"`
|
||||||
|
Message_id string `json:"message_id"`
|
||||||
|
Create_at int `json:"create_at"`
|
||||||
|
Task_id string `json:"task_id"`
|
||||||
|
Id string `json:"id"`
|
||||||
|
Answer string `json:"answer"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func AI(message string) string {
|
||||||
|
//escapedMessage := strings.ReplaceAll(message, `"`, `\"`)
|
||||||
|
body := `{
|
||||||
|
"inputs": {},
|
||||||
|
"query": "` + message + `",
|
||||||
|
"response_mode": "streaming",
|
||||||
|
"conversation_id": "",
|
||||||
|
"user": "abc-123",
|
||||||
|
"files": []
|
||||||
|
}`
|
||||||
|
|
||||||
|
req, err := http.NewRequest("POST", "http://8.155.13.51/v1/chat-messages", strings.NewReader(body))
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("创建请求失败: %v", err)
|
||||||
|
}
|
||||||
|
req.Header.Set("Content-Type", "application/json")
|
||||||
|
req.Header.Set("Authorization", "Bearer app-14eZWBFFIyUSZi7OYrnOGhSR")
|
||||||
|
|
||||||
|
client := &http.Client{}
|
||||||
|
resp, err := client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
// handle error
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
reader := bufio.NewReader(resp.Body)
|
||||||
|
var str string
|
||||||
|
for {
|
||||||
|
var message AiMessage
|
||||||
|
line, readErr := reader.ReadString('\n')
|
||||||
|
if readErr != nil {
|
||||||
|
if readErr == io.EOF {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
log.Fatalf("读取响应时出错: %v", readErr)
|
||||||
|
}
|
||||||
|
newLine := strings.TrimPrefix(line, "data:")
|
||||||
|
|
||||||
|
// 解析 JSON 数据
|
||||||
|
if err := json.Unmarshal([]byte(newLine), &message); err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if message.Event == "message" {
|
||||||
|
str += message.Answer
|
||||||
|
}
|
||||||
|
// 处理数据
|
||||||
|
|
||||||
|
//fmt.Print("收到数据: ", line)
|
||||||
|
}
|
||||||
|
|
||||||
|
r := regexp.MustCompile(`<think>[\s\S]*?<\/think>`)
|
||||||
|
str = r.ReplaceAllString(str, "")
|
||||||
|
str = strings.TrimLeft(str, "\n")
|
||||||
|
if str == "" {
|
||||||
|
str = "我不知道你在说什么"
|
||||||
|
}
|
||||||
|
return str
|
||||||
|
}
|
||||||
91
feishu/unit_test.go
Normal file
91
feishu/unit_test.go
Normal file
@ -0,0 +1,91 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"encoding/json"
|
||||||
|
"feishu/client"
|
||||||
|
"feishu/data"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
type message struct {
|
||||||
|
Event string `json:"event"`
|
||||||
|
Conversation_id string `json:"conversation_id"`
|
||||||
|
Message_id string `json:"message_id"`
|
||||||
|
Create_at int `json:"create_at"`
|
||||||
|
Task_id string `json:"task_id"`
|
||||||
|
Id string `json:"id"`
|
||||||
|
Answer string `json:"answer"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCard(t *testing.T) {
|
||||||
|
data, err := data.GetOperation()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("GetOperation error: %v", err)
|
||||||
|
}
|
||||||
|
err = client.SendMsg(data)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("SendMsg error: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ...existing code...
|
||||||
|
func TestHTTPPost(t *testing.T) {
|
||||||
|
body := `{
|
||||||
|
"inputs": {},
|
||||||
|
"query": "游戏订单提交失败",
|
||||||
|
"response_mode": "streaming",
|
||||||
|
"conversation_id": "",
|
||||||
|
"user": "abc-123",
|
||||||
|
"files": []
|
||||||
|
}`
|
||||||
|
|
||||||
|
req, err := http.NewRequest("POST", "http://8.155.13.51/v1/chat-messages", strings.NewReader(body))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("创建请求失败: %v", err)
|
||||||
|
}
|
||||||
|
req.Header.Set("Content-Type", "application/json")
|
||||||
|
req.Header.Set("Authorization", "Bearer app-14eZWBFFIyUSZi7OYrnOGhSR")
|
||||||
|
|
||||||
|
client := &http.Client{}
|
||||||
|
resp, err := client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
// handle error
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
reader := bufio.NewReader(resp.Body)
|
||||||
|
var str string
|
||||||
|
for {
|
||||||
|
var message message
|
||||||
|
line, readErr := reader.ReadString('\n')
|
||||||
|
if readErr != nil {
|
||||||
|
if readErr == io.EOF {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
t.Fatalf("读取响应时出错: %v", readErr)
|
||||||
|
}
|
||||||
|
newLine := strings.TrimPrefix(line, "data:")
|
||||||
|
|
||||||
|
// 解析 JSON 数据
|
||||||
|
if err := json.Unmarshal([]byte(newLine), &message); err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if message.Event == "message" {
|
||||||
|
str += message.Answer
|
||||||
|
}
|
||||||
|
// 处理数据
|
||||||
|
|
||||||
|
//fmt.Print("收到数据: ", line)
|
||||||
|
}
|
||||||
|
|
||||||
|
r := regexp.MustCompile(`<think>[\s\S]*?<\/think>`)
|
||||||
|
str = r.ReplaceAllString(str, "")
|
||||||
|
|
||||||
|
fmt.Print("收到数据: ", str)
|
||||||
|
}
|
||||||
34
feishu/util/msyql.go
Normal file
34
feishu/util/msyql.go
Normal file
@ -0,0 +1,34 @@
|
|||||||
|
package util
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
_ "github.com/go-sql-driver/mysql"
|
||||||
|
"github.com/jmoiron/sqlx"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
MYSQL_USER = "root"
|
||||||
|
MYSQL_PASS = "Z4rf7eZZe500dxa"
|
||||||
|
MYSQL_HOST = "rm-f8zd2030feam53n43io.mysql.rds.aliyuncs.com"
|
||||||
|
MYSQL_PORT = 3306
|
||||||
|
MYSQL_DATABASE = "merge_pet_london"
|
||||||
|
)
|
||||||
|
|
||||||
|
func MySQL() (*sqlx.DB, error) {
|
||||||
|
dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local", MYSQL_USER, MYSQL_PASS, MYSQL_HOST, MYSQL_PORT, MYSQL_DATABASE)
|
||||||
|
db, err := sqlx.Open("mysql", dsn)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return db, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func MYSQL2() (*sqlx.DB, error) {
|
||||||
|
dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local", MYSQL_USER, MYSQL_PASS, "8.208.47.208", MYSQL_PORT, "merge_pet_london_1")
|
||||||
|
db, err := sqlx.Open("mysql", dsn)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return db, nil
|
||||||
|
}
|
||||||
10
feishu/util/util.go
Normal file
10
feishu/util/util.go
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
package util
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
func ZeroTimestampByTz(tz string) int64 {
|
||||||
|
loc, _ := time.LoadLocation(tz)
|
||||||
|
now := time.Now().In(loc)
|
||||||
|
midnight := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location())
|
||||||
|
return midnight.Unix()
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user