kuiper-11-扩展-function-fft

任务

进行kuiper的function扩展，增加fft处理流程。

最终目标：

在mosquitto发送float数组（[]float）消息，在kuiper中进行fft变换后，输出对应的幅值信息。

例如，输入[]float数组[1,2]，在matlab中验证结果：

所以应该输出float数组[1,3]

准备工作

fft算法的golang实现

//输入时域数据，输出振幅
func fftabs(data []float64)[]float64{
	slice:=make([]complex128,len(data))
	for i,item :=range data{
		slice[i]=complex(item,0)
	}

	fft(slice)
	ret := make([]float64,len(data))
	for i,item :=range slice{
		ret[i] = cmplx.Abs(item)
		println(ret[i])
	}
	return ret
}

//输入复数域数组，直接进行fft变换，得到改变后的复数域数组
func fft(data []complex128) {
	if len(data) == 1 {
		return
	}
	hl := len(data) / 2
	even := make([]complex128, hl)
	odd := make([]complex128, hl)
	for i := 0; i < hl; i++ {
		even[i] = data[i] + data[i+hl]
		odd[i] = (data[i] - data[i+hl]) *
			cmplx.Exp(complex(0, 2*float64(i)*math.Pi/float64(len(data))))
	}
	Fft(even)
	Fft(odd)
	for i := 0; i < hl; i++ {
		data[2*i] = even[i]
		data[2*i+1] = odd[i]
	}
}

func Fft(data []complex128) {
	fft(data)
}

编写dataIn.go

编写过程参照官方文档
主要是修改了Exec()方法

package main

import (
	"fmt"
	"math"
	"math/cmplx"
)

type dataIn struct {
}

//校验输入格式
func (f *dataIn) Validate(args []interface{}) error{
	if len(args) != 1{
		return fmt.Errorf("echo function only supports 1 parameter but got %d", len(args))
	}
	return nil
}

//处理
func (f *dataIn) Exec(args []interface{}) (interface{}, bool) {
	m :=args[0].([]float64)
	ret :=fftabs(m)
	//out:= dataIn{ret}
	return ret,true
}
//是不是xx函数
func (f *dataIn) IsAggregate() bool {
	return false
}
此处省略fft算法的实现函数，上面写过了，拷贝过来即可

var DataIn dataIn

测试

编译成.so 文件

上一次已经配置好环境，所以直接可以编译。
将.go文件放在插件项目下，执行：

go build --buildmode=plugin -o /home/zhang/gopath/src/kuiper/_build/kuiper-0.3.0-4-g9423612-linux-x86_64/plugins/functions/DataIn@v1.0.3.so functions/dataIn.go

在kuiper目录下就得到了插件文件

编写规则文件ruleDataIn.txt

{
  "id": "ruleDataIn",
  "sql": "SELECT dataIn(prarr) from streamDataIn",
  "actions": [
    {
      "log": {
      }
    }
  ]
}

启动kuiper

bin/server

定义流streamDataIn

定义一个接收1个float数组的流：

bin/cli create stream streamDataIn '(prarr array(float)) with (datasource="topicDataIn", format="json")'

定义规则ruleDataIn

bin/cli create rule ruleDataIn -f ruleDataIn.txt

发送消息：

mosquitto_pub -h localhost -t "topicDataIn" -m "{\"prarr\":[1,2]}"

结果

消息发送后，马上接收到了结果：

结果和预期相同。
查看stream.log文件：

数据成功被记录