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!"))
}

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

相关推荐
一洽客服系统11 分钟前
网页嵌入与接入功能说明
开发语言·前端·javascript
mit6.82427 分钟前
[Meetily后端框架] Whisper转录服务器 | 后端服务管理脚本
c++·人工智能·后端·python
zhangfeng113329 分钟前
python 数据分析 单细胞测序数据分析 相关的图表,常见于肿瘤免疫微环境、细胞亚群功能研究 ,各图表类型及逻辑关系如下
开发语言·python·数据分析·医学
Sylvia-girl35 分钟前
Java---IDEA
java·开发语言·intellij-idea
Z_W_H_42 分钟前
【Springboot】Bean解释
java·开发语言
Apipost的同学们1 小时前
AI时代的接口自动化优化实践:如何突破Postman的局限性
后端·ai·架构·postman·自定义函数·apipost·api+ai
L_autinue_Star2 小时前
手写vector容器:C++模板实战指南(从0到1掌握泛型编程)
java·c语言·开发语言·c++·学习·stl
王中阳Go2 小时前
面试完第一反应是想笑
后端·go
元气小嘉2 小时前
前端技术小结
开发语言·前端·javascript·vue.js·人工智能
Livingbody2 小时前
10分钟实现基于Ubuntu25.04本地推理ERNIE模型
后端