Improve code

.github/workflows/test.yaml

@@ -47,4 +47,10 @@ jobs:
           install-mode: "goinstall"
 
       - name: Run tests 🧪
-        run: make test
+        run: go test -p 1 -cover -covermode atomic -coverprofile=profile.cov -v ./...
+
+      # Uncomment the following lines to upload the coverage report to Coveralls
+      # - name: Upload coverage report 📈
+      #   uses: shogo82148/actions-goveralls@v1
+      #   with:
+      #     path-to-profile: profile.cov

README.md

@@ -32,3 +32,11 @@ Running the above command causes the `go mod tidy` and `go build` to run for com
 ## Sentry
 
 This plugin uses [Sentry](https://sentry.io) for error tracking. Sentry can be configured using the `SENTRY_DSN` environment variable. If `SENTRY_DSN` is not set, Sentry will not be used. -->
+
+## Contributing
+
+We welcome contributions from everyone.<!-- Please read our [contributing guide](https://gatewayd-io.github.io/CONTIBUTING.md) for more details.--> Just open an [issue](https://github.com/gatewayd-io/gatewayd-plugin-sql-ids-ips/issues) or send us a [pull request](https://github.com/gatewayd-io/gatewayd-plugin-sql-ids-ips/pulls).
+
+## License
+
+This plugin is licensed under the [Affero General Public License v3.0](https://github.com/gatewayd-io/gatewayd-plugin-sql-ids-ips/blob/main/LICENSE).

gatewayd_plugin.yaml

@@ -8,7 +8,6 @@ reloadOnCrash: True
 timeout: 30s
 
 plugins:
-  # Plugin name
   - name: gatewayd-plugin-sql-ids-ips
     # whether to enable or disable the plugin on the next run
     enabled: True

plugin/plugin.go

@@ -29,7 +29,6 @@ const (
 	ResponseField     string = "response"
 	OutputsField      string = "outputs"
 	TokensField       string = "tokens"
-	RequestField      string = "request"
 	StringField       string = "String"
 
 	DeepLearningModel string = "deep_learning_model"
@@ -102,13 +101,7 @@ func (p *Plugin) OnTrafficFromClient(ctx context.Context, req *v1.Struct) (*v1.S
 	req, err := postgres.HandleClientMessage(req, p.Logger)
 	if err != nil {
 		p.Logger.Debug("Failed to handle client message", ErrorField, err)
-  return req, err
-	}
-
-	// Get the client request from the GatewayD request.
-	request := cast.ToString(sdkPlugin.GetAttr(req, RequestField, ""))
-	if request == "" {
-		return req, nil
+		return req, err
 	}
 
 	// Get the query from the request.

plugin/plugin_test.go

@@ -78,7 +78,6 @@ func Test_OnTrafficFromClinet(t *testing.T) {
 
 	server := httptest.NewServer(
 		http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-			fmt.Println(r.URL.Path)
 			switch r.URL.Path {
 			case TokenizeAndSequencePath:
 				w.WriteHeader(http.StatusOK)
@@ -156,7 +155,6 @@ func Test_OnTrafficFromClinetFailedTokenization(t *testing.T) {
 
 	server := httptest.NewServer(
 		http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-			fmt.Println(r.URL.Path)
 			switch r.URL.Path {
 			case TokenizeAndSequencePath:
 				w.WriteHeader(http.StatusInternalServerError)
@@ -228,7 +226,6 @@ func Test_OnTrafficFromClinetFailedPrediction(t *testing.T) {
 
 	server := httptest.NewServer(
 		http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-			fmt.Println(r.URL.Path)
 			switch r.URL.Path {
 			case TokenizeAndSequencePath:
 				w.WriteHeader(http.StatusOK)