+ ENV

helper/env.process.go

@@ -0,0 +1,22 @@
+package helper
+
+import (
+	"os"
+
+	"github.com/yaoapp/gou"
+)
+
+// ProcessEnvGet  xiang.helper.EnvGet 读取ENV
+func ProcessEnvGet(process *gou.Process) interface{} {
+	process.ValidateArgNums(1)
+	name := process.ArgsString(0)
+	return os.Getenv(name)
+}
+
+// ProcessEnvSet  xiang.helper.EnvSet 设置ENV
+func ProcessEnvSet(process *gou.Process) interface{} {
+	process.ValidateArgNums(2)
+	name := process.ArgsString(0)
+	value := process.ArgsString(1)
+	return os.Setenv(name, value)
+}

helper/env_test.go

@@ -0,0 +1,15 @@
+package helper
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/yaoapp/gou"
+)
+
+func TestProcessEnv(t *testing.T) {
+	err := gou.NewProcess("xiang.helper.EnvSet", "XIANG_UNIT_TEST", "FOO").Run()
+	assert.Nil(t, err)
+	test := gou.NewProcess("xiang.helper.EnvGet", "XIANG_UNIT_TEST").Run().(string)
+	assert.Equal(t, "FOO", test)
+}

helper/process.go

@@ -33,6 +33,9 @@ func init() {
 	gou.RegisterProcessHandler("xiang.helper.Case", ProcessCase)
 	gou.RegisterProcessHandler("xiang.helper.IF", ProcessIF)
 
+	gou.RegisterProcessHandler("xiang.helper.EnvSet", ProcessEnvSet)
+	gou.RegisterProcessHandler("xiang.helper.EnvGet", ProcessEnvGet)
+
 	gou.RegisterProcessHandler("xiang.helper.Print", ProcessPrint)
 
 }