本文主要介绍如何使用go编写一个命令行工具,具体一些细节不在本文分析

编写一个简单的curl命令行工具

使用到go的 flag库

例如

第一个参数为命令行参数，第二个默认值GET 第三个usage提示输入信息

flag.String("X", "GET", "GET POST")

代码如下

package main

import (
    "bufio"
    "bytes"
    "encoding/json"
    "errors"
    "flag"
    "fmt"
    "io"
    "net/http"
    "net/url"
    "time"
)

func main() {

    // 命令行参数 参数名称前面加个 -
    // 第一个参数为命令行参数，第二个默认值GET 第三个usage提示信息 
    method := flag.String("X", "GET", "GET POST")
    // 如果时GET请求不需要写body
    body := flag.String("d", "{}", "json format")
    flag.Parse()

    // 没有 - 的参数,但是需要放到最末尾
    args := flag.Args()
    if len(args) != 1 {
        // 约定只有一个这样子的参数，且必须是url
        fmt.Println("No parameter is passed, the url must be passed at the end")
        return
    }
    // 创建一个httpclient
    client := NewHttpClient()

    switch *method {
    // 只做GET POST支持
    case "POST", "GET":
        // 将body参数转成map 必须要是json格式
        param := make(map[string]any)
        err := json.Unmarshal([]byte(*body), &param)
        if err != nil {
            fmt.Println("Json format is not valid", err.Error())
            return
        }

        data, err := client.Json(*method, args[0], param)
        if err != nil {
            fmt.Println(err.Error())
            return
        }
        // json格式化打印
        result := bytes.Buffer{}
        _ = json.Indent(&result, data, "", "\t")
        fmt.Println(result.String())
    default:
        fmt.Printf("No support %s", *method)
        return
    }
}

// HttpClient http客户端
type HttpClient struct {
    client  http.Client
    Address string
}

func NewHttpClient() *HttpClient {
    client := http.Client{
        Timeout: time.Duration(3) * time.Second,
        Transport: &http.Transport{
            MaxIdleConnsPerHost:   5,
            MaxConnsPerHost:       100,
            IdleConnTimeout:       90 * time.Second,
            TLSHandshakeTimeout:   10 * time.Second,
            ExpectContinueTimeout: 1 * time.Second,
        },
    }
    return &HttpClient{client: client}
}

// Json 接收json格式返回
func (c *HttpClient) Json(method string, url string, args map[string]any) ([]byte, error) {
    jsonStr, _ := json.Marshal(args)
    req, err := http.NewRequest(method, url, bytes.NewReader(jsonStr))
    req.Header.Set("Content-Type", "application/json")
    if err != nil {
        return nil, err
    }
    return c.handleResponse(req)
}

func (c *HttpClient) handleResponse(req *http.Request) ([]byte, error) {
    // go原生方法 执行http请求
    response, err := c.client.Do(req)
    if err != nil {
        return nil, err
    }

    if response.StatusCode != http.StatusOK {
        return nil, errors.New(fmt.Sprintf("response code is %d", response.StatusCode))
    }

    // 以带缓冲字节的方式读取body 与其它语言类似
    buff := make([]byte, 128)
    var body []byte
    reader := bufio.NewReader(response.Body)
    for {
        n, err := reader.Read(buff)

        if err != nil && err != io.EOF {
            return nil, err
        }

        body = append(body, buff[:n]...)

        // 读完了
        if err == io.EOF || n == 0 {
            break
        }

        if n < 128 {
            break
        }

    }
    defer response.Body.Close()
    return body, nil
}

// 将map转成go中的url.Values
func (c *HttpClient) toValues(args map[string]any) string {
    if args != nil && len(args) > 0 {
        params := url.Values{}
        for k, v := range args {
            params.Set(k, fmt.Sprintf("%v", v))
        }
        return params.Encode()
    }
    return ""
}

打开终端命令行执行 go build -o gocurl.exe .

会在当前项目生成 gocurl.exe的windows可执行文件，如果时其它操作系统可以起其它后缀

D:\GoProject\curl> go build  -o gocurl.exe  .   

帮助信息 --help 或者 -h

1667407279798.png

准备两个服务

一个GET请求一个POST，这里需要大家自己准备,我这里准备的两个服务postman调用结果如下

1667324887023.png

1667324953997.png

测试

使用终端执行gocurl.exe

1667325133066.png

PS D:\GoProject\curl> .\gocurl.exe -X=GET http://localhost:7001/product/get?pid=1001
{
        "code": "200",
        "msg": "ok",
        "data": {
                "pid": "1001",
                "pName": "iphone14-7676",
                "count": 0,
                "PType": ""
        }
}
PS D:\GoProject\curl> .\gocurl.exe -X=POST -d='{\"pid\":\"1002\"}' http://localhost:7001/product/post
{
        "code": "200",
        "msg": "ok",
        "data": {
                "pid": "1002",
                "pName": "iphone14-7676",
                "count": 0,
                "PType": ""
        }
}

好了，一个简单的命令行工具已经开法完成。大家也可以尝试着去开发自己的小工具。编译成对应系统的可执行文件。比如开发一个并发测试工具 ab可以并发请求url并且计算qps等信息。