ws_split.go

/*
 MySQL UDF ws split
 */
package main

import (
	/*
		#cgo CFLAGS: -I/usr/include/mysql
		#include <stdio.h>
		#include <stdlib.h>
		#include <string.h>
		#include <mysql.h>
		#include <limits.h>
	*/
	"C"
	"github.com/shigenobu/mysql_ws_split/func"
	"unsafe"
)

// convert argc, argv into go structure
func argToGostrings(count C.uint, args **C.char, lengths *C.ulong) []string {
	// https://github.com/golang/go/wiki/cgo#turning-c-arrays-into-go-slices
	length := count
	argslice := (*[1 << 30]*C.char)(unsafe.Pointer(args))[:length:length]
	lengthsslice := (*[1 << 2]C.ulong)(unsafe.Pointer(lengths))[:length:length]

	gostrings := make([]string, count)

	for i, s := range argslice {
		l := C.int(lengthsslice[i])
		gostrings[i] = C.GoStringN(s, l)
	}
	return gostrings
}

//export ws_split_init
func ws_split_init(initid *C.UDF_INIT, args *C.UDF_ARGS, message *C.char) C.my_bool {
	length := int(args.arg_count)
	if length != 2 {
		return 1
	}

	typeslice := (*[2]C.uint)(unsafe.Pointer(args.arg_type))[:length:length]
	if typeslice[0] != C.STRING_RESULT {
		return 1
	}
	if typeslice[1] != C.STRING_RESULT {
		return 1
	}

	return 0
}

//export ws_split
func ws_split(initid *C.UDF_INIT, args *C.UDF_ARGS, result *C.char, length *C.ulong, isNull *C.char, error *C.char) *C.char {
	C.free(unsafe.Pointer(initid.ptr))

	argsString := argToGostrings(args.arg_count, args.args, args.lengths)

	// -----
	// ここが処理本体
	rslt := _func.Split(argsString[0], argsString[1])
	// -----

	initid.ptr = C.CString(rslt)
	*length = C.ulong(len(rslt))
	return initid.ptr
}

//export ws_split_deinit
func ws_split_deinit(initid *C.UDF_INIT) {
	C.free(unsafe.Pointer(initid.ptr))
}

func main() {
}