fixes

Author: asmyasnikov

cmd/sqlc-gen-ydb-database-sql/main.go

@@ -27,21 +27,26 @@ func run(ctx context.Context) error {
 	if err != nil {
 		return err
 	}
+
 	var req pb.GenerateRequest
 	if err := proto.Unmarshal(reqBlob, &req); err != nil {
 		return fmt.Errorf("unmarshal request: %w", err)
 	}
-	resp, err := codegen.Generate(ctx, &req, codegen.DatabaseSQL)
+
+	resp, err := codegen.New(codegen.DatabaseSql).Generate(ctx, &req)
 	if err != nil {
 		return err
 	}
+
 	respBlob, err := proto.Marshal(resp)
 	if err != nil {
 		return fmt.Errorf("marshal response: %w", err)
 	}
+
 	w := bufio.NewWriter(os.Stdout)
 	if _, err := w.Write(respBlob); err != nil {
 		return err
 	}
+	
 	return w.Flush()
 }

cmd/sqlc-gen-ydb-go-sdk/main.go

@@ -27,21 +27,26 @@ func run(ctx context.Context) error {
 	if err != nil {
 		return err
 	}
+
 	var req pb.GenerateRequest
 	if err := proto.Unmarshal(reqBlob, &req); err != nil {
 		return fmt.Errorf("unmarshal request: %w", err)
 	}
-	resp, err := codegen.Generate(ctx, &req, codegen.YdbGoSDK)
+
+	resp, err := codegen.New(codegen.GoSDK).Generate(ctx, &req)
 	if err != nil {
 		return err
 	}
+
 	respBlob, err := proto.Marshal(resp)
 	if err != nil {
 		return fmt.Errorf("marshal response: %w", err)
 	}
+
 	w := bufio.NewWriter(os.Stdout)
 	if _, err := w.Write(respBlob); err != nil {
 		return err
 	}
+	
 	return w.Flush()
 }

cmd/sqlc-gen-ydb-python-sdk/main.go

@@ -27,21 +27,26 @@ func run(ctx context.Context) error {
 	if err != nil {
 		return err
 	}
+
 	var req pb.GenerateRequest
 	if err := proto.Unmarshal(reqBlob, &req); err != nil {
 		return fmt.Errorf("unmarshal request: %w", err)
 	}
-	resp, err := codegen.Generate(ctx, &req, codegen.YdbPythonSDK)
+
+	resp, err := codegen.New(codegen.PythonSDK).Generate(ctx, &req)
 	if err != nil {
 		return err
 	}
+
 	respBlob, err := proto.Marshal(resp)
 	if err != nil {
 		return fmt.Errorf("marshal response: %w", err)
 	}
+
 	w := bufio.NewWriter(os.Stdout)
 	if _, err := w.Write(respBlob); err != nil {
 		return err
 	}
+
 	return w.Flush()
 }