分类 "Golang" 下的文章

最近文章快凉凉了,大家记得给文章点赞支持~谢谢大家的喜欢!关注我,分享有趣的知识共同学习!

AI 大模型太火爆了!工作中经常使用,感觉已经离不开了大模型了!最近测试了智谱AI大模型,实测下来真的很好用!

今天的股市太绿了,我让它给我写个文案,没想到还给股民按摩了哈哈!

你是一名文学家,请你为我创作文案,主题是 2024 年,A 股走势绿得可怕。要求:语言流畅通顺,观点独特,行文风骚。

2024年,A股走势绿意盎然,恍若春日里的翠竹,节节败退。那抹绿,如同一幅泼墨山水,挥毫之间,惊心动魄。
这一年,股市的绿意犹如江南的梅雨,绵延不绝,投资者们的心情也随着这绿意,起伏不定。有人调侃:“绿”成了心头大患,也有人戏言:“要想生活过得去,头上总得带点绿。”
然而,在这片绿海之中,智者看到了机遇,勇者捕捉到了逆袭的曙光。绿得可怕,恰似风雨前的乌云,预示着一场翻天覆地的变革即将来临。
风骚独步的投资者,把这片绿色视为一种挑战,一种磨砺。他们深知,在这片绿意盎然的A股市场,唯有独具慧眼,方能洞察先机,把握那稍纵即逝的盈利机会。
绿得可怕,是因为市场在洗牌,在重塑。那些站在风口浪尖的企业,或将涅槃重生,或将黯然离场。而作为投资者的我们,唯有紧跟时代步伐,拥抱变革,方能在这片绿色的股市中,破茧成蝶,翩翩起舞。
2024年,A股走势绿得可怕,但这绿色背后,却蕴藏着无尽的机遇与挑战。让我们以独特的视角,审视这片绿色的股市,携手共进,书写属于我们的辉煌篇章。

我平时使用 Go 语言,但我发现官方只给了 Python 和 Java 的接口!

Python: https://github.com/zhipuai/zhipuai-sdk-python-v4

Java: https://github.com/zhipuai/zhipuai-sdk-java-v4

这对于我们 Go 语言开发者来说,确实有点不方便。

翻遍了全球最大的男性交友平台——GitHub!Go 语言一个 SDK 都没有!难道要自己手写?

对我这种喜欢偷懒的人来说,是不可能自己写的。

我坏了十个键盘,每一个都是坏了 Ctrl C V 这三个键。

研究了下接口文档(智谱AI开放平台),发现智谱 AI 的 v4 版本接口和 OpenAI 的接口长得挺像的!

那就用 go-openai 这个包试试水!GitHub - sashabaranov/go-openai: OpenAI ChatGPT, GPT-3, GPT-4, DALL·E, Whisper API wrapper for Go

折腾下来,发现两个问题:

第一个问题是,智谱 AI 的接口认证要用 jwt 生成 token,这个在网上一搜,没人做过。

根据文档,我自己写了个函数,得到 Authorization 头部后,传给 go-openai 作为请求的 Header 头发出去!

import "github.com/golang-jwt/jwt/v5"

// apikey 为智谱提供的 API Key
func splitAPIKey(apikey string) (string, string) {
    parts := strings.Split(apikey, ".")
    if len(parts) != 2 {
        return "", ""
    }
    return parts[0], parts[1]
}

func getAuthorization(authToken string) string {
    id, secret := splitAPIKey(authToken)
    payload := jwt.MapClaims{
        "api_key":   id,
        "exp":       time.Now().Add(3 * time.Minute).UnixMilli(),
        "timestamp": time.Now().UnixMilli(),
    }
    token := jwt.NewWithClaims(jwt.SigningMethodHS256, payload)
    token.Header["sign_type"] = "SIGN"
    res, _ := token.SignedString([]byte(secret))
    return res
}

第二个问题是,你能通过认证,但调用失败!!!这个我折腾了一下午,总算找到了罪魁祸首!

排查的方式是,将我的请求复制成 cURL 到终端,发现是能正常响应的!!!

而 go-openai 包在底层发送请求的时候,会自己加上以下头部:

    req.Header.Set("Content-Type", "application/json")
    req.Header.Set("Accept", "text/event-stream")
    req.Header.Set("Cache-Control", "no-cache")
    req.Header.Set("Connection", "keep-alive")

    resp, err := client.config.HTTPClient.Do(req) //nolint:bodyclose // body is closed in stream.Close()
    if err != nil {
        return new(streamReader[T]), err
    }

智谱 AI 服务器端,是不支持 Accept: text/event-stream 的!!!

感兴趣的同学可以自己测试下,只要带上 Accept 头就会报错,即便返回接口就是 text/event-stream,原因不明。

那该如何解决这个问题呢?我发现 go-openai 包里的 requestBuilder 是个 interface,但可惜没有暴露出来!

如果它暴露出来了,我可以套娃后替换它!不过请求头是在 requestBuilder 之后,所以这个方案只能放弃。

但它提供了自定义的 HTTPClient!

而 HTTPClient 最终会请求到 client.config.HTTPClient.Do(req) 方法!

所以用 RoundTrip(*Request) (*Response, error) 接口截获请求就完事了!

怎么搞呢?灰常简单:

type removeAcceptHeaderRoundTrip struct{}

func (s *removeAcceptHeaderRoundTrip) RoundTrip(req *http.Request) (resp *http.Response, err error) {
    req.Header.Del("Accept")
    resp, err = http.DefaultTransport.RoundTrip(req)
    return
}

然后初始化的时候,换成我们的 RoundTrip 就好了!

    cfg := openai.DefaultConfig(authorization)
    cfg.BaseURL = "https://open.bigmodel.cn/api/paas/v4"
    cfg.HTTPClient.Transport = &removeAcceptHeaderRoundTrip{} // 换成我们的 RoundTrip
    cli := openai.NewClientWithConfig(cfg)
    stream, err := cli.CreateChatCompletionStream(ctx, openai.ChatCompletionRequest{
        // Model: "glm-3-turbo",
        Model: "glm-4",
        Messages: []openai.ChatCompletionMessage{
            {Role: openai.ChatMessageRoleUser, Content: input},
        },
    })

是不是很惊艳!!!interface 大法好!抽象得当,有时候写代码真的就是几分钟的事情!!!

不得不说 Go 的标准库是真的舒服,简单翻一下代码,就知道怎么写了,1 分钟的事情干完下班!!!

(如果觉得文章有帮助,建议收藏起来,方便下次阅读)

之前写文章的时候,有人喷我用 interface 用 wire 把代码搞得太复杂了……

仁扬:我是如何组织 Go 代码的(目录结构 依赖注入 wire)

现在这个就是活生生的例子……

假如没有暴露 RoundTrip 接口,那你什么都做不了,你只能硬质头皮自己写 SDK 写反射写各种恶心的黑科技……

总之,写代码抽象思维很重要,需要时刻注意代码的领域和边界!!!

夜已深,我把完整的代码贴上来吧,大家自行食用:

package main

import (
    "context"
    "errors"
    "fmt"
    "io"
    "net/http"
    "strings"
    "time"

    "github.com/golang-jwt/jwt/v5"
    "github.com/sashabaranov/go-openai"
)

func main() {
    input := `你是一名文学家,请你为我创作文案,主题是2024年,A股走势绿得可怕。要求:语言流畅通顺,观点独特,行文风骚。
`
    ctx := context.Background()
    authToken := "......"
    authorization := getAuthorization(authToken)
    cfg := openai.DefaultConfig(authorization)
    cfg.BaseURL = "https://open.bigmodel.cn/api/paas/v4"
    cfg.HTTPClient.Transport = &removeAcceptHeaderRoundTrip{} // 替换成我们自己的
    cli := openai.NewClientWithConfig(cfg)
    stream, err := cli.CreateChatCompletionStream(ctx, openai.ChatCompletionRequest{
        // Model: "glm-3-turbo",
        Model: "glm-4",
        Messages: []openai.ChatCompletionMessage{
            {Role: openai.ChatMessageRoleUser, Content: input},
        },
    })
    if err != nil {
        fmt.Println(err.Error())
        return
    }
    defer stream.Close()

    fmt.Print(input)
    fmt.Println("")

    for {
        response, e := stream.Recv()
        if errors.Is(e, io.EOF) {
            break
        }
        if e != nil {
            break
        }
        content := response.Choices[0].Delta.Content
        fmt.Print(content)
    }

    fmt.Println("")

    return
}

func getAuthorization(authToken string) string {
    id, secret := splitAPIKey(authToken)
    payload := jwt.MapClaims{
        "api_key":   id,
        "exp":       time.Now().Add(3 * time.Minute).UnixMilli(),
        "timestamp": time.Now().UnixMilli(),
    }
    token := jwt.NewWithClaims(jwt.SigningMethodHS256, payload)
    token.Header["sign_type"] = "SIGN"
    res, _ := token.SignedString([]byte(secret))
    return res
}

func splitAPIKey(apikey string) (string, string) {
    parts := strings.Split(apikey, ".")
    if len(parts) != 2 {
        return "", ""
    }
    return parts[0], parts[1]
}

type removeAcceptHeaderRoundTrip struct{}

func (s *removeAcceptHeaderRoundTrip) RoundTrip(req *http.Request) (resp *http.Response, err error) {
    req.Header.Del("Accept")
    resp, err = http.DefaultTransport.RoundTrip(req)
    return
}

如果觉得文章有帮助,建议收藏起来,方便下次阅读。

喜欢我的文章可以关注我,分享有趣的知识共同学习!

想不想知道 e5 洋垃圾的性能?好家伙,我手痒了!!!

前阵子我用 Go 写了一个小工具 —— GoHyperPi!通过计算圆周率 π 的值,测试 CPU 的单核和多核性能。

代码是开源的,可以移步到 GitHub 查看 → https://github.com/Lofanmi/GoHyperPi

与传统的性能测试软件 SuperPi(只能单核)、HyperPi(支持多核但不跨平台)相比,

GoHyperPi 能更充分利用多核处理器的并行计算能力,跨平台地进行性能测试,以更全面和准确地评估计算机的处理器性能。

为什么我要写这个软件呢?

1、Super PI 这个软件是单核的,而 HyperPi 支持多核但是又不跨平台;

2、Hyper PI 使用的人偏少,跑分数据不全面;

3、我组装了双路 E5-2690 v4,我想知道它的性能是怎么样的!!!

而 Go 语言有调度器有协程,写一个并行计算的程序,对我来说这不和呼吸一样简单?

嘿嘿,仅用 200 行代码不到~

它的原理是什么?

1、-n 指定圆周率小数点后的位数。我的测试都是计算圆周率 10 万位;

2、-proc 并行数量,默认程序自动获取 CPU 的核心数量(超线程会当做真正的核心)。程序会启动一定数量的协程,计算圆周率,并测量时间;

3、-times 并行倍数。最终的跑分结果为小数点第几位每秒,倍数越高得分越高,默认 2 倍。

废话不多说,先看跑分!双路 E5-2690 v4 洋垃圾,28 核心 56 线程,单核跑分 1838.56,多核跑分 102608.14!

阅读全文

今天线上出现了一个很奇怪的问题,简单讲就是某个比较通用的服务反序列化 JSON 的时候,发现有些 int64 的 ID 解开之后不是预期的数字。

废话不多说!直接上代码!不看答案,你能知道输出什么吗?

package main

import (
    "encoding/json"
    "fmt"
    "log"
    "sync"
    "time"
)

type Msg struct {
    DrId int `json:"drid"`
}

type Data struct {
    Msg interface{} `json:"msg"`
}

func main() {
    msg := Msg{DrId: 21041079142844926}
    data := Data{Msg: msg}
    str2, _ := json.Marshal(data)
    // 1. {"msg":{"drid":21041079142844926}}
    fmt.Println(string(str2))
    var data2 Data
    _ = json.Unmarshal(str2, &data2)
    // fmt.Println(data2)
    str3, _ := json.Marshal(data2)
    // 2. ?
    fmt.Println(string(str3))
}

阅读全文

五一长假,大家都出去玩了。

独自一人在家闲着没事,心血来潮开撸日志查询系统——Gobana(Gobana = Golang + Kibana)!

是的,它使用 Go 语言开发,我希望可以替代日常使用的 Kibana!

Kibana 其实有挺多缺点,简单的日志查询写语句还是比较简单的,但是对于更多关键词,你需要写括号 (),或者是逻辑运算符 ANDOR

仔细想想,它其实并不是必须的:

因为,大部分情况下,我们查日志

都是需要命中某些关键词,或者是排除某些关键词

所以如果提供输入框,约定好查询的格式,我想就可以节省很多时间了!

欢迎参观,我会抽空开发,项目地址:https://github.com/Lofanmi/gobana/

6 月啦,补充一张图片上来~

Gobana

规划中的功能:

  • 支持 Elasticsearch
  • 支持阿里云 SLS 查询
  • 支持 Kibana-Proxy 工作模式(如果运维同学没有 Elasticsearch 地址,只提供 Kibana 的权限的话)【首创!!!】
  • 支持图表展示
  • 按照业务归类常用的查询语句
  • 内嵌 Lua 解析引擎,理论上支持任意类型的日志!【首创!!!】

预计 6 月份发初版,敬请期待吧。