Merge pull request #53 from gatewayd-io/method-call-counter-metrics

Method call counter metrics

Author: mostafa

plugin/metrics.go

@@ -7,6 +7,27 @@ import (
 )
 
 var (
+	GetPluginConfigCounter = promauto.NewCounter(prometheus.CounterOpts{
+		Namespace: metrics.Namespace,
+		Name:      "get_plugin_config_total",
+		Help:      "The total number of calls to the getPluginConfig method",
+	})
+	OnClosedCounter = promauto.NewCounter(prometheus.CounterOpts{
+		Namespace: metrics.Namespace,
+		Name:      "on_closed_total",
+		Help:      "The total number of calls to the onClosed method",
+	})
+	OnTrafficFromClientCounter = promauto.NewCounter(prometheus.CounterOpts{
+		Namespace: metrics.Namespace,
+		Name:      "on_traffic_from_client_total",
+		Help:      "The total number of calls to the onTrafficFromClient method",
+	})
+	OnTrafficFromServerCounter = promauto.NewCounter(prometheus.CounterOpts{
+		Namespace: metrics.Namespace,
+		Name:      "on_traffic_from_server_total",
+		Help:      "The total number of calls to the onTrafficFromServer method",
+	})
+
 	CacheHitsCounter = promauto.NewCounter(prometheus.CounterOpts{
 		Namespace: metrics.Namespace,
 		Name:      "cache_hits_total",

plugin/plugin.go

@@ -65,13 +65,15 @@ func (p *CachePlugin) GRPCClient(_ context.Context, _ *goplugin.GRPCBroker, c *g
 func (p *Plugin) GetPluginConfig(
 	_ context.Context, _ *structpb.Struct,
 ) (*structpb.Struct, error) {
+	GetPluginConfigCounter.Inc()
 	return structpb.NewStruct(PluginConfig)
 }
 
 // OnTrafficFromClient is called when a request is received by GatewayD from the client.
 func (p *Plugin) OnTrafficFromClient(
 	ctx context.Context, req *structpb.Struct,
 ) (*structpb.Struct, error) {
+	OnTrafficFromClientCounter.Inc()
 	req, err := postgres.HandleClientMessage(req, p.Logger)
 	if err != nil {
 		p.Logger.Info("Failed to handle client message", "error", err)
@@ -147,6 +149,7 @@ func (p *Plugin) OnTrafficFromClient(
 func (p *Plugin) OnTrafficFromServer(
 	ctx context.Context, resp *structpb.Struct,
 ) (*structpb.Struct, error) {
+	OnTrafficFromServerCounter.Inc()
 	resp, err := postgres.HandleServerMessage(resp, p.Logger)
 	if err != nil {
 		p.Logger.Info("Failed to handle server message", "error", err)
@@ -221,6 +224,7 @@ func (p *Plugin) OnTrafficFromServer(
 }
 
 func (p *Plugin) OnClosed(ctx context.Context, req *structpb.Struct) (*structpb.Struct, error) {
+	OnClosedCounter.Inc()
 	client := cast.ToStringMapString(sdkPlugin.GetAttr(req, "client", nil))
 	if client != nil {
 		if err := p.RedisClient.Del(ctx, client["remote"]).Err(); err != nil {