-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #10 from vearne/develop
Develop
- Loading branch information
Showing
7 changed files
with
189 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Empty file.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,127 @@ | ||
| package rule | ||
|
|
||
| import ( | ||
| "github.com/go-resty/resty/v2" | ||
| "github.com/vearne/zaplog" | ||
| "github.com/yuin/gopher-lua" | ||
| "go.uber.org/zap" | ||
| luajson "layeh.com/gopher-json" | ||
| "strconv" | ||
| "sync" | ||
| ) | ||
|
|
||
| var L *lua.LState | ||
|
|
||
| /* | ||
| Lua virtual machine is not thread-safe | ||
| so we need LuaVMLock to protect L | ||
| */ | ||
| var LuaVMLock sync.Mutex | ||
|
|
||
| func init() { | ||
| L = lua.NewState() | ||
| //defer L.Close() | ||
| // register json lib | ||
| luajson.Preload(L) | ||
| registerHttpRespType(L) | ||
| } | ||
|
|
||
| type HttpResp struct { | ||
| Code string | ||
| Body string | ||
| } | ||
|
|
||
| const luaHttpRespTypeName = "HttpResp" | ||
|
|
||
| var httpRespMethods = map[string]lua.LGFunction{ | ||
| "code": getSetCode, | ||
| "body": getSetBody, | ||
| } | ||
|
|
||
| // Getter and setter for the HttpResp#Code | ||
| func getSetCode(L *lua.LState) int { | ||
| p := checkHttpResp(L) | ||
| if L.GetTop() == 3 { | ||
| p.Code = L.CheckString(2) | ||
| return 0 | ||
| } | ||
| L.Push(lua.LString(p.Code)) | ||
| return 1 | ||
| } | ||
|
|
||
| // Getter and setter for the HttpResp#Body | ||
| func getSetBody(L *lua.LState) int { | ||
| p := checkHttpResp(L) | ||
| if L.GetTop() == 3 { | ||
| p.Body = L.CheckString(3) | ||
| return 0 | ||
| } | ||
| L.Push(lua.LString(p.Body)) | ||
| return 1 | ||
| } | ||
|
|
||
| // Checks whether the first lua argument is a *LUserData with *Person and returns this *Person. | ||
| func checkHttpResp(L *lua.LState) *HttpResp { | ||
| ud := L.CheckUserData(1) | ||
| if v, ok := ud.Value.(*HttpResp); ok { | ||
| return v | ||
| } | ||
| L.ArgError(1, "HttpResp expected") | ||
| return nil | ||
| } | ||
|
|
||
| // Constructor | ||
| func newHttpResp(L *lua.LState) int { | ||
| resp := &HttpResp{L.CheckString(1), L.CheckString(2)} | ||
| ud := L.NewUserData() | ||
| ud.Value = resp | ||
| L.SetMetatable(ud, L.GetTypeMetatable(luaHttpRespTypeName)) | ||
| L.Push(ud) | ||
| return 1 | ||
| } | ||
|
|
||
| // Registers my person type to given L. | ||
| func registerHttpRespType(L *lua.LState) { | ||
| mt := L.NewTypeMetatable(luaHttpRespTypeName) | ||
| L.SetGlobal("HttpResp", mt) | ||
| // static attributes | ||
| L.SetField(mt, "new", L.NewFunction(newHttpResp)) | ||
| // methods | ||
| L.SetField(mt, "__index", L.SetFuncs(L.NewTable(), httpRespMethods)) | ||
| } | ||
|
|
||
| type HttpLuaRule struct { | ||
| LuaStr string `json:"lua"` | ||
| } | ||
|
|
||
| func (r *HttpLuaRule) Name() string { | ||
| return "HttpLuaRule" | ||
| } | ||
|
|
||
| func (r *HttpLuaRule) Verify(resp *resty.Response) bool { | ||
| L.SetGlobal("codeStr", lua.LString(strconv.Itoa(resp.StatusCode()))) | ||
| L.SetGlobal("bodyStr", lua.LString(resp.String())) | ||
|
|
||
| source := r.LuaStr + | ||
| ` | ||
| r = HttpResp.new(codeStr, bodyStr); | ||
| return verify(r); | ||
| ` | ||
| if err := runLuaStr(source); err != nil { | ||
| zaplog.Error("HttpLuaRule-Verify", | ||
| zap.Int("status", resp.StatusCode()), | ||
| zap.String("body", resp.String()), | ||
| zap.String("LuaStr", r.LuaStr), | ||
| zap.Error(err)) | ||
| return false | ||
| } | ||
| lv := L.Get(-1) | ||
| return lv == lua.LTrue | ||
| } | ||
|
|
||
| func runLuaStr(source string) error { | ||
| LuaVMLock.Lock() | ||
| defer LuaVMLock.Unlock() | ||
|
|
||
| return L.DoString(source) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,43 @@ | ||
| package rule | ||
|
|
||
| import ( | ||
| "github.com/go-resty/resty/v2" | ||
| "github.com/stretchr/testify/assert" | ||
| lua "github.com/yuin/gopher-lua" | ||
| "testing" | ||
| ) | ||
|
|
||
| func TestLua(t *testing.T) { | ||
| if err := runLuaStr(` | ||
| function verify(r) | ||
| local json = require "json"; | ||
| local car = json.decode(r:body()); | ||
| return r:code() == "200" and car.age == 10 and car.name == "buick"; | ||
| end | ||
| r = HttpResp.new("200", "{\"age\": 10,\"name\": \"buick\"}") | ||
| return verify(r) | ||
| `); err != nil { | ||
| panic(err) | ||
| } | ||
|
|
||
| result := false | ||
| lv := L.Get(-1) | ||
| if lv == lua.LTrue { | ||
| result = true | ||
| } | ||
| assert.True(t, result) | ||
| } | ||
|
|
||
| func TestHttpLuaRule(t *testing.T) { | ||
| luaStr := ` | ||
| function verify(r) | ||
| local json = require "json"; | ||
| local person = json.decode(r:body()); | ||
| return person.age == 10 and person.name == "John"; | ||
| end | ||
| ` | ||
| var resp resty.Response | ||
| resp.SetBody([]byte(`{"age": 10, "name": "John"}`)) | ||
| rule := HttpLuaRule{LuaStr: luaStr} | ||
| assert.True(t, rule.Verify(&resp)) | ||
| } |