kis-flow/flow/kis_config/kis_func_config.go

package flow

import (
	"fmt"
	"kis-flow/common"
)

// FParam 在当前Flow中Function定制固定配置参数类型
type FParam map[string]string

// KisSource 表示当前Function的业务源
type KisSource struct {
	Name string   `yaml:"name"` //本层Function的数据源描述
	Must []string `yaml:"must"` //source必传字段
}

// KisFuncOption 可选配置
type KisFuncOption struct {
	Cid          string `yaml:"cid"`
	RetryTimes   int    `yaml:"retry_times"`     //选填,Function调度重试(不包括正常调度)最大次数
	RetryDuriton int    `yaml:"return_duration"` //选填,Function调度每次重试最大时间间隔(单位:ms)
	Params       FParam `yaml:"default_params"`  //选填,在当前Flow中Function定制固定配置参数
}

// KisFuncConfig 一个KisFunction策略配置
type KisFuncConfig struct {
	KisType string        `yaml:"kistype"`
	Fid     string        `yaml:"fid"`
	Fname   string        `yaml:"fname"`
	Fmode   string        `yaml:"fmode"`
	Source  KisSource     `yaml:"source"`
	Option  KisFuncOption `yaml:"option"`
}

// NewFuncConfig 创建一个Function策略配置对象, 用于描述一个KisFunction信息
func NewFuncConfig(
	funcId string, funcName string, mode common.KisMode,
	source *KisSource, option *KisFuncOption) *KisFuncConfig {

	config := new(KisFuncConfig)
	config.Fid = funcId
	config.Fname = funcName

	if source == nil {
		fmt.Printf("funcName NewConfig Error, source is nil, funcName = %s\n", funcId)
		return nil
	}
	config.Source = *source

	config.Fmode = string(mode)

	//FunctionS 和 L 需要必传KisConnector参数,原因是S和L需要通过Connector进行建立流式关系
	if mode == common.S || mode == common.L {
		if option == nil {
			fmt.Printf("Funcion S/L need option->Cid\n")
			return nil
		} else if option.Cid == "" {
			fmt.Printf("Funcion S/L need option->Cid\n")
			return nil
		}
	}

	if option != nil {
		config.Option = *option
	}

	return config
}