Merge pull request #1175 from anatoly32322/fix-function-id

Add go with anonymous function case in gstack

Author: asmyasnikov

CHANGELOG.md

@@ -1,4 +1,5 @@
 * Added `Struct` support for `Variant` in `ydb.ParamsBuilder()`
+* Added `go` with anonymous function case in `gstack`
 
 ## v3.61.2
 * Changed default transaction control to `NoTx` for execute query through query service client 

internal/cmd/gstack/main.go

@@ -79,6 +79,12 @@ func getCallExpressionsFromStmt(statement ast.Stmt) (listOfCallExpressions []*as
 		body = stmt.Body
 	case *ast.ForStmt:
 		body = stmt.Body
+	case *ast.GoStmt:
+		if fun, ok := stmt.Call.Fun.(*ast.FuncLit); ok {
+			listOfCallExpressions = append(listOfCallExpressions, getListOfCallExpressionsFromBlockStmt(fun.Body)...)
+		} else {
+			listOfCallExpressions = append(listOfCallExpressions, stmt.Call)
+		}
 	case *ast.RangeStmt:
 		body = stmt.Body
 	case *ast.DeclStmt:

internal/stack/function_id_test.go

@@ -1,7 +1,9 @@
 package stack
 
 import (
+	"sync"
 	"testing"
+	"time"
 
 	"github.com/stretchr/testify/require"
 )
@@ -22,6 +24,22 @@ func (e *starType) starredCall() string {
 	return FunctionID("").FunctionID()
 }
 
+func anonymousFunctionCall() string {
+	var result string
+	var mu sync.Mutex
+	go func() {
+		mu.Lock()
+		defer mu.Unlock()
+		result = FunctionID("").FunctionID()
+	}()
+	time.Sleep(time.Second)
+
+	mu.Lock()
+	defer mu.Unlock()
+
+	return result
+}
+
 func TestFunctionIDForGenericType(t *testing.T) {
 	t.Run("StaticFunc", func(t *testing.T) {
 		require.Equal(t,
@@ -42,4 +60,10 @@ func TestFunctionIDForGenericType(t *testing.T) {
 			x.starredCall(),
 		)
 	})
+	t.Run("AnonymousFunctionCall", func(t *testing.T) {
+		require.Equal(t,
+			"github.com/ydb-platform/ydb-go-sdk/v3/internal/stack.anonymousFunctionCall",
+			anonymousFunctionCall(),
+		)
+	})
 }