2024-08-11



package main
 
import (
    "context"
    "fmt"
    "github.com/apache/rocketmq-client-go/v2"
    "github.com/apache/rocketmq-client-go/v2/consumer"
    "github.com/apache/rocketmq-client-go/v2/primitive"
    "time"
)
 
func main() {
    // 创建RocketMQ Producer
    producer, err := rocketmq.NewProducer(
        producer.WithGroupName("test_group"),
        producer.WithNameServer([]string{"127.0.0.1:9876"}),
    )
    if err != nil {
        fmt.Println(err)
        return
    }
 
    // 启动Producer
    err = producer.Start()
    if err != nil {
        fmt.Println(err)
        return
    }
 
    // 创建RocketMQ Consumer
    consumer, err := rocketmq.NewPushConsumer(
        consumer.WithGroupName("test_group"),
        consumer.WithNameServer([]string{"127.0.0.1:9876"}),
    )
    if err != nil {
        fmt.Println(err)
        return
    }
 
    // 订阅主题
    err = consumer.Subscribe(
        "TopicTest",
        consumer.MessageSelector{},
        func(context context.Context, msg primitive.Message) (consumer.ConsumeResult, error) {
            fmt.Printf("Received message: %s\n", msg.Body)
            return consumer.ConsumeSuccess, nil
        },
    )
    if err != nil {
        fmt.Println(err)
        return
    }
 
    // 启动Consumer
    err = consumer.Start()
    if err != nil {
        fmt.Println(err)
        return
    }
 
    // 发送消息
    msg := &primitive.Message{
        Topic: "TopicTest",
        Body:  []byte("Hello RocketMQ"),
    }
    res, err := producer.SendSync(context.Background(), msg)
    if err != nil {
        fmt.Println(err)
    } else {
        fmt.Printf("Send message success, result:%v\n", res)
    }
 
    // 等待一段时间以便Consumer接收和处理消息
    time.Sleep(10 * time.Second)
 
    // 关闭Producer和Consumer
    err = producer.Shutdown()
    if err != nil {
        fmt.Println(err)
    }
    err = consumer.Shutdown()
    if err != nil {
        fmt.Println(err)
    }
}

这段代码展示了如何在Go语言中创建和启动RocketMQ的Producer和Consumer,并且如何发送和接收消息。代码中包含了错误处理,确保在出错时能够打印错误信息并优雅地关闭资源。

2024-08-11



package main
 
import (
    "fmt"
    "github.com/gin-gonic/gin"
)
 
func main() {
    // 设置Gin为发布模式
    gin.SetMode(gin.ReleaseMode)
 
    // 创建一个Gin路由器
    r := gin.Default()
 
    // 组路由分组
    v1 := r.Group("/v1")
    {
        // 路由处理器 - 获取路径参数
        v1.GET("/posts/:id", func(c *gin.Context) {
            // 获取路径参数
            id := c.Param("id")
            fmt.Printf("路径参数: id = %s\n", id)
        })
 
        // 路由处理器 - 获取查询字符串参数
        v1.GET("/posts", func(c *gin.Context) {
            // 获取查询字符串参数,如果不存在则返回默认值-1
            page := c.DefaultQuery("page", "-1")
            fmt.Printf("查询字符串参数: page = %s\n", page)
        })
    }
 
    // 启动服务器
    r.Run(":8080")
}

这段代码定义了一个简单的Gin服务器,它有两个路由处理器,分别用于获取路径参数和查询字符串参数。在实际的Web服务中,这些参数通常用于查询数据库或者进行下一步的操作。

2024-08-11

在Redis中,compareStringObjects函数用于比较两个Redis字符串对象的内容。如果两个对象相同,则返回1;如果不同,则返回0。

以下是compareStringObjects函数的核心实现:




/* Compares two string objects. Return 1 if equal, 0 otherwise. */
int compareStringObjects(robj *o1, robj *o2) {
    return (o1->encoding == OBJ_ENCODING_RAW &&
            o2->encoding == OBJ_ENCODING_RAW) ?
           (strcmp(o1->ptr,o2->ptr) == 0) :
           (memcmp(o1->ptr,o2->ptr,sdslen(o1->ptr)) == 0);
}

这个函数首先检查两个对象是否都是原始编码(raw encoding),如果是,则使用strcmp函数来比较它们的字符串指针;如果不是,则使用memcmp函数来比较它们的字符串内容,直到找到不同的字节或到达字符串的末尾。

这个函数是Redis内部用于比较字符串对象的实现,它被许多其他函数所调用,以确定对象的相等性或进行其他操作。

2024-08-11



package main
 
import (
    "gorm.io/driver/sqlite"
    "gorm.io/gorm"
)
 
// 定义用户结构体
type User struct {
    gorm.Model
    Username string
    Age      int
    Email    string
}
 
func main() {
    // 连接数据库
    db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
    if err != nil {
        panic("数据库连接失败")
    }
 
    // 自动迁移模式,确保表结构最新
    db.AutoMigrate(&User{})
 
    // 添加用户示例
    db.Create(&User{Username: "张三", Age: 25, Email: "zhangsan@example.com"})
 
    // 查询用户示例
    var users []User
    db.Find(&users)
 
    // 输出用户信息
    for _, user := range users {
        println(user.Username, user.Age, user.Email)
    }
}

这段代码首先定义了一个User结构体,用于表示用户信息。然后,它创建了一个SQLite数据库的连接,自动迁移User结构体对应的数据库表,并演示了如何添加和查询用户信息。这个过程展示了Golang中使用Gorm进行数据库操作的基本步骤。

2024-08-11



package main
 
import "fmt"
 
// 二分查找函数
func search(nums []int, target int) int {
    left, right := 0, len(nums)-1
    for left <= right {
        mid := left + (right-left)/2
        if nums[mid] == target {
            return mid
        } else if nums[mid] < target {
            left = mid + 1
        } else {
            right = mid - 1
        }
    }
    return -1
}
 
func searchBitonicSequence(nums []int, target int) int {
    n := len(nums)
    left, right := 0, n-1
    while left <= right {
        mid := left + (right-left)/2
        if nums[mid] == target {
            return mid
        }
        // 当前点不是target,则可能在递增或递减的序列里
        // 如果中点大于下一个元素,说明我们在递增序列里,继续在左半边查找
        if mid+1 < n && nums[mid] > nums[mid+1] {
            right = mid - 1
        } else {
            left = mid + 1
        }
    }
    return -1
}
 
func main() {
    nums := []int{1, 3, 8, 12, 15, 19, 20}
    target := 15
    fmt.Println(searchBitonicSequence(nums, target)) // 输出 4
}

这段代码首先定义了一个二分查找函数search,然后定义了searchBitonicSequence函数来处理位于峰顶序列中的目标值的查找。searchBitonicSequence函数通过检查中点是否在递增序列的边界来决定是在左半边还是右半边继续查找,直到找到目标值或确定目标值不在数组中。最后在main函数中通过样例数组和目标值进行测试。

2024-08-11



package main
 
import (
    "fmt"
    "runtime"
    "sync"
    "sync/atomic"
)
 
func main() {
    var (
        wg           sync.WaitGroup
        writeIndex   uint64
        data         [8]uint64 // 假设我们有8个元素的数组,模拟缓存行
        cacheLineSize = uint64(runtime.CacheLineSize)
    )
 
    // 假设我们有8个goroutine,每个都在写入数组
    wg.Add(8)
    for i := 0; i < 8; i++ {
        go func() {
            defer wg.Done()
            for {
                // 通过原子操作更新索引,保证多个goroutine安全更新同一个缓存行
                index := atomic.AddUint64(&writeIndex, cacheLineSize) % uint64(len(data))
                data[index] = index // 模拟写入操作
 
                // 假设达到某个条件退出循环
                if data[0] == cacheLineSize*8 {
                    break
                }
            }
        }()
    }
 
    wg.Wait() // 等待所有goroutine完成
    fmt.Println("Done")
}

这段代码模拟了8个并发的goroutine,每个都在不同的缓存行上进行写操作。通过保证索引的更新是原子操作,我们确保了一个缓存行不会同时被多个goroutine访问,这样就能够有效地利用局部性原理,减少缓存冲突。

2024-08-11

Go语言的goroutine是由Go运行时的调度器进行调度管理的。Go的调度器采用了协同式调度(G-P-M模型),其中G表示goroutine,P表示processor(处理器),M表示machine(线程)。

调度器的主要工作原理如下:

  1. 当一个M线程启动时,它会在自己的本地队列中寻找可运行的G。
  2. 如果本地队列为空或者本地队列长度不足,M线程会从全局运行队列中抢取G。
  3. 一旦M线程找到了G,它会将G绑定到自己并开始运行该goroutine。
  4. 当G执行了一段时间或者调用了runtime.Gosched()主动放弃CPU使用权时,M线程会将G放回到全局运行队列,然后再从队列中选择G执行。
  5. 如果全局队列为空,但是P列表中的处理器不足以满足需求,调度器会创建新的线程(M)来帮助处理。

下面是一个简单的例子,演示了如何创建goroutine:




package main
 
import (
    "fmt"
    "runtime"
)
 
func printNumbers() {
    for i := 1; i <= 3; i++ {
        fmt.Println(i)
    }
}
 
func main() {
    // 设置CPU核心数
    runtime.GOMAXPROCS(1)
 
    fmt.Println("Starting main go routine.")
 
    // 创建一个goroutine
    go printNumbers()
 
    // 模拟其他工作
    for i := 1; i <= 3; i++ {
        fmt.Println("Doing some work in main go routine.")
    }
 
    // 确保主goroutine不会提前退出
    var input string
    fmt.Scanln(&input)
}

在这个例子中,我们创建了一个goroutine来执行printNumbers函数,并通过runtime.GOMAXPROCS(1)设置了运行时的CPU核心数为1,以保证在单核心环境中能清晰地观察到调度的行为。

2024-08-11

以下是一个简化的代码实例,展示了如何使用Go语言创建一个简单的分布式网络爬虫:




package main
 
import (
    "fmt"
    "github.com/corpix/uarand"
    "github.com/temoto/robotstxt"
    "log"
    "net/http"
    "net/url"
    "sync"
)
 
var (
    wg sync.WaitGroup
)
 
func main() {
    // 初始化User-Agent列表
    uarand.Init("./ua.txt")
 
    // 设置robots.txt的处理器
    robots, err := robotstxt.NewRobotstxt("http://www.example.com/robots.txt")
    if err != nil {
        log.Fatal(err)
    }
 
    // 初始化一个任务队列
    queue := make(chan string)
 
    // 启动一个goroutine来生产URL
    go func() {
        for i := 0; i < 100; i++ {
            queue <- fmt.Sprintf("http://www.example.com/page-%d", i)
        }
        close(queue)
    }()
 
    // 启动多个goroutine来消费URL
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            for u := range queue {
                if !robots.TestAgent(uarand.Get(), u) {
                    continue
                }
                crawl(u)
            }
        }()
    }
 
    wg.Wait()
}
 
func crawl(u string) {
    req, err := http.NewRequest("GET", u, nil)
    if err != nil {
        log.Print(err)
        return
    }
 
    req.Header.Set("User-Agent", uarand.Get())
    resp, err := http.DefaultClient.Do(req)
    if err != nil {
        log.Print(err)
        return
    }
    defer resp.Body.Close()
 
    if resp.StatusCode == http.StatusOK {
        // 这里可以添加解析响应体的代码
        fmt.Printf("Crawled %s\n", u)
    } else {
        log.Printf("Received status code %d for %s\n", resp.StatusCode, u)
    }
}

这个简化的代码实例展示了如何使用Go语言创建一个简单的分布式网络爬虫。它初始化了一个用户代理列表,检查了robots.txt,并启动了一个goroutine来生产待爬取的URL。然后,它启动了多个goroutine来消费URL,并执行爬取操作。这个例子教会开发者如何在Go中实现并发网络爬虫的基本概念。

2024-08-11

在Go语言中,有一些编程知识点和注意事项,这里列举一些常见的概念和相关代码示例:

  1. 变量声明和初始化



var a int = 10      // 声明并初始化变量a
var b = 20          // 类型推导,初始化变量b
c := 30             // 简短变量声明,初始化变量c
  1. 指针



i := 10
p := &i // 获取变量i的内存地址,p是指向i的指针
fmt.Println(*p) // 使用*p来访问p指向的值,输出10
  1. 数组和切片



arr := [5]int{1, 2, 3, 4, 5} // 声明一个固定大小的数组
slice := []int{1, 2, 3, 4, 5} // 声明一个切片,底层是数组
  1. 函数



func add(a, b int) int { // 声明一个函数,接受两个int参数,返回一个int值
    return a + b
}
  1. 错误处理



_, err := os.Open("does_not_exist.txt")
if err != nil {
    fmt.Println(err)
}
  1. 并发



var wg sync.WaitGroup
wg.Add(2)
go func() {
    fmt.Println("Hello from the first goroutine")
    wg.Done()
}()
go func() {
    fmt.Println("Hello from the second goroutine")
    wg.Done()
}()
wg.Wait() // Wait for the goroutines to finish
  1. 结构体和方法



type Rectangle struct {
    width, height int
}
 
func (r Rectangle) Area() int {
    return r.width * r.height
}
 
rect := Rectangle{width: 10, height: 20}
fmt.Println(rect.Area()) // 输出结果200
  1. 接口



type Shape interface {
    Area() int
}
 
type Rectangle struct {
    width, height int
}
 
func (r Rectangle) Area() int {
    return r.width * r.height
}
 
func printArea(s Shape) {
    fmt.Println(s.Area())
}
 
rect := Rectangle{width: 10, height: 20}
printArea(rect) // 输出结果200

这些代码示例展示了Go语言中的一些基本概念和语法,同时也提醒开发者注意相关的编程实践和错误处理。

2024-08-11



package main
 
import (
    "fmt"
    "github.com/gin-gonic/gin"
)
 
func main() {
    router := gin.Default()
 
    // 处理GET请求
    router.GET("/get", func(c *gin.Context) {
        c.JSON(200, gin.H{
            "message": "Hello, World!",
        })
    })
 
    // 处理POST请求
    router.POST("/post", func(c *gin.Context) {
        var json struct {
            Name string `json:"name"`
        }
 
        // 绑定请求体中的JSON到结构体
        if err := c.ShouldBindJSON(&json); err != nil {
            c.JSON(400, gin.H{
                "error": err.Error(),
            })
            return
        }
 
        c.JSON(200, gin.H{
            "message": fmt.Sprintf("Hello, %s!", json.Name),
        })
    })
 
    // 启动服务器
    router.Run(":8080")
}

这段代码使用Gin框架创建了一个简单的Web服务器,它可以同时处理GET和POST请求。对于GET请求,它返回一个简单的JSON响应;对于POST请求,它尝试解析请求体中的JSON并返回一个个性化的JSON响应。这个例子展示了如何使用Gin来快速创建RESTful API。