go语言回调函数的使用

前言

在 Go 语言中,回调函数是一种将一个函数作为参数传递给另一个函数,在特定的事件发生时被调用的编程模式。

一、回调函数的定义

复制代码
type OnTaskHandler func(r []byte)

type remoteTaskClient struct {
    sync.RWMutex
    onTask OnTaskHandler
}

以上定义了一个用于处理任务的结构体,可以通过设置回调函数的方式来灵活地处理不同的情况,同时使用读写锁来保证在并发环境下的安全性。

二、定义结构体方法

复制代码
func (c *remoteTaskClient) SetOnTaskHandler(onTask OnTaskHandler) {
    c.Lock()
    defer c.Unlock()
    c.onTask = onTask
}

func (c *remoteTaskClient) ProcessMessage(message []byte) {
    if c.onTask != nil {
        c.onTask(message) // 调用回调函数处理消息
    }
}

定义了两个方法一个是与回调函数绑定的方法,另一个是任务参数接收的方法。

三、主函数调用

复制代码
func main() {
    client := &remoteTaskClient{}

    // 模拟接收消息,此时没有回调函数被设置
    client.ProcessMessage([]byte("Hello, world!"))

    // 设置回调函数
    client.SetOnTaskHandler(func(data []byte) {
        fmt.Println("Received message:", string(data))
    })
    
    // 模拟接收消息,此时回调函数已被设置
    client.ProcessMessage([]byte("Hello again, world!"))
}

主函数中SetonTaskHandler方法的参数中包含了回调函数的具体实现,当接收到任务信息就会执行。

四、测试函数

复制代码
package main

import (
    "fmt"
)

type OnTaskHandler func(r []byte)

type remoteTaskClient struct {
    sync.RWMutex
    onTask OnTaskHandler
}

func (c *remoteTaskClient) SetOnTaskHandler(onTask OnTaskHandler) {
    c.Lock()
    defer c.Unlock()
    c.onTask = onTask
}

func (c *remoteTaskClient) ProcessMessage(message []byte) {
    if c.onTask != nil {
        c.onTask(message) // 调用回调函数处理消息
    }
}

func main() {
    client := &remoteTaskClient{}

    // 模拟接收消息,此时没有回调函数被设置
    client.ProcessMessage([]byte("Hello, world!"))

    // 设置回调函数
    client.SetOnTaskHandler(func(data []byte) {
        fmt.Println("Received message:", string(data))
    })
    
    // 再次模拟接收消息,此时回调函数已被设置
    client.ProcessMessage([]byte("Hello again, world!"))
}

主函数中回调函数具体通过匿名函数实现。

相关推荐
一只叫煤球的猫3 小时前
写代码很6,面试秒变菜鸟?不卖课,面试官视角走心探讨
前端·后端·面试
bobz9653 小时前
tcp/ip 中的多路复用
后端
bobz9654 小时前
tls ingress 简单记录
后端
你的人类朋友5 小时前
什么是OpenSSL
后端·安全·程序员
bobz9655 小时前
mcp 直接操作浏览器
后端
前端小张同学7 小时前
服务器部署 gitlab 占用空间太大怎么办,优化思路。
后端
databook7 小时前
Manim实现闪光轨迹特效
后端·python·动效
武子康8 小时前
大数据-98 Spark 从 DStream 到 Structured Streaming:Spark 实时计算的演进
大数据·后端·spark
该用户已不存在8 小时前
6个值得收藏的.NET ORM 框架
前端·后端·.net
文心快码BaiduComate9 小时前
文心快码入选2025服贸会“数智影响力”先锋案例
前端·后端·程序员