forked from megastep/mysql-udf
-
Notifications
You must be signed in to change notification settings - Fork 0
/
udf_strvalformax.cc
148 lines (117 loc) · 3.37 KB
/
udf_strvalformax.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
/*
returns the strvalformax of the regression function a data set
input parameters:
dependent data (real)
independent data (string)
output:
strvalformax of the regression function (real)
registering the function:
CREATE AGGREGATE FUNCTION strvalformax RETURNS REAL SONAME 'udf_strvalformax.so';
getting rid of the function:
DROP FUNCTION strvalformax;
*/
#ifdef STANDARD
#include <stdio.h>
#include <string.h>
#ifdef __WIN__
typedef unsigned __int64 ulonglong;
typedef __int64 longlong;
#else
typedef unsigned long long ulonglong;
typedef long long longlong;
#endif /*__WIN__*/
#else
#include <my_global.h>
#include <my_sys.h>
#endif
#include <mysql.h>
#include <m_ctype.h>
#include <m_string.h>
#ifdef HAVE_DLOPEN
extern "C" {
my_bool strvalformax_init( UDF_INIT* initid, UDF_ARGS* args, char* message );
void strvalformax_deinit( UDF_INIT* initid );
void strvalformax_clear(UDF_INIT *initid, char *is_null, char *is_error);
void strvalformax_reset( UDF_INIT* initid, UDF_ARGS* args, char* is_null, char *error );
void strvalformax_add( UDF_INIT* initid, UDF_ARGS* args, char* is_null, char *error );
char* strvalformax( UDF_INIT* initid, UDF_ARGS* args,char* result,unsigned long* res_length, char* is_null, char *error );
}
struct max_data
{
double max;
char* colval;
};
my_bool strvalformax_init( UDF_INIT* initid, UDF_ARGS* args, char* message )
{
if (args->arg_count != 2)
{
strcpy(message,"wrong number of arguments: strvalformax() requires two arguments");
return 1;
}
if (args->arg_type[1]!=STRING_RESULT)
{
strcpy(message,"correlation() requires a string as parameter 2");
return 1;
}
max_data *buffer = new max_data;
buffer->max = NULL;
buffer->colval= NULL;
initid->ptr = (char*)buffer;
return 0;
}
void strvalformax_deinit( UDF_INIT* initid )
{
max_data *buffer = (max_data*)initid->ptr;
if(buffer->colval!=NULL){
free(buffer->colval);
buffer->colval=NULL;
}
delete initid->ptr;
}
void strvalformax_clear(UDF_INIT *initid, char *is_null, char *is_error)
{
max_data *buffer = (max_data*)initid->ptr;
*is_null = 0;
*is_error = 0;
buffer->max=NULL;
buffer->colval=NULL;
}
void strvalformax_reset( UDF_INIT* initid, UDF_ARGS* args, char* is_null, char* is_error )
{
strvalformax_clear(initid, is_null, is_error);
strvalformax_add( initid, args, is_null, is_error );
}
void strvalformax_add( UDF_INIT* initid, UDF_ARGS* args, char* is_null, char* is_error )
{
max_data *buffer = (max_data*)initid->ptr;
if (args->args[0]!=NULL && args->args[1]!=NULL)
{
if(buffer->max==NULL){
buffer->max = *(double*)args->args[0];
if(buffer->colval!=NULL){
free(buffer->colval);
buffer->colval=NULL;
}
buffer->colval = (char *)malloc(args->attribute_lengths[1]+1);
strcpy(buffer->colval,args->args[1]);
}else{
if((*(double*)args->args[0])>(buffer->max)){
buffer->max = *(double *)args->args[0];
if(buffer->colval!=NULL){
free(buffer->colval);
buffer->colval=NULL;
}
buffer->colval = (char *)malloc(args->attribute_lengths[1]+1);
strcpy(buffer->colval,args->args[1]);
}
}
}
}
char *strvalformax ( UDF_INIT* initid, UDF_ARGS* args,char* result,unsigned long* res_length, char* is_null, char* is_error )
{
max_data* buffer = (max_data*)initid->ptr;
result = buffer->colval;
*res_length = strlen(buffer->colval);
return result;
}
#endif