Mirrors/miller
1
0
Fork 0
mirror of https://github.com/johnkerl/miller.git synced 2026-07-22 15:37:59 +00:00
Code Issues Projects Releases Wiki Activity

emit-write iterate

Browse source
This commit is contained in:
John Kerl 2016-07-20 19:42:40 -04:00
parent 017c9eebd2
commit 8e4002358d
2 changed files with 18 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

15
c/mapping/mlr_dsl_cst.c
View file

@ -551,6 +551,7 @@ static mlr_dsl_cst_statement_t* alloc_blank() {
pstatement->prhs_evaluator = NULL;
pstatement->poutput_filename_evaluator = NULL;
pstatement->pmulti_out = NULL;
pstatement->pmulti_lrec_writer = NULL;
pstatement->poosvar_rhs_keylist_evaluators = NULL;
pstatement->pemit_oosvar_namelist_evaluators = NULL;
pstatement->pvarargs = NULL;
@ -852,6 +853,7 @@ static mlr_dsl_cst_statement_t* alloc_emitf_write(mlr_dsl_ast_node_t* pnode, int
pstatement->poutput_filename_evaluator = rval_evaluator_alloc_from_ast(pfilenode,
type_inferencing, context_flags);
pstatement->pmulti_lrec_writer = multi_lrec_writer_alloc();
pstatement->pnode_handler = handle_emitf_write;
return pstatement;
@ -1830,6 +1832,10 @@ static void cst_statement_free(mlr_dsl_cst_statement_t* pstatement) {
}
multi_out_free(pstatement->pmulti_out);
if (pstatement->pmulti_lrec_writer != NULL) {
// xxx drain all
multi_lrec_writer_free(pstatement->pmulti_lrec_writer);
}
if (pstatement->poosvar_rhs_keylist_evaluators != NULL) {
for (sllve_t* pe = pstatement->poosvar_rhs_keylist_evaluators->phead; pe != NULL; pe = pe->pnext) {
@ -2245,6 +2251,7 @@ static void handle_emitf_write(
mv_t filename = poutput_filename_evaluator->pprocess_func(poutput_filename_evaluator->pvstate, pvars);
lrec_t* prec_to_emit = lrec_unbacked_alloc();
sllv_t* poutrecs = sllv_alloc();
for (sllve_t* pf = pnode->pvarargs->phead; pf != NULL; pf = pf->pnext) {
mlr_dsl_cst_statement_vararg_t* pvararg = pf->pvvalue;
char* emitf_or_unset_srec_field_name = pvararg->emitf_or_unset_srec_field_name;
@ -2264,7 +2271,13 @@ static void handle_emitf_write(
}
}
sllv_append(pcst_outputs->poutrecs, prec_to_emit);
sllv_append(poutrecs, prec_to_emit);
// xxx to-string ...
// xxx flush param ...
multi_lrec_writer_write(pnode->pmulti_lrec_writer, poutrecs, filename.u.strv, TRUE/*flush_every_record*/);
sllv_free(poutrecs);
mv_free(&filename);
}

6
c/mapping/mlr_dsl_cst.h
View file

@ -7,6 +7,7 @@
#include "containers/bind_stack.h"
#include "containers/loop_stack.h"
#include "output/multi_out.h"
#include "output/multi_lrec_writer.h"
// ================================================================
// Concrete syntax tree (CST) derived from an abstract syntax tree (AST).
@ -120,9 +121,10 @@ typedef struct _mlr_dsl_cst_statement_t {
// Assignments to srec or oosvar, as well as the boolean expression in filter, cond, and bare-boolean
rval_evaluator_t* prhs_evaluator;
// For 'print > filename_expression, value_expression'
// For print-to-file and dump-to-file, and emit-to-file
rval_evaluator_t* poutput_filename_evaluator;
multi_out_t* pmulti_out;
multi_out_t* pmulti_out; // print-to-file and dump-to-file
multi_lrec_writer_t* pmulti_lrec_writer; // emit-to-file
// Assigning full srec from oosvar:
sllv_t* poosvar_rhs_keylist_evaluators;