96 lines
2.1 KiB
Go
96 lines
2.1 KiB
Go
package chart
|
|
|
|
import (
|
|
"fmt"
|
|
"path/filepath"
|
|
|
|
jsoniter "github.com/json-iterator/go"
|
|
"github.com/yaoapp/gou"
|
|
"github.com/yaoapp/kun/exception"
|
|
"github.com/yaoapp/kun/log"
|
|
"github.com/yaoapp/yao/config"
|
|
"github.com/yaoapp/yao/share"
|
|
)
|
|
|
|
// Charts 已载入图表
|
|
var Charts = map[string]*Chart{}
|
|
|
|
// Load 加载数据表格
|
|
func Load(cfg config.Config) error {
|
|
if share.BUILDIN {
|
|
return LoadBuildIn("charts", "")
|
|
}
|
|
return LoadFrom(filepath.Join(cfg.Root, "charts"), "")
|
|
}
|
|
|
|
// LoadBuildIn 从制品中读取
|
|
func LoadBuildIn(dir string, prefix string) error {
|
|
return nil
|
|
}
|
|
|
|
// LoadFrom 从特定目录加载
|
|
func LoadFrom(dir string, prefix string) error {
|
|
|
|
if share.DirNotExists(dir) {
|
|
return fmt.Errorf("%s does not exists", dir)
|
|
}
|
|
|
|
err := share.Walk(dir, ".json", func(root, filename string) {
|
|
name := prefix + share.SpecName(root, filename)
|
|
content := share.ReadFile(filename)
|
|
_, err := LoadChart(content, name)
|
|
if err != nil {
|
|
log.With(log.F{"root": root, "file": filename}).Error(err.Error())
|
|
}
|
|
})
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Load Script
|
|
err = share.Walk(dir, ".js", func(root, filename string) {
|
|
name := prefix + share.SpecName(root, filename)
|
|
chart := Select(name)
|
|
if chart != nil {
|
|
script := share.ScriptName(filename)
|
|
content := share.ReadFile(filename)
|
|
chart.LoadScript(string(content), script)
|
|
}
|
|
})
|
|
|
|
return err
|
|
}
|
|
|
|
// LoadChart 载入数据表格
|
|
func LoadChart(source []byte, name string) (*Chart, error) {
|
|
chart := &Chart{
|
|
Flow: &gou.Flow{Name: name},
|
|
}
|
|
err := jsoniter.Unmarshal(source, chart)
|
|
if err != nil {
|
|
log.With(log.F{"name": name, "source": source}).Error(err.Error())
|
|
return nil, err
|
|
}
|
|
chart.Prepare()
|
|
chart.SetupAPIs()
|
|
Charts[name] = chart
|
|
return chart, nil
|
|
}
|
|
|
|
// Select 读取已加载图表
|
|
func Select(name string) *Chart {
|
|
chart, has := Charts[name]
|
|
if !has {
|
|
exception.New(
|
|
fmt.Sprintf("Chart:%s; 尚未加载", name),
|
|
400,
|
|
).Throw()
|
|
}
|
|
return chart
|
|
}
|
|
|
|
// GetData 运行 flow 返回数值
|
|
func (chart Chart) GetData(params map[string]interface{}) interface{} {
|
|
return chart.Flow.Exec(params)
|
|
}
|