2024-08-19



package main
 
import (
    "fmt"
    "github.com/juju/ratelimit"
    "time"
)
 
// 令牌桶限流器示例
func main() {
    // 创建一个每秒填充5个令牌的令牌桶限流器
    limiter := ratelimit.NewBucket(5, 1*time.Second)
 
    // 尝试获取令牌
    for i := 0; i < 10; i++ {
        time.Sleep(200 * time.Millisecond) // 为了更好地观察,在获取令牌前等待200毫秒
        if limiter.TakeAvailable(1) {
            fmt.Printf("Request allowed: %d\n", i)
        } else {
            fmt.Printf("Request rejected: %d\n", i)
        }
    }
}

这段代码演示了如何使用ratelimit库中的NewBucket函数创建一个令牌桶限流器,并通过TakeAvailable方法尝试获取令牌。在每次循环中,程序会等待200毫秒,然后尝试从桶中获取一个令牌。如果获取成功,则允许请求,否则拒绝请求。这种方式可以帮助我们控制数据流量和防止峰值流量打垮系统。

2024-08-19



package main
 
import (
    "fmt"
    "sync"
)
 
// 使用sync.Once实现单例模式
type singleton struct{}
 
var once sync.Once
var instance *singleton
 
func GetInstance() *singleton {
    return instance
}
 
func init() {
    instance = &singleton{}
}
 
func (s *singleton) Initialize() {
    // 初始化代码
    fmt.Println("Initialized singleton")
}
 
func main() {
    // 安全地调用初始化方法,只会执行一次
    once.Do(GetInstance().Initialize)
    // 程序其他部分...
}

这段代码展示了如何使用sync.Once来确保单例模式中的初始化代码只执行一次,即便面临多个goroutine同时调用。这是一个在Go中实现单例模式的简单方法,同时也演示了sync.Once的用法。

2024-08-19

在CentOS 7上使用yum安装Go语言环境的步骤如下:

  1. 首先,你需要添加Go的官方仓库。创建一个新的repo文件:



sudo tee /etc/yum.repos.d/golang.repo<<EOF
[golang]
name=Go Repository
baseurl=https://mirrors.aliyun.com/golang/yum/el7/
enabled=1
gpgcheck=1
gpgkey=https://mirrors.aliyun.com/golang/yum/el7/gpgkey
EOF
  1. 导入Go的公钥:



sudo rpm --import https://mirrors.aliyun.com/golang/yum/el7/gpgkey
  1. 接下来,你可以使用yum安装Go语言:



sudo yum install golang
  1. 安装完成后,你可以检查Go的版本确认安装成功:



go version
  1. 设置环境变量。编辑你的shell配置文件(比如~/.bash_profile~/.bashrc),添加以下内容:



export GOROOT=/usr/lib/golang
export GOPATH=$HOME/go
export PATH=$PATH:$GOROOT/bin:$GOPATH/bin
  1. 更新你的shell配置使其生效:



source ~/.bash_profile
# 或者
source ~/.bashrc

完成以上步骤后,Go语言应该已经成功安装并配置在你的CentOS 7系统上。

2024-08-19



package main
 
import (
    "context"
    "fmt"
    "log"
    "time"
 
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
)
 
func main() {
    clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")
    client, err := mongo.Connect(context.Background(), clientOptions)
    if err != nil {
        log.Fatal(err)
    }
    defer client.Disconnect(context.Background())
 
    collection := client.Database("testdb").Collection("testcollection")
 
    ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
    defer cancel()
 
    // 插入文档
    insertResult, err := collection.InsertOne(ctx, bson.D{{"name", "John Doe"}, {"age", 30}})
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("文档插入成功,ID: %v\n", insertResult.InsertedID)
 
    // 查询单个文档
    var result bson.M
    err = collection.FindOne(ctx, bson.D{{"name", "John Doe"}}).Decode(&result)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("查询结果: %v\n", result)
 
    // 更新文档
    updateResult, err := collection.UpdateOne(ctx, bson.D{{"name", "John Doe"}}, bson.D{{"$set", bson.D{{"age", 31}}}})
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("文档更新成功,影响行数: %v\n", updateResult.MatchedCount)
 
    // 删除文档
    deleteResult, err := collection.DeleteOne(ctx, bson.D{{"name", "John Doe"}})
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("文档删除成功,删除行数: %v\n", deleteResult.DeletedCount)
}

这段代码展示了如何使用MongoDB的Go语言驱动程序来连接到MongoDB实例,对一个集合进行插入、查询、更新和删除操作。代码简洁,注重于展示核心功能,并包含了错误处理。

2024-08-19

Golang的GMP模型是指Go程序运行时的内存管理、垃圾回收和并发调度。

  1. 内存管理:

    • 使用Tcmalloc替代glibc的malloc实现,减少内存分配的锁竞争。
    • 对象分配器,用于高效管理小对象。
    • 三色标记清除垃圾回收算法。
  2. 调度:

    • M: OS线程,负责执行Go代码。
    • P: 代表处理器,负责调度M和管理任务队列。
    • G: Go程,即Go语言中的goroutine。

GMP模型调度流程示例代码:




// 创建一个新的Goroutine
go func() {
    // 你的代码
}()
 
// 创建P
p := newP()
 
// 创建M
m := newM()
 
// 将新创建的M和P绑定
newm(m, p)
 
// 执行Goroutine
execute(gp)

在实际的Golang运行时中,这些操作都是隐式进行的,无需手动干预。开发者只需编写Go代码并让Go运行时负责管理内存和调度goroutine即可。

2024-08-19

在Go语言中,通过反射(reflect)给对象赋值是一个常见的操作。但是,反射操作可能会导致一些不确定的行为,如果不正确地处理,可能会引发自定义异常。以下是一个简单的例子,展示如何通过反射给对象赋值,并在遇到错误时抛出自定义异常。




package main
 
import (
    "errors"
    "fmt"
    "reflect"
)
 
// 自定义异常
var ErrInvalidType = errors.New("invalid type")
 
// 通过反射给对象赋值
func setValueByReflect(obj interface{}, fieldName string, value interface{}) error {
    objValue := reflect.ValueOf(obj)
    if objValue.Kind() == reflect.Ptr {
        objValue = objValue.Elem()
    }
 
    if !objValue.IsValid() {
        return errors.New("object is invalid")
    }
 
    fieldValue := objValue.FieldByName(fieldName)
    if !fieldValue.IsValid() {
        return fmt.Errorf("field %s not found", fieldName)
    }
 
    if !fieldValue.CanSet() {
        return fmt.Errorf("field %s cannot be set", fieldName)
    }
 
    // 检查值的类型是否匹配
    valueType := reflect.TypeOf(value)
    if fieldValue.Type() != valueType {
        return ErrInvalidType
    }
 
    fieldValue.Set(reflect.ValueOf(value))
    return nil
}
 
func main() {
    type example struct {
        Field string
    }
 
    ex := &example{}
    err := setValueByReflect(ex, "Field", "value")
    if err != nil {
        if err == ErrInvalidType {
            fmt.Println("Invalid type error:", err)
        } else {
            fmt.Println("Other error:", err)
        }
        return
    }
 
    fmt.Println("Value set successfully:", ex.Field)
}

在这个例子中,setValueByReflect 函数接受一个对象、字段名和要赋的值,通过反射来设置字段的值。在赋值前,它会检查类型是否匹配,如果不匹配则抛出ErrInvalidType异常。在main函数中,我们创建了一个example结构体的实例,并尝试给它的Field字段赋一个不匹配类型的值,以此来演示异常处理的流程。

2024-08-19

在Go语言中,可以使用标准库中的sort包提供的SearchInts函数来高效地判断有序数组中是否包含指定元素。如果数组未排序,可以先排序然后再使用SearchInts

以下是一个判断数组中是否包含指定元素的示例代码:




package main
 
import (
    "fmt"
    "sort"
)
 
func contains(s []int, e int) bool {
    i := sort.SearchInts(s, e)
    return i < len(s) && s[i] == e
}
 
func main() {
    array := []int{1, 2, 3, 4, 5}
    element := 3
 
    if contains(array, element) {
        fmt.Printf("%d is in the array.\n", element)
    } else {
        fmt.Printf("%d is not in the array.\n", element)
    }
}

如果数组是其他类型的切片,可以使用Search函数的泛型版本Search或者自定义搜索函数来适应特定的切片类型。

2024-08-19

在Go语言中,有多个库可以用来解析JSON,例如encoding/jsonjson-iterator/goeasyjson等。以下是使用encoding/json库解析JSON的示例代码:




package main
 
import (
    "encoding/json"
    "fmt"
    "log"
)
 
// 定义结构体,与JSON数据结构匹配
type Person struct {
    Name string `json:"name"`
    Age  int    `json:"age"`
}
 
func main() {
    // JSON数据
    jsonData := `{"name": "John", "age": 30}`
 
    // 将要解析的数据解引用
    var person Person
 
    // 解析JSON数据
    err := json.Unmarshal([]byte(jsonData), &person)
    if err != nil {
        log.Fatalf("JSON Unmarshal error: %v", err)
    }
 
    // 输出解析后的数据
    fmt.Printf("Name: %v, Age: %v\n", person.Name, person.Age)
}

这段代码首先定义了一个Person结构体,与JSON数据的结构相匹配。然后,它使用json.Unmarshal函数将JSON数据解析到结构体实例中。如果解析成功,它会打印出解析后的数据。

如果你想要一个更高效的解析方式,可以考虑使用json-iterator/go库,它号称比encoding/json更快。使用方法类似,只是导入的包不同:




import "github.com/json-iterator/go"
 
var json = jsoniter.ConfigCompatibleWithStandardLibrary
 
// 解析JSON数据
err := json.Unmarshal([]byte(jsonData), &person)
if err != nil {
    log.Fatalf("JSON Unmarshal error: %v", err)
}

在选择库时,你需要考虑项目的具体需求,比如是否需要高性能,是否需要与其他库(如easyjson)进行对比测试等。通常,对于大多数应用,encoding/json已经足够好用,但在处理非常大的JSON数据或者高性能要求的场景下,可能会考虑使用其他库。

2024-08-19

在VSCode中设置Go语言的开发环境,需要安装Go语言扩展、设置GOPATH环境变量、安装Go语言工具链,并可选择配置代理。以下是简要步骤和示例代码:

  1. 安装Go语言扩展:

    打开VSCode,通过扩展市场安装Go插件。

  2. 设置GOPATH环境变量:

    在操作系统的环境变量中设置GOPATH。

    • 在Windows上:

      
      
      
      setx GOPATH "C:\path\to\your\workspace"
    • 在Linux或macOS上:

      
      
      
      export GOPATH=/path/to/your/workspace
  3. 安装Go语言工具链:

    访问Go官方下载页面,下载并安装适合你操作系统的Go语言工具链。

  4. 配置代理(可选):

    如果你在中国大陆等地使用Go语言,可能需要配置代理以访问Google的服务。

    • 在Windows上:

      
      
      
      setx GOPROXY=https://goproxy.io,direct
    • 在Linux或macOS上:

      
      
      
      export GOPROXY=https://goproxy.io,direct
  5. 创建Go项目和文件:

    在你的GOPATH下的src目录中创建你的项目文件夹,并在其中创建Go源文件。

    
    
    
    mkdir -p $GOPATH/src/yourproject
    cd $GOPATH/src/yourproject
    touch main.go

    然后,在main.go中写入以下代码:

    
    
    
    package main
     
    import "fmt"
     
    func main() {
        fmt.Println("Hello, Go!")
    }
  6. 运行Go程序:

    在VSCode中打开终端,运行以下命令:

    
    
    
    go run main.go

    如果一切设置正确,终端将输出:"Hello, Go!"

以上步骤为在VSCode中设置Go语言开发环境的基本过程。

2024-08-19



package main
 
import (
    "fmt"
    "net"
    "syscall"
    "time"
)
 
// 定义网络IO事件类型
type Event int
 
const (
    READ Event = 1 << iota
    WRITE
)
 
// 定义网络连接结构体
type Conn struct {
    fd      int
    events  Event
    lastUse time.Time
}
 
// 创建新的网络连接
func newConn(fd int, events Event) *Conn {
    return &Conn{
        fd:      fd,
        events:  events,
        lastUse: time.Now(),
    }
}
 
// 检查连接是否可读
func (c *Conn) isReadable() bool {
    return c.events&READ == READ
}
 
// 检查连接是否可写
func (c *Conn) isWritable() bool {
    return c.events&WRITE == WRITE
}
 
// 定义事件循环结构体
type EventLoop struct {
    // 这里可以包含其他必要的字段,例如epoll文件描述符、连接池等
}
 
// 初始化事件循环
func (el *EventLoop) Init() error {
    // 初始化epoll,并设置EPOLLONESHOT,防止轮询
    epfd, err := syscall.EpollCreate1(syscall.EPOLL_CLOEXEC)
    if err != nil {
        return err
    }
    // 初始化其他字段,例如连接池等
    el.epfd = epfd
    return nil
}
 
// 添加连接到事件循环
func (el *EventLoop) AddConn(conn *Conn, events Event) error {
    // 设置非阻塞模式
    if err := setNonblock(conn.fd); err != nil {
        return err
    }
    // 添加到epoll
    event := syscall.EpollEvent{
        Fd:     int32(conn.fd),
        Events: uint32(events),
    }
    if err := syscall.EpollCtl(el.epfd, syscall.EPOLL_CTL_ADD, conn.fd, &event); err != nil {
        return err
    }
    // 添加到连接池等
    return nil
}
 
// 运行事件循环
func (el *EventLoop) Run() error {
    events := make([]syscall.EpollEvent, 10)
    for {
        nevents, err := syscall.EpollWait(el.epfd, events[:], -1)
        if err != nil {
            return err
        }
        for i := 0; i < nevents; i++ {
            // 处理事件
            conn := getConn(events[i].Fd) // 假设有方法可以根据fd获取连接
            if events[i].Events&syscall.EPOLLIN == syscall.EPOLLIN {
                // 读取数据
            }
            if events[i].Events&syscall.EPOLLOUT == syscall.EPOLLOUT {
                // 发送数据
            }
            // 更新连接的最后使用时间等
        }
    }
}
 
// 设置socket为非阻塞模式
func setNonblock(fd int) error {
    return syscall.SetNonblock(fd, true)
}
 
func main() {
    // 初始化事件循环
    eventLoop := &EventLoop{}
    if err := eventLoop.Init(); err != nil {
        fmt.Println("初始化事件循环失败:", err)
        retu