Add CI flow

Rename codegeneration.go -> main.go
ydb-platform · Mar 23, 2024 · de98eb7 · de98eb7
1 parent eca954d
commit de98eb7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/internal/cmd/gstack/main.go b/internal/cmd/gstack/main.go
@@ -58,11 +58,10 @@ func getExprFromDeclStmt(statement *ast.DeclStmt) (listOfExpressions []ast.Expr)
 	}
 	for _, spec := range decl.Specs {
 		if spec, ok := spec.(*ast.ValueSpec); ok {
-			for _, expr := range spec.Values {
-				listOfExpressions = append(listOfExpressions, expr)
-			}
+			listOfExpressions = append(listOfExpressions, spec.Values...)
 		}
 	}
+
 	return listOfExpressions
 }
 

diff --git a/internal/query/session.go b/internal/query/session.go
@@ -119,7 +119,7 @@ func (s *Session) attach(ctx context.Context) (finalErr error) {
 		onDone(finalErr)
 	}()
 
-	attachCtx, cancelAttach := xcontext.WithCancel(xcontext.WithoutDeadline(ctx))
+	attachCtx, cancelAttach := xcontext.WithCancel(xcontext.ValueOnly(ctx))
 
 	attach, err := s.grpcClient.AttachSession(attachCtx, &Ydb_Query.AttachSessionRequest{
 		SessionId: s.id,