Add ability to add comments to the allow list

2020-02-04 00:20:25 -05:00
parent 62fd1eac55
commit c85d379fe2
3 changed files with 53 additions and 27 deletions
--- a/serv/core.go
+++ b/serv/core.go
@@ -242,7 +242,7 @@ func (c *coreContext) resolveSQL() ([]byte, *stmt, error) {
 	}

 	if allowList.IsPersist() {
-		if err := allowList.Add(c.req.Vars, c.req.Query, c.req.ref); err != nil {
+		if err := allowList.Set(c.req.Vars, c.req.Query, c.req.ref); err != nil {
 			return nil, nil, err
 		}
 	}