| Message ID | 20190129150159.10588-8-alban.gruin@gmail.com (mailing list archive) |
|---|---|
| State | New, archived |
| Headers | show |
| Series | sequencer: refactor functions working on a todo_list | expand |
Hi Alban On 29/01/2019 15:01, Alban Gruin wrote: > This refactors sequencer_add_exec_commands() to work on a todo_list to > avoid redundant reads and writes to the disk. > > Instead of inserting the `exec' commands between the other commands and > re-parsing the buffer at the end, they are appended to the buffer once, > and a new list of items is created. Items from the old list are copied > across and new `exec' items are appended when necessary. This > eliminates the need to reparse the buffer, but this also means we have > to use todo_list_write_to_disk() to write the file. > > todo_list_add_exec_commands() and sequencer_add_exec_commands() are > modified to take a string list instead of a string -- one item for each > command. This makes it easier to insert a new command to the todo list > for each command to execute. > > sequencer_add_exec_commands() still reads the todo list from the disk, > as it is needed by rebase -p. > > complete_action() still uses sequencer_add_exec_commands() for now. > This will be changed in a future commit. > > Signed-off-by: Alban Gruin <alban.gruin@gmail.com> > --- > builtin/rebase--interactive.c | 15 +++-- > sequencer.c | 110 +++++++++++++++++++++------------- > sequencer.h | 5 +- > 3 files changed, 82 insertions(+), 48 deletions(-) > > diff --git a/builtin/rebase--interactive.c b/builtin/rebase--interactive.c > index df19ccaeb9..53056ee713 100644 > --- a/builtin/rebase--interactive.c > +++ b/builtin/rebase--interactive.c > @@ -65,7 +65,7 @@ static int do_interactive_rebase(struct replay_opts *opts, unsigned flags, > const char *onto, const char *onto_name, > const char *squash_onto, const char *head_name, > const char *restrict_revision, char *raw_strategies, > - const char *cmd, unsigned autosquash) > + struct string_list *commands, unsigned autosquash) > { > int ret; > const char *head_hash = NULL; > @@ -116,7 +116,7 @@ static int do_interactive_rebase(struct replay_opts *opts, unsigned flags, > discard_cache(); > ret = complete_action(the_repository, opts, flags, > shortrevisions, onto_name, onto, > - head_hash, cmd, autosquash); > + head_hash, commands, autosquash); > } > > free(revisions); > @@ -139,6 +139,7 @@ int cmd_rebase__interactive(int argc, const char **argv, const char *prefix) > const char *onto = NULL, *onto_name = NULL, *restrict_revision = NULL, > *squash_onto = NULL, *upstream = NULL, *head_name = NULL, > *switch_to = NULL, *cmd = NULL; > + struct string_list commands = STRING_LIST_INIT_DUP; > char *raw_strategies = NULL; > enum { > NONE = 0, CONTINUE, SKIP, EDIT_TODO, SHOW_CURRENT_PATCH, > @@ -221,6 +222,12 @@ int cmd_rebase__interactive(int argc, const char **argv, const char *prefix) > warning(_("--[no-]rebase-cousins has no effect without " > "--rebase-merges")); > > + if (cmd && *cmd) { > + string_list_split(&commands, cmd, '\n', -1); This whole splitting and later skipping 'exec ' is a bit of a shame - it would be much nicer if we could just have one exec command per -x option but I think that is outside the scope of this series (If I have time I'd like to look at calling do_interactive_rebase() directly from builtin/rebase.c without forking rebase--interactive). > + if (strlen(commands.items[commands.nr - 1].string) == 0) I'd be tempted just to test the string using !* rather than calling strlen. Also is there ever a case where the last string isn't empty? > + --commands.nr; > + } > + > switch (command) { > case NONE: > if (!onto && !upstream) > @@ -228,7 +235,7 @@ int cmd_rebase__interactive(int argc, const char **argv, const char *prefix) > > ret = do_interactive_rebase(&opts, flags, switch_to, upstream, onto, > onto_name, squash_onto, head_name, restrict_revision, > - raw_strategies, cmd, autosquash); > + raw_strategies, &commands, autosquash); > break; > case SKIP: { > struct string_list merge_rr = STRING_LIST_INIT_DUP; > @@ -262,7 +269,7 @@ int cmd_rebase__interactive(int argc, const char **argv, const char *prefix) > ret = rearrange_squash(the_repository); > break; > case ADD_EXEC: > - ret = sequencer_add_exec_commands(the_repository, cmd); > + ret = sequencer_add_exec_commands(the_repository, &commands); > break; > default: > BUG("invalid command '%d'", command); > diff --git a/sequencer.c b/sequencer.c > index 266f80d704..3a90b419d7 100644 > --- a/sequencer.c > +++ b/sequencer.c > @@ -4446,25 +4446,27 @@ int sequencer_make_script(struct repository *r, FILE *out, > return 0; > } > > -/* > - * Add commands after pick and (series of) squash/fixup commands > - * in the todo list. > - */ > -int sequencer_add_exec_commands(struct repository *r, > - const char *commands) > +static void todo_list_add_exec_commands(struct todo_list *todo_list, > + struct string_list *commands) > { > - const char *todo_file = rebase_path_todo(); > - struct todo_list todo_list = TODO_LIST_INIT; > - struct strbuf *buf = &todo_list.buf; > - size_t offset = 0, commands_len = strlen(commands); > - int i, insert; > + struct strbuf *buf = &todo_list->buf; > + size_t base_offset = buf->len; > + int i, insert, nr = 0, alloc = 0; > + struct todo_item *items = NULL, *base_items = NULL; > > - if (strbuf_read_file(&todo_list.buf, todo_file, 0) < 0) > - return error(_("could not read '%s'."), todo_file); > + base_items = xcalloc(commands->nr, sizeof(struct todo_item)); > + for (i = 0; i < commands->nr; ++i) { > + size_t command_len = strlen(commands->items[i].string); > > - if (todo_list_parse_insn_buffer(r, todo_list.buf.buf, &todo_list)) { > - todo_list_release(&todo_list); > - return error(_("unusable todo list: '%s'"), todo_file); > + strbuf_addstr(buf, commands->items[i].string); > + strbuf_addch(buf, '\n'); > + > + base_items[i].command = TODO_EXEC; > + base_items[i].offset_in_buf = base_offset; > + base_items[i].arg_offset = base_offset + strlen("exec "); > + base_items[i].arg_len = command_len - strlen("exec "); > + > + base_offset += command_len + 1; > } > > /* > @@ -4473,38 +4475,62 @@ int sequencer_add_exec_commands(struct repository *r, > * those chains if there are any. > */ > insert = -1; > - for (i = 0; i < todo_list.nr; i++) { > - enum todo_command command = todo_list.items[i].command; > - > - if (insert >= 0) { > - /* skip fixup/squash chains */ > - if (command == TODO_COMMENT) > - continue; > - else if (is_fixup(command)) { > - insert = i + 1; > - continue; > - } > - strbuf_insert(buf, > - todo_list.items[insert].offset_in_buf + > - offset, commands, commands_len); In a todo list that looks like pick abc message #pick cde empty commit This inserts the exec command for the first pick above the commented out pick. I think your translation puts it below the commented out pick as it ignores the value of insert. I think it's probably easiest to add an INSERT_ARRAY macro to insert it in the right place. An alternative might be to track the last insert position and only copy commands across when there is another exec to insert but that might get complicated in cases such as pick abc message #squash cde squash! message //empty commit for rewording fixup 123 fixup! message #pick 456 empty commit Best Wishes Phillip > - offset += commands_len; > + for (i = 0; i < todo_list->nr; i++) { > + enum todo_command command = todo_list->items[i].command; > + if (insert >= 0 && command != TODO_COMMENT && !is_fixup(command)) { > + ALLOC_GROW(items, nr + commands->nr, alloc); > + COPY_ARRAY(items + nr, base_items, commands->nr); > + nr += commands->nr; > insert = -1; > } > > - if (command == TODO_PICK || command == TODO_MERGE) > + ALLOC_GROW(items, nr + 1, alloc); > + items[nr++] = todo_list->items[i]; > + > + if (command == TODO_PICK || command == TODO_MERGE || is_fixup(command)) > insert = i + 1; > } > > /* insert or append final <commands> */ > - if (insert >= 0 && insert < todo_list.nr) > - strbuf_insert(buf, todo_list.items[insert].offset_in_buf + > - offset, commands, commands_len); > - else if (insert >= 0 || !offset) > - strbuf_add(buf, commands, commands_len); > + if (insert >= 0 || nr == todo_list->nr) { > + ALLOC_GROW(items, nr + commands->nr, alloc); > + COPY_ARRAY(items + nr, base_items, commands->nr); > + nr += commands->nr; > + } > + > + free(base_items); > + FREE_AND_NULL(todo_list->items); > + todo_list->items = items; > + todo_list->nr = nr; > + todo_list->alloc = alloc;> +} > > - i = write_message(buf->buf, buf->len, todo_file, 0); > +/* > + * Add commands after pick and (series of) squash/fixup commands > + * in the todo list. > + */ > +int sequencer_add_exec_commands(struct repository *r, > + struct string_list *commands) > +{ > + const char *todo_file = rebase_path_todo(); > + struct todo_list todo_list = TODO_LIST_INIT; > + int res; > + > + if (strbuf_read_file(&todo_list.buf, todo_file, 0) < 0) > + return error_errno(_("could not read '%s'."), todo_file); > + > + if (todo_list_parse_insn_buffer(r, todo_list.buf.buf, &todo_list)) { > + todo_list_release(&todo_list); > + return error(_("unusable todo list: '%s'"), todo_file); > + } > + > + todo_list_add_exec_commands(&todo_list, commands); > + res = todo_list_write_to_file(r, &todo_list, todo_file, NULL, NULL, -1, 0); > todo_list_release(&todo_list); > - return i; > + > + if (res) > + return error_errno(_("could not write '%s'."), todo_file); > + return 0; > } > > static void todo_list_to_strbuf(struct repository *r, struct todo_list *todo_list, > @@ -4735,7 +4761,7 @@ static int skip_unnecessary_picks(struct repository *r, struct object_id *output > > int complete_action(struct repository *r, struct replay_opts *opts, unsigned flags, > const char *shortrevisions, const char *onto_name, > - const char *onto, const char *orig_head, const char *cmd, > + const char *onto, const char *orig_head, struct string_list *commands, > unsigned autosquash) > { > const char *shortonto, *todo_file = rebase_path_todo(); > @@ -4754,8 +4780,8 @@ int complete_action(struct repository *r, struct replay_opts *opts, unsigned fla > if (autosquash && rearrange_squash(r)) > return -1; > > - if (cmd && *cmd) > - sequencer_add_exec_commands(r, cmd); > + if (commands->nr) > + sequencer_add_exec_commands(r, commands); > > if (strbuf_read_file(buf, todo_file, 0) < 0) > return error_errno(_("could not read '%s'."), todo_file); > diff --git a/sequencer.h b/sequencer.h > index 1de97f188d..e79f03e213 100644 > --- a/sequencer.h > +++ b/sequencer.h > @@ -146,12 +146,13 @@ int sequencer_make_script(struct repository *r, FILE *out, int argc, > const char **argv, > unsigned flags); > > -int sequencer_add_exec_commands(struct repository *r, const char *command); > +int sequencer_add_exec_commands(struct repository *r, > + struct string_list *commands); > int transform_todo_file(struct repository *r, unsigned flags); > int check_todo_list_from_file(struct repository *r); > int complete_action(struct repository *r, struct replay_opts *opts, unsigned flags, > const char *shortrevisions, const char *onto_name, > - const char *onto, const char *orig_head, const char *cmd, > + const char *onto, const char *orig_head, struct string_list *commands, > unsigned autosquash); > int rearrange_squash(struct repository *r); > >
Hi Phillip, Le 31/01/2019 à 15:30, Phillip Wood a écrit : > Hi Alban > > On 29/01/2019 15:01, Alban Gruin wrote: >> This refactors sequencer_add_exec_commands() to work on a todo_list to >> avoid redundant reads and writes to the disk. >> >> Instead of inserting the `exec' commands between the other commands and >> re-parsing the buffer at the end, they are appended to the buffer once, >> and a new list of items is created. Items from the old list are copied >> across and new `exec' items are appended when necessary. This >> eliminates the need to reparse the buffer, but this also means we have >> to use todo_list_write_to_disk() to write the file. >> >> todo_list_add_exec_commands() and sequencer_add_exec_commands() are >> modified to take a string list instead of a string -- one item for each >> command. This makes it easier to insert a new command to the todo list >> for each command to execute. >> >> sequencer_add_exec_commands() still reads the todo list from the disk, >> as it is needed by rebase -p. >> >> complete_action() still uses sequencer_add_exec_commands() for now. >> This will be changed in a future commit. >> >> Signed-off-by: Alban Gruin <alban.gruin@gmail.com> >> --- >> builtin/rebase--interactive.c | 15 +++-- >> sequencer.c | 110 +++++++++++++++++++++------------- >> sequencer.h | 5 +- >> 3 files changed, 82 insertions(+), 48 deletions(-) >> >> diff --git a/builtin/rebase--interactive.c >> b/builtin/rebase--interactive.c >> index df19ccaeb9..53056ee713 100644 >> --- a/builtin/rebase--interactive.c >> +++ b/builtin/rebase--interactive.c >> @@ -65,7 +65,7 @@ static int do_interactive_rebase(struct replay_opts >> *opts, unsigned flags, >> const char *onto, const char *onto_name, >> const char *squash_onto, const char *head_name, >> const char *restrict_revision, char *raw_strategies, >> - const char *cmd, unsigned autosquash) >> + struct string_list *commands, unsigned autosquash) >> { >> int ret; >> const char *head_hash = NULL; >> @@ -116,7 +116,7 @@ static int do_interactive_rebase(struct >> replay_opts *opts, unsigned flags, >> discard_cache(); >> ret = complete_action(the_repository, opts, flags, >> shortrevisions, onto_name, onto, >> - head_hash, cmd, autosquash); >> + head_hash, commands, autosquash); >> } >> >> free(revisions); >> @@ -139,6 +139,7 @@ int cmd_rebase__interactive(int argc, const char >> **argv, const char *prefix) >> const char *onto = NULL, *onto_name = NULL, *restrict_revision = >> NULL, >> *squash_onto = NULL, *upstream = NULL, *head_name = NULL, >> *switch_to = NULL, *cmd = NULL; >> + struct string_list commands = STRING_LIST_INIT_DUP; >> char *raw_strategies = NULL; >> enum { >> NONE = 0, CONTINUE, SKIP, EDIT_TODO, SHOW_CURRENT_PATCH, >> @@ -221,6 +222,12 @@ int cmd_rebase__interactive(int argc, const char >> **argv, const char *prefix) >> warning(_("--[no-]rebase-cousins has no effect without " >> "--rebase-merges")); >> >> + if (cmd && *cmd) { >> + string_list_split(&commands, cmd, '\n', -1); > > This whole splitting and later skipping 'exec ' is a bit of a shame - it > would be much nicer if we could just have one exec command per -x option > but I think that is outside the scope of this series (If I have time I'd > like to look at calling do_interactive_rebase() directly from > builtin/rebase.c without forking rebase--interactive). > Yes, I completely agree with you. I thought to do this in preparation to drop rebase -r. >> + if (strlen(commands.items[commands.nr - 1].string) == 0) > > I'd be tempted just to test the string using !* rather than calling > strlen. > Right. I’m still not used to this pattern. > Also is there ever a case where the last string isn't empty? I don’t think so. When rebase.c prepares the arguments for rebase--interactive, it always add a newline at the end[1]. Do you want me to drop this check? >> + --commands.nr; >> + } >> + >> switch (command) { >> case NONE: >> if (!onto && !upstream) >> @@ -228,7 +235,7 @@ int cmd_rebase__interactive(int argc, const char >> **argv, const char *prefix) >> >> ret = do_interactive_rebase(&opts, flags, switch_to, >> upstream, onto, >> onto_name, squash_onto, head_name, >> restrict_revision, >> - raw_strategies, cmd, autosquash); >> + raw_strategies, &commands, autosquash); >> break; >> case SKIP: { >> struct string_list merge_rr = STRING_LIST_INIT_DUP; >> @@ -262,7 +269,7 @@ int cmd_rebase__interactive(int argc, const char >> **argv, const char *prefix) >> ret = rearrange_squash(the_repository); >> break; >> case ADD_EXEC: >> - ret = sequencer_add_exec_commands(the_repository, cmd); >> + ret = sequencer_add_exec_commands(the_repository, &commands); >> break; >> default: >> BUG("invalid command '%d'", command); >> diff --git a/sequencer.c b/sequencer.c >> index 266f80d704..3a90b419d7 100644 >> --- a/sequencer.c >> +++ b/sequencer.c >> @@ -4446,25 +4446,27 @@ int sequencer_make_script(struct repository >> *r, FILE *out, >> return 0; >> } >> >> -/* >> - * Add commands after pick and (series of) squash/fixup commands >> - * in the todo list. >> - */ >> -int sequencer_add_exec_commands(struct repository *r, >> - const char *commands) >> +static void todo_list_add_exec_commands(struct todo_list *todo_list, >> + struct string_list *commands) >> { >> - const char *todo_file = rebase_path_todo(); >> - struct todo_list todo_list = TODO_LIST_INIT; >> - struct strbuf *buf = &todo_list.buf; >> - size_t offset = 0, commands_len = strlen(commands); >> - int i, insert; >> + struct strbuf *buf = &todo_list->buf; >> + size_t base_offset = buf->len; >> + int i, insert, nr = 0, alloc = 0; >> + struct todo_item *items = NULL, *base_items = NULL; >> >> - if (strbuf_read_file(&todo_list.buf, todo_file, 0) < 0) >> - return error(_("could not read '%s'."), todo_file); >> + base_items = xcalloc(commands->nr, sizeof(struct todo_item)); >> + for (i = 0; i < commands->nr; ++i) { >> + size_t command_len = strlen(commands->items[i].string); >> >> - if (todo_list_parse_insn_buffer(r, todo_list.buf.buf, &todo_list)) { >> - todo_list_release(&todo_list); >> - return error(_("unusable todo list: '%s'"), todo_file); >> + strbuf_addstr(buf, commands->items[i].string); >> + strbuf_addch(buf, '\n'); >> + >> + base_items[i].command = TODO_EXEC; >> + base_items[i].offset_in_buf = base_offset; >> + base_items[i].arg_offset = base_offset + strlen("exec "); >> + base_items[i].arg_len = command_len - strlen("exec "); >> + >> + base_offset += command_len + 1; >> } >> >> /* >> @@ -4473,38 +4475,62 @@ int sequencer_add_exec_commands(struct >> repository *r, >> * those chains if there are any. >> */ >> insert = -1; >> - for (i = 0; i < todo_list.nr; i++) { >> - enum todo_command command = todo_list.items[i].command; >> - >> - if (insert >= 0) { >> - /* skip fixup/squash chains */ >> - if (command == TODO_COMMENT) >> - continue; >> - else if (is_fixup(command)) { >> - insert = i + 1; >> - continue; >> - } >> - strbuf_insert(buf, >> - todo_list.items[insert].offset_in_buf + >> - offset, commands, commands_len); > > In a todo list that looks like > pick abc message > #pick cde empty commit > This inserts the exec command for the first pick above the commented out > pick. I think your translation puts it below the commented out pick as > it ignores the value of insert. I think it's probably easiest to add an > INSERT_ARRAY macro to insert it in the right place. An alternative might > be to track the last insert position and only copy commands across when > there is another exec to insert but that might get complicated in cases > such as > > pick abc message > #squash cde squash! message //empty commit for rewording > fixup 123 fixup! message > #pick 456 empty commit > I could do this with MOVE_ARRAY(), no? > Best Wishes > > Phillip > [1] https://github.com/git/git/blob/master/builtin/rebase.c#L1182-L1191 Cheers, Alban
Hi Alban On 31/01/2019 20:37, Alban Gruin wrote: > Hi Phillip, > > Le 31/01/2019 à 15:30, Phillip Wood a écrit : >> Hi Alban >> >> On 29/01/2019 15:01, Alban Gruin wrote: >>> This refactors sequencer_add_exec_commands() to work on a todo_list to >>> avoid redundant reads and writes to the disk. >>> >>> Instead of inserting the `exec' commands between the other commands and >>> re-parsing the buffer at the end, they are appended to the buffer once, >>> and a new list of items is created. Items from the old list are copied >>> across and new `exec' items are appended when necessary. This >>> eliminates the need to reparse the buffer, but this also means we have >>> to use todo_list_write_to_disk() to write the file. >>> >>> todo_list_add_exec_commands() and sequencer_add_exec_commands() are >>> modified to take a string list instead of a string -- one item for each >>> command. This makes it easier to insert a new command to the todo list >>> for each command to execute. >>> >>> sequencer_add_exec_commands() still reads the todo list from the disk, >>> as it is needed by rebase -p. >>> >>> complete_action() still uses sequencer_add_exec_commands() for now. >>> This will be changed in a future commit. >>> >>> Signed-off-by: Alban Gruin <alban.gruin@gmail.com> >>> --- >>> builtin/rebase--interactive.c | 15 +++-- >>> sequencer.c | 110 +++++++++++++++++++++------------- >>> sequencer.h | 5 +- >>> 3 files changed, 82 insertions(+), 48 deletions(-) >>> >>> diff --git a/builtin/rebase--interactive.c >>> b/builtin/rebase--interactive.c >>> index df19ccaeb9..53056ee713 100644 >>> --- a/builtin/rebase--interactive.c >>> +++ b/builtin/rebase--interactive.c >>> @@ -65,7 +65,7 @@ static int do_interactive_rebase(struct replay_opts >>> *opts, unsigned flags, >>> const char *onto, const char *onto_name, >>> const char *squash_onto, const char *head_name, >>> const char *restrict_revision, char *raw_strategies, >>> - const char *cmd, unsigned autosquash) >>> + struct string_list *commands, unsigned autosquash) >>> { >>> int ret; >>> const char *head_hash = NULL; >>> @@ -116,7 +116,7 @@ static int do_interactive_rebase(struct >>> replay_opts *opts, unsigned flags, >>> discard_cache(); >>> ret = complete_action(the_repository, opts, flags, >>> shortrevisions, onto_name, onto, >>> - head_hash, cmd, autosquash); >>> + head_hash, commands, autosquash); >>> } >>> >>> free(revisions); >>> @@ -139,6 +139,7 @@ int cmd_rebase__interactive(int argc, const char >>> **argv, const char *prefix) >>> const char *onto = NULL, *onto_name = NULL, *restrict_revision = >>> NULL, >>> *squash_onto = NULL, *upstream = NULL, *head_name = NULL, >>> *switch_to = NULL, *cmd = NULL; >>> + struct string_list commands = STRING_LIST_INIT_DUP; >>> char *raw_strategies = NULL; >>> enum { >>> NONE = 0, CONTINUE, SKIP, EDIT_TODO, SHOW_CURRENT_PATCH, >>> @@ -221,6 +222,12 @@ int cmd_rebase__interactive(int argc, const char >>> **argv, const char *prefix) >>> warning(_("--[no-]rebase-cousins has no effect without " >>> "--rebase-merges")); >>> >>> + if (cmd && *cmd) { >>> + string_list_split(&commands, cmd, '\n', -1); >> >> This whole splitting and later skipping 'exec ' is a bit of a shame - it >> would be much nicer if we could just have one exec command per -x option >> but I think that is outside the scope of this series (If I have time I'd >> like to look at calling do_interactive_rebase() directly from >> builtin/rebase.c without forking rebase--interactive). >> > > Yes, I completely agree with you. I thought to do this in preparation > to drop rebase -r. > >>> + if (strlen(commands.items[commands.nr - 1].string) == 0) >> >> I'd be tempted just to test the string using !* rather than calling >> strlen. >> > > Right. I’m still not used to this pattern. > >> Also is there ever a case where the last string isn't empty? > > I don’t think so. When rebase.c prepares the arguments for > rebase--interactive, it always add a newline at the end[1]. Do you want > me to drop this check? I think that would be clearer > > >>> + --commands.nr; >>> + } >>> + >>> switch (command) { >>> case NONE: >>> if (!onto && !upstream) >>> @@ -228,7 +235,7 @@ int cmd_rebase__interactive(int argc, const char >>> **argv, const char *prefix) >>> >>> ret = do_interactive_rebase(&opts, flags, switch_to, >>> upstream, onto, >>> onto_name, squash_onto, head_name, >>> restrict_revision, >>> - raw_strategies, cmd, autosquash); >>> + raw_strategies, &commands, autosquash); >>> break; >>> case SKIP: { >>> struct string_list merge_rr = STRING_LIST_INIT_DUP; >>> @@ -262,7 +269,7 @@ int cmd_rebase__interactive(int argc, const char >>> **argv, const char *prefix) >>> ret = rearrange_squash(the_repository); >>> break; >>> case ADD_EXEC: >>> - ret = sequencer_add_exec_commands(the_repository, cmd); >>> + ret = sequencer_add_exec_commands(the_repository, &commands); >>> break; >>> default: >>> BUG("invalid command '%d'", command); >>> diff --git a/sequencer.c b/sequencer.c >>> index 266f80d704..3a90b419d7 100644 >>> --- a/sequencer.c >>> +++ b/sequencer.c >>> @@ -4446,25 +4446,27 @@ int sequencer_make_script(struct repository >>> *r, FILE *out, >>> return 0; >>> } >>> >>> -/* >>> - * Add commands after pick and (series of) squash/fixup commands >>> - * in the todo list. >>> - */ >>> -int sequencer_add_exec_commands(struct repository *r, >>> - const char *commands) >>> +static void todo_list_add_exec_commands(struct todo_list *todo_list, >>> + struct string_list *commands) >>> { >>> - const char *todo_file = rebase_path_todo(); >>> - struct todo_list todo_list = TODO_LIST_INIT; >>> - struct strbuf *buf = &todo_list.buf; >>> - size_t offset = 0, commands_len = strlen(commands); >>> - int i, insert; >>> + struct strbuf *buf = &todo_list->buf; >>> + size_t base_offset = buf->len; >>> + int i, insert, nr = 0, alloc = 0; >>> + struct todo_item *items = NULL, *base_items = NULL; >>> >>> - if (strbuf_read_file(&todo_list.buf, todo_file, 0) < 0) >>> - return error(_("could not read '%s'."), todo_file); >>> + base_items = xcalloc(commands->nr, sizeof(struct todo_item)); >>> + for (i = 0; i < commands->nr; ++i) { >>> + size_t command_len = strlen(commands->items[i].string); >>> >>> - if (todo_list_parse_insn_buffer(r, todo_list.buf.buf, &todo_list)) { >>> - todo_list_release(&todo_list); >>> - return error(_("unusable todo list: '%s'"), todo_file); >>> + strbuf_addstr(buf, commands->items[i].string); >>> + strbuf_addch(buf, '\n'); >>> + >>> + base_items[i].command = TODO_EXEC; >>> + base_items[i].offset_in_buf = base_offset; >>> + base_items[i].arg_offset = base_offset + strlen("exec "); >>> + base_items[i].arg_len = command_len - strlen("exec "); >>> + >>> + base_offset += command_len + 1; >>> } >>> >>> /* >>> @@ -4473,38 +4475,62 @@ int sequencer_add_exec_commands(struct >>> repository *r, >>> * those chains if there are any. >>> */ >>> insert = -1; >>> - for (i = 0; i < todo_list.nr; i++) { >>> - enum todo_command command = todo_list.items[i].command; >>> - >>> - if (insert >= 0) { >>> - /* skip fixup/squash chains */ >>> - if (command == TODO_COMMENT) >>> - continue; >>> - else if (is_fixup(command)) { >>> - insert = i + 1; >>> - continue; >>> - } >>> - strbuf_insert(buf, >>> - todo_list.items[insert].offset_in_buf + >>> - offset, commands, commands_len); >> >> In a todo list that looks like >> pick abc message >> #pick cde empty commit >> This inserts the exec command for the first pick above the commented out >> pick. I think your translation puts it below the commented out pick as >> it ignores the value of insert. I think it's probably easiest to add an >> INSERT_ARRAY macro to insert it in the right place. An alternative might >> be to track the last insert position and only copy commands across when >> there is another exec to insert but that might get complicated in cases >> such as >> >> pick abc message >> #squash cde squash! message //empty commit for rewording >> fixup 123 fixup! message >> #pick 456 empty commit >> > > I could do this with MOVE_ARRAY(), no? Yes, if you extend the array first then you could use MOVE_ARRAY() and COPY_ARRAY() to move the comment down and then insert the exec commands so maybe we don't need a new macro after all. I've looked through most of the rest of this series (I think I've got three patches left to check) and they all look fine, I'll try and look at the rest tomorrow, but if not I'll get round to it next week. Best Wishes Phillip >> Best Wishes >> >> Phillip >> > > [1] https://github.com/git/git/blob/master/builtin/rebase.c#L1182-L1191 > > Cheers, > Alban > >
On 31/01/2019 14:30, Phillip Wood wrote: > Hi Alban > > On 29/01/2019 15:01, Alban Gruin wrote: >> This refactors sequencer_add_exec_commands() to work on a todo_list to >> avoid redundant reads and writes to the disk. >> >> Instead of inserting the `exec' commands between the other commands and >> re-parsing the buffer at the end, they are appended to the buffer once, >> and a new list of items is created. Items from the old list are copied >> across and new `exec' items are appended when necessary. This >> eliminates the need to reparse the buffer, but this also means we have >> to use todo_list_write_to_disk() to write the file. >> >> todo_list_add_exec_commands() and sequencer_add_exec_commands() are >> modified to take a string list instead of a string -- one item for each >> command. This makes it easier to insert a new command to the todo list >> for each command to execute. >> >> sequencer_add_exec_commands() still reads the todo list from the disk, >> as it is needed by rebase -p. >> >> complete_action() still uses sequencer_add_exec_commands() for now. >> This will be changed in a future commit. >> >> Signed-off-by: Alban Gruin <alban.gruin@gmail.com> >> --- >> builtin/rebase--interactive.c | 15 +++-- >> sequencer.c | 110 +++++++++++++++++++++------------- >> sequencer.h | 5 +- >> 3 files changed, 82 insertions(+), 48 deletions(-) >> >> diff --git a/builtin/rebase--interactive.c >> b/builtin/rebase--interactive.c >> index df19ccaeb9..53056ee713 100644 >> --- a/builtin/rebase--interactive.c >> +++ b/builtin/rebase--interactive.c >> @@ -65,7 +65,7 @@ static int do_interactive_rebase(struct replay_opts >> *opts, unsigned flags, >> const char *onto, const char *onto_name, >> const char *squash_onto, const char *head_name, >> const char *restrict_revision, char *raw_strategies, >> - const char *cmd, unsigned autosquash) >> + struct string_list *commands, unsigned autosquash) >> { >> int ret; >> const char *head_hash = NULL; >> @@ -116,7 +116,7 @@ static int do_interactive_rebase(struct >> replay_opts *opts, unsigned flags, >> discard_cache(); >> ret = complete_action(the_repository, opts, flags, >> shortrevisions, onto_name, onto, >> - head_hash, cmd, autosquash); >> + head_hash, commands, autosquash); >> } >> >> free(revisions); >> @@ -139,6 +139,7 @@ int cmd_rebase__interactive(int argc, const char >> **argv, const char *prefix) >> const char *onto = NULL, *onto_name = NULL, *restrict_revision = >> NULL, >> *squash_onto = NULL, *upstream = NULL, *head_name = NULL, >> *switch_to = NULL, *cmd = NULL; >> + struct string_list commands = STRING_LIST_INIT_DUP; >> char *raw_strategies = NULL; >> enum { >> NONE = 0, CONTINUE, SKIP, EDIT_TODO, SHOW_CURRENT_PATCH, >> @@ -221,6 +222,12 @@ int cmd_rebase__interactive(int argc, const char >> **argv, const char *prefix) >> warning(_("--[no-]rebase-cousins has no effect without " >> "--rebase-merges")); >> >> + if (cmd && *cmd) { >> + string_list_split(&commands, cmd, '\n', -1); > > This whole splitting and later skipping 'exec ' is a bit of a shame - it > would be much nicer if we could just have one exec command per -x option > but I think that is outside the scope of this series (If I have time I'd > like to look at calling do_interactive_rebase() directly from > builtin/rebase.c without forking rebase--interactive). > >> + if (strlen(commands.items[commands.nr - 1].string) == 0) > > I'd be tempted just to test the string using !* rather than calling > strlen. Also is there ever a case where the last string isn't empty? > >> + --commands.nr; >> + } >> + >> switch (command) { >> case NONE: >> if (!onto && !upstream) >> @@ -228,7 +235,7 @@ int cmd_rebase__interactive(int argc, const char >> **argv, const char *prefix) >> >> ret = do_interactive_rebase(&opts, flags, switch_to, >> upstream, onto, >> onto_name, squash_onto, head_name, >> restrict_revision, >> - raw_strategies, cmd, autosquash); >> + raw_strategies, &commands, autosquash); >> break; >> case SKIP: { >> struct string_list merge_rr = STRING_LIST_INIT_DUP; >> @@ -262,7 +269,7 @@ int cmd_rebase__interactive(int argc, const char >> **argv, const char *prefix) >> ret = rearrange_squash(the_repository); >> break; >> case ADD_EXEC: >> - ret = sequencer_add_exec_commands(the_repository, cmd); >> + ret = sequencer_add_exec_commands(the_repository, &commands); >> break; >> default: >> BUG("invalid command '%d'", command); >> diff --git a/sequencer.c b/sequencer.c >> index 266f80d704..3a90b419d7 100644 >> --- a/sequencer.c >> +++ b/sequencer.c >> @@ -4446,25 +4446,27 @@ int sequencer_make_script(struct repository >> *r, FILE *out, >> return 0; >> } >> >> -/* >> - * Add commands after pick and (series of) squash/fixup commands >> - * in the todo list. >> - */ >> -int sequencer_add_exec_commands(struct repository *r, >> - const char *commands) >> +static void todo_list_add_exec_commands(struct todo_list *todo_list, >> + struct string_list *commands) >> { >> - const char *todo_file = rebase_path_todo(); >> - struct todo_list todo_list = TODO_LIST_INIT; >> - struct strbuf *buf = &todo_list.buf; >> - size_t offset = 0, commands_len = strlen(commands); >> - int i, insert; >> + struct strbuf *buf = &todo_list->buf; >> + size_t base_offset = buf->len; >> + int i, insert, nr = 0, alloc = 0; >> + struct todo_item *items = NULL, *base_items = NULL; >> >> - if (strbuf_read_file(&todo_list.buf, todo_file, 0) < 0) >> - return error(_("could not read '%s'."), todo_file); >> + base_items = xcalloc(commands->nr, sizeof(struct todo_item)); >> + for (i = 0; i < commands->nr; ++i) { >> + size_t command_len = strlen(commands->items[i].string); >> >> - if (todo_list_parse_insn_buffer(r, todo_list.buf.buf, &todo_list)) { >> - todo_list_release(&todo_list); >> - return error(_("unusable todo list: '%s'"), todo_file); >> + strbuf_addstr(buf, commands->items[i].string); >> + strbuf_addch(buf, '\n'); >> + >> + base_items[i].command = TODO_EXEC; >> + base_items[i].offset_in_buf = base_offset; >> + base_items[i].arg_offset = base_offset + strlen("exec "); >> + base_items[i].arg_len = command_len - strlen("exec "); >> + >> + base_offset += command_len + 1; >> } >> >> /* >> @@ -4473,38 +4475,62 @@ int sequencer_add_exec_commands(struct >> repository *r, >> * those chains if there are any. >> */ >> insert = -1; >> - for (i = 0; i < todo_list.nr; i++) { >> - enum todo_command command = todo_list.items[i].command; >> - >> - if (insert >= 0) { >> - /* skip fixup/squash chains */ >> - if (command == TODO_COMMENT) >> - continue; >> - else if (is_fixup(command)) { >> - insert = i + 1; >> - continue; >> - } >> - strbuf_insert(buf, >> - todo_list.items[insert].offset_in_buf + >> - offset, commands, commands_len); > > In a todo list that looks like > pick abc message > #pick cde empty commit > This inserts the exec command for the first pick above the commented out > pick. I think your translation puts it below the commented out pick as > it ignores the value of insert. I think it's probably easiest to add an > INSERT_ARRAY macro to insert it in the right place. An alternative might > be to track the last insert position and only copy commands across when > there is another exec to insert but that might get complicated in cases > such as > > pick abc message > #squash cde squash! message //empty commit for rewording > fixup 123 fixup! message > #pick 456 empty commit Thinking about this, I'm not sure it can happen as the empty squash commit will be commented out before rearrange_squash() is called so I think it would actually look like pick abc message fixup 123 fixup! message #pick cde squash! message #pick 456 empty commit So I wonder if we can get away with treating insert as a flag and inserting exec commands when insert && command == TODO_COMMENT We could probably do with some tests for this to be sure. Best Wishes Phillip > > Best Wishes > > Phillip > >> - offset += commands_len; >> + for (i = 0; i < todo_list->nr; i++) { >> + enum todo_command command = todo_list->items[i].command; >> + if (insert >= 0 && command != TODO_COMMENT && >> !is_fixup(command)) { >> + ALLOC_GROW(items, nr + commands->nr, alloc); >> + COPY_ARRAY(items + nr, base_items, commands->nr); >> + nr += commands->nr; >> insert = -1; >> } >> >> - if (command == TODO_PICK || command == TODO_MERGE) >> + ALLOC_GROW(items, nr + 1, alloc); >> + items[nr++] = todo_list->items[i]; >> + >> + if (command == TODO_PICK || command == TODO_MERGE || >> is_fixup(command)) >> insert = i + 1; >> } >> >> /* insert or append final <commands> */ >> - if (insert >= 0 && insert < todo_list.nr) >> - strbuf_insert(buf, todo_list.items[insert].offset_in_buf + >> - offset, commands, commands_len); >> - else if (insert >= 0 || !offset) >> - strbuf_add(buf, commands, commands_len); >> + if (insert >= 0 || nr == todo_list->nr) { >> + ALLOC_GROW(items, nr + commands->nr, alloc); >> + COPY_ARRAY(items + nr, base_items, commands->nr); >> + nr += commands->nr; >> + } >> + >> + free(base_items); >> + FREE_AND_NULL(todo_list->items); >> + todo_list->items = items; >> + todo_list->nr = nr; >> + todo_list->alloc = alloc;> +} >> >> - i = write_message(buf->buf, buf->len, todo_file, 0); >> +/* >> + * Add commands after pick and (series of) squash/fixup commands >> + * in the todo list. >> + */ >> +int sequencer_add_exec_commands(struct repository *r, >> + struct string_list *commands) >> +{ >> + const char *todo_file = rebase_path_todo(); >> + struct todo_list todo_list = TODO_LIST_INIT; >> + int res; >> + >> + if (strbuf_read_file(&todo_list.buf, todo_file, 0) < 0) >> + return error_errno(_("could not read '%s'."), todo_file); >> + >> + if (todo_list_parse_insn_buffer(r, todo_list.buf.buf, &todo_list)) { >> + todo_list_release(&todo_list); >> + return error(_("unusable todo list: '%s'"), todo_file); >> + } >> + >> + todo_list_add_exec_commands(&todo_list, commands); >> + res = todo_list_write_to_file(r, &todo_list, todo_file, NULL, >> NULL, -1, 0); >> todo_list_release(&todo_list); >> - return i; >> + >> + if (res) >> + return error_errno(_("could not write '%s'."), todo_file); >> + return 0; >> } >> >> static void todo_list_to_strbuf(struct repository *r, struct >> todo_list *todo_list, >> @@ -4735,7 +4761,7 @@ static int skip_unnecessary_picks(struct >> repository *r, struct object_id *output >> >> int complete_action(struct repository *r, struct replay_opts *opts, >> unsigned flags, >> const char *shortrevisions, const char *onto_name, >> - const char *onto, const char *orig_head, const char *cmd, >> + const char *onto, const char *orig_head, struct >> string_list *commands, >> unsigned autosquash) >> { >> const char *shortonto, *todo_file = rebase_path_todo(); >> @@ -4754,8 +4780,8 @@ int complete_action(struct repository *r, struct >> replay_opts *opts, unsigned fla >> if (autosquash && rearrange_squash(r)) >> return -1; >> >> - if (cmd && *cmd) >> - sequencer_add_exec_commands(r, cmd); >> + if (commands->nr) >> + sequencer_add_exec_commands(r, commands); >> >> if (strbuf_read_file(buf, todo_file, 0) < 0) >> return error_errno(_("could not read '%s'."), todo_file); >> diff --git a/sequencer.h b/sequencer.h >> index 1de97f188d..e79f03e213 100644 >> --- a/sequencer.h >> +++ b/sequencer.h >> @@ -146,12 +146,13 @@ int sequencer_make_script(struct repository *r, >> FILE *out, int argc, >> const char **argv, >> unsigned flags); >> >> -int sequencer_add_exec_commands(struct repository *r, const char >> *command); >> +int sequencer_add_exec_commands(struct repository *r, >> + struct string_list *commands); >> int transform_todo_file(struct repository *r, unsigned flags); >> int check_todo_list_from_file(struct repository *r); >> int complete_action(struct repository *r, struct replay_opts *opts, >> unsigned flags, >> const char *shortrevisions, const char *onto_name, >> - const char *onto, const char *orig_head, const char *cmd, >> + const char *onto, const char *orig_head, struct >> string_list *commands, >> unsigned autosquash); >> int rearrange_squash(struct repository *r); >> >> >
Le 01/02/2019 à 15:51, Phillip Wood a écrit : >>> @@ -4473,38 +4475,62 @@ int sequencer_add_exec_commands(struct >>> repository *r, >>> * those chains if there are any. >>> */ >>> insert = -1; >>> - for (i = 0; i < todo_list.nr; i++) { >>> - enum todo_command command = todo_list.items[i].command; >>> - >>> - if (insert >= 0) { >>> - /* skip fixup/squash chains */ >>> - if (command == TODO_COMMENT) >>> - continue; >>> - else if (is_fixup(command)) { >>> - insert = i + 1; >>> - continue; >>> - } >>> - strbuf_insert(buf, >>> - todo_list.items[insert].offset_in_buf + >>> - offset, commands, commands_len); >> >> In a todo list that looks like >> pick abc message >> #pick cde empty commit >> This inserts the exec command for the first pick above the commented >> out pick. I think your translation puts it below the commented out >> pick as it ignores the value of insert. I think it's probably easiest >> to add an INSERT_ARRAY macro to insert it in the right place. An >> alternative might be to track the last insert position and only copy >> commands across when there is another exec to insert but that might >> get complicated in cases such as >> >> pick abc message >> #squash cde squash! message //empty commit for rewording >> fixup 123 fixup! message >> #pick 456 empty commit > > Thinking about this, I'm not sure it can happen as the empty squash > commit will be commented out before rearrange_squash() is called so I > think it would actually look like > > pick abc message > fixup 123 fixup! message > #pick cde squash! message > #pick 456 empty commit > > So I wonder if we can get away with treating insert as a flag and > inserting exec commands when > insert && command == TODO_COMMENT > > We could probably do with some tests for this to be sure. > I will try this approach. -- Alban > Best Wishes > > Phillip >
Hi Alban On 29/01/2019 15:01, Alban Gruin wrote: > This refactors sequencer_add_exec_commands() to work on a todo_list to > avoid redundant reads and writes to the disk. > > Instead of inserting the `exec' commands between the other commands and > re-parsing the buffer at the end, they are appended to the buffer once, > and a new list of items is created. Items from the old list are copied > across and new `exec' items are appended when necessary. This > eliminates the need to reparse the buffer, but this also means we have > to use todo_list_write_to_disk() to write the file. > > todo_list_add_exec_commands() and sequencer_add_exec_commands() are > modified to take a string list instead of a string -- one item for each > command. This makes it easier to insert a new command to the todo list > for each command to execute. > > sequencer_add_exec_commands() still reads the todo list from the disk, > as it is needed by rebase -p. > > complete_action() still uses sequencer_add_exec_commands() for now. > This will be changed in a future commit. > > Signed-off-by: Alban Gruin <alban.gruin@gmail.com> > --- > builtin/rebase--interactive.c | 15 +++-- > sequencer.c | 110 +++++++++++++++++++++------------- > sequencer.h | 5 +- > 3 files changed, 82 insertions(+), 48 deletions(-) > > diff --git a/builtin/rebase--interactive.c b/builtin/rebase--interactive.c > index df19ccaeb9..53056ee713 100644 > --- a/builtin/rebase--interactive.c > +++ b/builtin/rebase--interactive.c > @@ -65,7 +65,7 @@ static int do_interactive_rebase(struct replay_opts *opts, unsigned flags, > const char *onto, const char *onto_name, > const char *squash_onto, const char *head_name, > const char *restrict_revision, char *raw_strategies, > - const char *cmd, unsigned autosquash) > + struct string_list *commands, unsigned autosquash) > { > int ret; > const char *head_hash = NULL; > @@ -116,7 +116,7 @@ static int do_interactive_rebase(struct replay_opts *opts, unsigned flags, > discard_cache(); > ret = complete_action(the_repository, opts, flags, > shortrevisions, onto_name, onto, > - head_hash, cmd, autosquash); > + head_hash, commands, autosquash); > } > > free(revisions); > @@ -139,6 +139,7 @@ int cmd_rebase__interactive(int argc, const char **argv, const char *prefix) > const char *onto = NULL, *onto_name = NULL, *restrict_revision = NULL, > *squash_onto = NULL, *upstream = NULL, *head_name = NULL, > *switch_to = NULL, *cmd = NULL; > + struct string_list commands = STRING_LIST_INIT_DUP; This is leaked, I think we should call string_list_clear() at the end. Best Wishes Phillip > char *raw_strategies = NULL; > enum { > NONE = 0, CONTINUE, SKIP, EDIT_TODO, SHOW_CURRENT_PATCH, > @@ -221,6 +222,12 @@ int cmd_rebase__interactive(int argc, const char **argv, const char *prefix) > warning(_("--[no-]rebase-cousins has no effect without " > "--rebase-merges")); > > + if (cmd && *cmd) { > + string_list_split(&commands, cmd, '\n', -1); > + if (strlen(commands.items[commands.nr - 1].string) == 0) > + --commands.nr; > + } > + > switch (command) { > case NONE: > if (!onto && !upstream) > @@ -228,7 +235,7 @@ int cmd_rebase__interactive(int argc, const char **argv, const char *prefix) > > ret = do_interactive_rebase(&opts, flags, switch_to, upstream, onto, > onto_name, squash_onto, head_name, restrict_revision, > - raw_strategies, cmd, autosquash); > + raw_strategies, &commands, autosquash); > break; > case SKIP: { > struct string_list merge_rr = STRING_LIST_INIT_DUP; > @@ -262,7 +269,7 @@ int cmd_rebase__interactive(int argc, const char **argv, const char *prefix) > ret = rearrange_squash(the_repository); > break; > case ADD_EXEC: > - ret = sequencer_add_exec_commands(the_repository, cmd); > + ret = sequencer_add_exec_commands(the_repository, &commands); > break; > default: > BUG("invalid command '%d'", command); > diff --git a/sequencer.c b/sequencer.c > index 266f80d704..3a90b419d7 100644 > --- a/sequencer.c > +++ b/sequencer.c > @@ -4446,25 +4446,27 @@ int sequencer_make_script(struct repository *r, FILE *out, > return 0; > } > > -/* > - * Add commands after pick and (series of) squash/fixup commands > - * in the todo list. > - */ > -int sequencer_add_exec_commands(struct repository *r, > - const char *commands) > +static void todo_list_add_exec_commands(struct todo_list *todo_list, > + struct string_list *commands) > { > - const char *todo_file = rebase_path_todo(); > - struct todo_list todo_list = TODO_LIST_INIT; > - struct strbuf *buf = &todo_list.buf; > - size_t offset = 0, commands_len = strlen(commands); > - int i, insert; > + struct strbuf *buf = &todo_list->buf; > + size_t base_offset = buf->len; > + int i, insert, nr = 0, alloc = 0; > + struct todo_item *items = NULL, *base_items = NULL; > > - if (strbuf_read_file(&todo_list.buf, todo_file, 0) < 0) > - return error(_("could not read '%s'."), todo_file); > + base_items = xcalloc(commands->nr, sizeof(struct todo_item)); > + for (i = 0; i < commands->nr; ++i) { > + size_t command_len = strlen(commands->items[i].string); > > - if (todo_list_parse_insn_buffer(r, todo_list.buf.buf, &todo_list)) { > - todo_list_release(&todo_list); > - return error(_("unusable todo list: '%s'"), todo_file); > + strbuf_addstr(buf, commands->items[i].string); > + strbuf_addch(buf, '\n'); > + > + base_items[i].command = TODO_EXEC; > + base_items[i].offset_in_buf = base_offset; > + base_items[i].arg_offset = base_offset + strlen("exec "); > + base_items[i].arg_len = command_len - strlen("exec "); > + > + base_offset += command_len + 1; > } > > /* > @@ -4473,38 +4475,62 @@ int sequencer_add_exec_commands(struct repository *r, > * those chains if there are any. > */ > insert = -1; > - for (i = 0; i < todo_list.nr; i++) { > - enum todo_command command = todo_list.items[i].command; > - > - if (insert >= 0) { > - /* skip fixup/squash chains */ > - if (command == TODO_COMMENT) > - continue; > - else if (is_fixup(command)) { > - insert = i + 1; > - continue; > - } > - strbuf_insert(buf, > - todo_list.items[insert].offset_in_buf + > - offset, commands, commands_len); > - offset += commands_len; > + for (i = 0; i < todo_list->nr; i++) { > + enum todo_command command = todo_list->items[i].command; > + if (insert >= 0 && command != TODO_COMMENT && !is_fixup(command)) { > + ALLOC_GROW(items, nr + commands->nr, alloc); > + COPY_ARRAY(items + nr, base_items, commands->nr); > + nr += commands->nr; > insert = -1; > } > > - if (command == TODO_PICK || command == TODO_MERGE) > + ALLOC_GROW(items, nr + 1, alloc); > + items[nr++] = todo_list->items[i]; > + > + if (command == TODO_PICK || command == TODO_MERGE || is_fixup(command)) > insert = i + 1; > } > > /* insert or append final <commands> */ > - if (insert >= 0 && insert < todo_list.nr) > - strbuf_insert(buf, todo_list.items[insert].offset_in_buf + > - offset, commands, commands_len); > - else if (insert >= 0 || !offset) > - strbuf_add(buf, commands, commands_len); > + if (insert >= 0 || nr == todo_list->nr) { > + ALLOC_GROW(items, nr + commands->nr, alloc); > + COPY_ARRAY(items + nr, base_items, commands->nr); > + nr += commands->nr; > + } > + > + free(base_items); > + FREE_AND_NULL(todo_list->items); > + todo_list->items = items; > + todo_list->nr = nr; > + todo_list->alloc = alloc; > +} > > - i = write_message(buf->buf, buf->len, todo_file, 0); > +/* > + * Add commands after pick and (series of) squash/fixup commands > + * in the todo list. > + */ > +int sequencer_add_exec_commands(struct repository *r, > + struct string_list *commands) > +{ > + const char *todo_file = rebase_path_todo(); > + struct todo_list todo_list = TODO_LIST_INIT; > + int res; > + > + if (strbuf_read_file(&todo_list.buf, todo_file, 0) < 0) > + return error_errno(_("could not read '%s'."), todo_file); > + > + if (todo_list_parse_insn_buffer(r, todo_list.buf.buf, &todo_list)) { > + todo_list_release(&todo_list); > + return error(_("unusable todo list: '%s'"), todo_file); > + } > + > + todo_list_add_exec_commands(&todo_list, commands); > + res = todo_list_write_to_file(r, &todo_list, todo_file, NULL, NULL, -1, 0); > todo_list_release(&todo_list); > - return i; > + > + if (res) > + return error_errno(_("could not write '%s'."), todo_file); > + return 0; > } > > static void todo_list_to_strbuf(struct repository *r, struct todo_list *todo_list, > @@ -4735,7 +4761,7 @@ static int skip_unnecessary_picks(struct repository *r, struct object_id *output > > int complete_action(struct repository *r, struct replay_opts *opts, unsigned flags, > const char *shortrevisions, const char *onto_name, > - const char *onto, const char *orig_head, const char *cmd, > + const char *onto, const char *orig_head, struct string_list *commands, > unsigned autosquash) > { > const char *shortonto, *todo_file = rebase_path_todo(); > @@ -4754,8 +4780,8 @@ int complete_action(struct repository *r, struct replay_opts *opts, unsigned fla > if (autosquash && rearrange_squash(r)) > return -1; > > - if (cmd && *cmd) > - sequencer_add_exec_commands(r, cmd); > + if (commands->nr) > + sequencer_add_exec_commands(r, commands); > > if (strbuf_read_file(buf, todo_file, 0) < 0) > return error_errno(_("could not read '%s'."), todo_file); > diff --git a/sequencer.h b/sequencer.h > index 1de97f188d..e79f03e213 100644 > --- a/sequencer.h > +++ b/sequencer.h > @@ -146,12 +146,13 @@ int sequencer_make_script(struct repository *r, FILE *out, int argc, > const char **argv, > unsigned flags); > > -int sequencer_add_exec_commands(struct repository *r, const char *command); > +int sequencer_add_exec_commands(struct repository *r, > + struct string_list *commands); > int transform_todo_file(struct repository *r, unsigned flags); > int check_todo_list_from_file(struct repository *r); > int complete_action(struct repository *r, struct replay_opts *opts, unsigned flags, > const char *shortrevisions, const char *onto_name, > - const char *onto, const char *orig_head, const char *cmd, > + const char *onto, const char *orig_head, struct string_list *commands, > unsigned autosquash); > int rearrange_squash(struct repository *r); > >
diff --git a/builtin/rebase--interactive.c b/builtin/rebase--interactive.c index df19ccaeb9..53056ee713 100644 --- a/builtin/rebase--interactive.c +++ b/builtin/rebase--interactive.c @@ -65,7 +65,7 @@ static int do_interactive_rebase(struct replay_opts *opts, unsigned flags, const char *onto, const char *onto_name, const char *squash_onto, const char *head_name, const char *restrict_revision, char *raw_strategies, - const char *cmd, unsigned autosquash) + struct string_list *commands, unsigned autosquash) { int ret; const char *head_hash = NULL; @@ -116,7 +116,7 @@ static int do_interactive_rebase(struct replay_opts *opts, unsigned flags, discard_cache(); ret = complete_action(the_repository, opts, flags, shortrevisions, onto_name, onto, - head_hash, cmd, autosquash); + head_hash, commands, autosquash); } free(revisions); @@ -139,6 +139,7 @@ int cmd_rebase__interactive(int argc, const char **argv, const char *prefix) const char *onto = NULL, *onto_name = NULL, *restrict_revision = NULL, *squash_onto = NULL, *upstream = NULL, *head_name = NULL, *switch_to = NULL, *cmd = NULL; + struct string_list commands = STRING_LIST_INIT_DUP; char *raw_strategies = NULL; enum { NONE = 0, CONTINUE, SKIP, EDIT_TODO, SHOW_CURRENT_PATCH, @@ -221,6 +222,12 @@ int cmd_rebase__interactive(int argc, const char **argv, const char *prefix) warning(_("--[no-]rebase-cousins has no effect without " "--rebase-merges")); + if (cmd && *cmd) { + string_list_split(&commands, cmd, '\n', -1); + if (strlen(commands.items[commands.nr - 1].string) == 0) + --commands.nr; + } + switch (command) { case NONE: if (!onto && !upstream) @@ -228,7 +235,7 @@ int cmd_rebase__interactive(int argc, const char **argv, const char *prefix) ret = do_interactive_rebase(&opts, flags, switch_to, upstream, onto, onto_name, squash_onto, head_name, restrict_revision, - raw_strategies, cmd, autosquash); + raw_strategies, &commands, autosquash); break; case SKIP: { struct string_list merge_rr = STRING_LIST_INIT_DUP; @@ -262,7 +269,7 @@ int cmd_rebase__interactive(int argc, const char **argv, const char *prefix) ret = rearrange_squash(the_repository); break; case ADD_EXEC: - ret = sequencer_add_exec_commands(the_repository, cmd); + ret = sequencer_add_exec_commands(the_repository, &commands); break; default: BUG("invalid command '%d'", command); diff --git a/sequencer.c b/sequencer.c index 266f80d704..3a90b419d7 100644 --- a/sequencer.c +++ b/sequencer.c @@ -4446,25 +4446,27 @@ int sequencer_make_script(struct repository *r, FILE *out, return 0; } -/* - * Add commands after pick and (series of) squash/fixup commands - * in the todo list. - */ -int sequencer_add_exec_commands(struct repository *r, - const char *commands) +static void todo_list_add_exec_commands(struct todo_list *todo_list, + struct string_list *commands) { - const char *todo_file = rebase_path_todo(); - struct todo_list todo_list = TODO_LIST_INIT; - struct strbuf *buf = &todo_list.buf; - size_t offset = 0, commands_len = strlen(commands); - int i, insert; + struct strbuf *buf = &todo_list->buf; + size_t base_offset = buf->len; + int i, insert, nr = 0, alloc = 0; + struct todo_item *items = NULL, *base_items = NULL; - if (strbuf_read_file(&todo_list.buf, todo_file, 0) < 0) - return error(_("could not read '%s'."), todo_file); + base_items = xcalloc(commands->nr, sizeof(struct todo_item)); + for (i = 0; i < commands->nr; ++i) { + size_t command_len = strlen(commands->items[i].string); - if (todo_list_parse_insn_buffer(r, todo_list.buf.buf, &todo_list)) { - todo_list_release(&todo_list); - return error(_("unusable todo list: '%s'"), todo_file); + strbuf_addstr(buf, commands->items[i].string); + strbuf_addch(buf, '\n'); + + base_items[i].command = TODO_EXEC; + base_items[i].offset_in_buf = base_offset; + base_items[i].arg_offset = base_offset + strlen("exec "); + base_items[i].arg_len = command_len - strlen("exec "); + + base_offset += command_len + 1; } /* @@ -4473,38 +4475,62 @@ int sequencer_add_exec_commands(struct repository *r, * those chains if there are any. */ insert = -1; - for (i = 0; i < todo_list.nr; i++) { - enum todo_command command = todo_list.items[i].command; - - if (insert >= 0) { - /* skip fixup/squash chains */ - if (command == TODO_COMMENT) - continue; - else if (is_fixup(command)) { - insert = i + 1; - continue; - } - strbuf_insert(buf, - todo_list.items[insert].offset_in_buf + - offset, commands, commands_len); - offset += commands_len; + for (i = 0; i < todo_list->nr; i++) { + enum todo_command command = todo_list->items[i].command; + if (insert >= 0 && command != TODO_COMMENT && !is_fixup(command)) { + ALLOC_GROW(items, nr + commands->nr, alloc); + COPY_ARRAY(items + nr, base_items, commands->nr); + nr += commands->nr; insert = -1; } - if (command == TODO_PICK || command == TODO_MERGE) + ALLOC_GROW(items, nr + 1, alloc); + items[nr++] = todo_list->items[i]; + + if (command == TODO_PICK || command == TODO_MERGE || is_fixup(command)) insert = i + 1; } /* insert or append final <commands> */ - if (insert >= 0 && insert < todo_list.nr) - strbuf_insert(buf, todo_list.items[insert].offset_in_buf + - offset, commands, commands_len); - else if (insert >= 0 || !offset) - strbuf_add(buf, commands, commands_len); + if (insert >= 0 || nr == todo_list->nr) { + ALLOC_GROW(items, nr + commands->nr, alloc); + COPY_ARRAY(items + nr, base_items, commands->nr); + nr += commands->nr; + } + + free(base_items); + FREE_AND_NULL(todo_list->items); + todo_list->items = items; + todo_list->nr = nr; + todo_list->alloc = alloc; +} - i = write_message(buf->buf, buf->len, todo_file, 0); +/* + * Add commands after pick and (series of) squash/fixup commands + * in the todo list. + */ +int sequencer_add_exec_commands(struct repository *r, + struct string_list *commands) +{ + const char *todo_file = rebase_path_todo(); + struct todo_list todo_list = TODO_LIST_INIT; + int res; + + if (strbuf_read_file(&todo_list.buf, todo_file, 0) < 0) + return error_errno(_("could not read '%s'."), todo_file); + + if (todo_list_parse_insn_buffer(r, todo_list.buf.buf, &todo_list)) { + todo_list_release(&todo_list); + return error(_("unusable todo list: '%s'"), todo_file); + } + + todo_list_add_exec_commands(&todo_list, commands); + res = todo_list_write_to_file(r, &todo_list, todo_file, NULL, NULL, -1, 0); todo_list_release(&todo_list); - return i; + + if (res) + return error_errno(_("could not write '%s'."), todo_file); + return 0; } static void todo_list_to_strbuf(struct repository *r, struct todo_list *todo_list, @@ -4735,7 +4761,7 @@ static int skip_unnecessary_picks(struct repository *r, struct object_id *output int complete_action(struct repository *r, struct replay_opts *opts, unsigned flags, const char *shortrevisions, const char *onto_name, - const char *onto, const char *orig_head, const char *cmd, + const char *onto, const char *orig_head, struct string_list *commands, unsigned autosquash) { const char *shortonto, *todo_file = rebase_path_todo(); @@ -4754,8 +4780,8 @@ int complete_action(struct repository *r, struct replay_opts *opts, unsigned fla if (autosquash && rearrange_squash(r)) return -1; - if (cmd && *cmd) - sequencer_add_exec_commands(r, cmd); + if (commands->nr) + sequencer_add_exec_commands(r, commands); if (strbuf_read_file(buf, todo_file, 0) < 0) return error_errno(_("could not read '%s'."), todo_file); diff --git a/sequencer.h b/sequencer.h index 1de97f188d..e79f03e213 100644 --- a/sequencer.h +++ b/sequencer.h @@ -146,12 +146,13 @@ int sequencer_make_script(struct repository *r, FILE *out, int argc, const char **argv, unsigned flags); -int sequencer_add_exec_commands(struct repository *r, const char *command); +int sequencer_add_exec_commands(struct repository *r, + struct string_list *commands); int transform_todo_file(struct repository *r, unsigned flags); int check_todo_list_from_file(struct repository *r); int complete_action(struct repository *r, struct replay_opts *opts, unsigned flags, const char *shortrevisions, const char *onto_name, - const char *onto, const char *orig_head, const char *cmd, + const char *onto, const char *orig_head, struct string_list *commands, unsigned autosquash); int rearrange_squash(struct repository *r);
This refactors sequencer_add_exec_commands() to work on a todo_list to avoid redundant reads and writes to the disk. Instead of inserting the `exec' commands between the other commands and re-parsing the buffer at the end, they are appended to the buffer once, and a new list of items is created. Items from the old list are copied across and new `exec' items are appended when necessary. This eliminates the need to reparse the buffer, but this also means we have to use todo_list_write_to_disk() to write the file. todo_list_add_exec_commands() and sequencer_add_exec_commands() are modified to take a string list instead of a string -- one item for each command. This makes it easier to insert a new command to the todo list for each command to execute. sequencer_add_exec_commands() still reads the todo list from the disk, as it is needed by rebase -p. complete_action() still uses sequencer_add_exec_commands() for now. This will be changed in a future commit. Signed-off-by: Alban Gruin <alban.gruin@gmail.com> --- builtin/rebase--interactive.c | 15 +++-- sequencer.c | 110 +++++++++++++++++++++------------- sequencer.h | 5 +- 3 files changed, 82 insertions(+), 48 deletions(-)