MySQL自定义函数(User Define Function)开发实例——发送TCP/UDP消息

来源：cnblogs　　作者：左眼水星　　时间：2024/6/26 8:53:23　　对本文有异议

开发背景

当数据库中某个字段的值改为特定值时，实时发送消息通知到其他系统。

实现思路

监控数据库中特定字段值的变化可以用数据库触发器实现。还需要实现一个自定义的函数，接收一个字符串参数，然后将这个字符传通过udp消息发送到指定端口。

在触发器中执行这个自定义函数并在其他系统中监听指定端口的消息。从而实现数据库内容变化通知到其他系统的功能。

开发前准备

MySQL自定义函数仅支持C/C++开发，所以需要一些C/C++的基础。

这里以在Windows系统使用Visual Studio2022开发进行介绍。

添加依赖项libmysql.lib、ws2_32.lib，引入头文件"mysql.h" <winsock2.h> <ws2tcpip.h>

自定义函数命名为SendG，使用方式为SendG('abcde')

MySQL自定义函数设计说明

提前说明：本文只介绍满足前面提到的需要的情况下涉及到的自定义函数设计方法。

这里要涉及两个方法

bool SendG_init(UDF_INIT* init, UDF_ARGS* args, char* message) 这个后缀_init方法显然是一个前置函数

void SendG(UDF_INIT* init, UDF_ARGS* args, char* result, unsigned long* length, char* is_null, char* error) 主体函数

当我们每次调用SendG('abcde') 首先会调用 _init后缀的函数，然后才会调用主体函数中的业务逻辑。因此我们可以在SendG_init函数中检查参数是否正确，在SendG函数中执行具体的业务逻辑。

至于UDF_INIT UDF_ARGS等结构的使用方法则不用担心，MySQL的头文件中有详细介绍，基本一看就能大概知道怎么用，例如以下是UDF_ARGS的定义：

具体实现代码

#include "pch.h"
#include <winsock2.h>
#include "mysql.h"
#include <ws2tcpip.h>
extern "C" {
    __declspec(dllexport)
        bool SendG_init(UDF_INIT* init, UDF_ARGS* args, char* message) {
        if (args->arg_count != 1) 
            return true;
        else if (args->arg_type[0] != STRING_RESULT)
            return true;
        return false;
    }
    __declspec(dllexport)
    void SendG(UDF_INIT* init, UDF_ARGS* args, char* result, unsigned long* length, char* is_null, char* error) {
        SOCKET sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
        sockaddr_in RecvAddr{};
        RecvAddr.sin_family = AF_INET;
        RecvAddr.sin_port = htons(54321);
        inet_pton(AF_INET, "127.0.0.1", &RecvAddr.sin_addr);
        *is_null = 1;
        sendto(sock, args->args[0], args->lengths[0], 0, (SOCKADDR*)&RecvAddr,sizeof(RecvAddr));
        closesocket(sock);
    }
}