使用starlet简化Starlark脚本执行与数据转换

Starlark是一种类似Python的配置语言，常用于构建系统如Bazel。在Go生态中，starlark-go是官方实现，而starlet是基于它构建的简化插件库，提供了更友好的API和实用功能。

starlet核心功能

starlet主要提供以下便利功能：

1. 简化脚本执行流程
2. 自动类型转换(Go ↔ Starlark)
3. 预置常用模块(math, time, json等)
4. 错误处理优化

安装

go get github.com/1set/starlet

基础使用示例

package main

import (
	"fmt"
	"github.com/1set/starlet"
)

func main() {
	// 创建执行器
	vm := starlet.New()

	// 执行简单脚本
	result, err := vm.Execute("1 + 2 * 3")
	if err != nil {
		fmt.Printf("执行错误: %v\n", err)
		return
	}
	fmt.Printf("结果: %v\n", result) // 输出: 7
}

变量传递与获取

func main() {
	vm := starlet.New()
	
	// 设置变量到Starlark环境
	err := vm.Set("name", "Alice")
	if err != nil {
		panic(err)
	}
	
	// 执行脚本使用变量
	result, err := vm.Execute("'Hello, ' + name + '!'")
	if err != nil {
		panic(err)
	}
	fmt.Println(result) // 输出: Hello, Alice!
	
	// 从Starlark环境获取变量
	value, err := vm.Get("name")
	if err != nil {
		panic(err)
	}
	fmt.Printf("变量值: %v\n", value) // 输出: Alice
}

复杂数据类型处理

starlet自动处理Go与Starlark类型转换：

func main() {
	vm := starlet.New()
	
	// 传递结构体数据
	data := map[string]interface{}{
		"name":    "Bob",
		"age":     30,
		"hobbies": []string{"reading", "coding"},
	}
	
	err := vm.Set("user", data)
	if err != nil {
		panic(err)
	}
	
	// 执行脚本处理复杂数据
	script := `
user_info = "Name: {}, Age: {}, Hobbies: {}".format(
		user["name"],
		user["age"],
		", ".join(user["hobbies"])
	)
	`
	
	_, err = vm.Execute(script)
	if err != nil {
		panic(err)
	}
	
	// 获取处理结果
	info, err := vm.Get("user_info")
	if err != nil {
		panic(err)
	}
	fmt.Println(info) // 输出: Name: Bob, Age: 30, Hobbies: reading, coding
}

使用预置模块

starlet预加载了多个实用模块：

func main() {
	vm := starlet.New()
	
	// 使用time模块
	script := `
import time
now = time.now()
formatted = time.format(now, "2006-01-02 15:04:05")
	`
	
	_, err := vm.Execute(script)
	if err != nil {
		panic(err)
	}
	
	formatted, err := vm.Get("formatted")
	if err != nil {
		panic(err)
	}
	fmt.Printf("当前时间: %v\n", formatted)
	
	// 使用math模块
	result, err := vm.Execute("math.sqrt(16)")
	if err != nil {
		panic(err)
	}
	fmt.Printf("平方根: %v\n", result) // 输出: 4
}

函数调用

可以在Go中定义函数供Starlark调用：

func main() {
	vm := starlet.New()
	
	// 注册Go函数
	err := vm.Set("greet", func(name string) string {
		return fmt.Sprintf("Hello, %s!", name)
	})
	if err != nil {
		panic(err)
	}
	
	// 调用Go函数
	result, err := vm.Execute("greet('Charlie')")
	if err != nil {
		panic(err)
	}
	fmt.Println(result) // 输出: Hello, Charlie!
}

错误处理

starlet提供了更友好的错误处理：

func main() {
	vm := starlet.New()
	
	// 执行有错误的脚本
	_, err := vm.Execute("1 + 'a'")
	if err != nil {
		if starErr, ok := err.(*starlet.StarletError); ok {
			fmt.Printf("Starlark错误: %v\n", starErr.Message)
			fmt.Printf("位置: %v\n", starErr.Pos)
		} else {
			fmt.Printf("其他错误: %v\n", err)
		}
	}
}

性能考虑

对于需要多次执行的场景，可以预编译脚本：

func main() {
	vm := starlet.New()
	
	// 预编译脚本
	script := `
def calculate(x):
	return x * 2 + 1
	`
	
	_, err := vm.Execute(script)
	if err != nil {
		panic(err)
	}
	
	// 多次调用预编译的函数
	for i := 0; i < 5; i++ {
		result, err := vm.Execute(fmt.Sprintf("calculate(%d)", i))
		if err != nil {
			panic(err)
		}
		fmt.Printf("calculate(%d) = %v\n", i, result)
	}
}

starlet通过简化API和自动类型转换，使得在Go中集成Starlark脚本变得更加容易，特别适合需要灵活配置或规则引擎的场景。