Always handle non-normal (per-worktree or pseudo) refs in the files backend instead of alternate backends.
Sometimes a ref transaction will update both a per-worktree ref and a normal ref. For instance, an ordinary commit might update refs/heads/master and HEAD (or at least HEAD's reflog). Updates to normal refs continue to go through the chosen backend. Updates to non-normal refs are moved to a separate files backend transaction. Signed-off-by: David Turner <dtur...@twopensource.com> --- refs.c | 110 +++++++++++++++++++++++++++++++++++++++++++++++++++++------------ 1 file changed, 90 insertions(+), 20 deletions(-) diff --git a/refs.c b/refs.c index bdde808..b2acf0e 100644 --- a/refs.c +++ b/refs.c @@ -9,6 +9,11 @@ #include "object.h" #include "tag.h" +static const char split_transaction_fail_warning[] = N_( + "A ref transaction was split across two refs backends. Part of the " + "transaction succeeded, but then the update to the per-worktree refs " + "failed. Your repository may be in an inconsistent state."); + /* * We always have a files backend and it is the default. */ @@ -791,6 +796,13 @@ void ref_transaction_free(struct ref_transaction *transaction) free(transaction); } +static void add_update_obj(struct ref_transaction *transaction, + struct ref_update *update) +{ + ALLOC_GROW(transaction->updates, transaction->nr + 1, transaction->alloc); + transaction->updates[transaction->nr++] = update; +} + static struct ref_update *add_update(struct ref_transaction *transaction, const char *refname) { @@ -798,8 +810,7 @@ static struct ref_update *add_update(struct ref_transaction *transaction, struct ref_update *update = xcalloc(1, sizeof(*update) + len); memcpy((char *)update->refname, refname, len); /* includes NUL */ - ALLOC_GROW(transaction->updates, transaction->nr + 1, transaction->alloc); - transaction->updates[transaction->nr++] = update; + add_update_obj(transaction, update); return update; } @@ -1416,11 +1427,37 @@ static int dereference_symrefs(struct ref_transaction *transaction, return 0; } -int ref_transaction_commit(struct ref_transaction *transaction, - struct strbuf *err) +/* + * Move all non-normal ref updates into a specially-created + * files-backend transaction + */ +static int move_abnormal_ref_updates(struct ref_transaction *transaction, + struct ref_transaction *files_transaction, + struct strbuf *err) +{ + int i, normal_nr = 0; + + for (i = 0; i < transaction->nr; i++) { + struct ref_update *update = transaction->updates[i]; + + if (ref_type(update->refname) == REF_TYPE_NORMAL) + transaction->updates[normal_nr++] = update; + else + add_update_obj(files_transaction, update); + } + + transaction->nr = normal_nr; + return 0; +} + +static int do_ref_transaction_commit(struct ref_transaction *transaction, + struct strbuf *err, + ref_transaction_commit_fn *commit_fn) { int ret = -1; struct string_list affected_refnames = STRING_LIST_INIT_NODUP; + struct string_list files_affected_refnames = STRING_LIST_INIT_NODUP; + struct ref_transaction *files_transaction = NULL; assert(err); @@ -1436,18 +1473,60 @@ int ref_transaction_commit(struct ref_transaction *transaction, if (ret) goto done; + if (the_refs_backend != &refs_be_files) { + files_transaction = ref_transaction_begin(err); + if (!files_transaction) + goto done; + + ret = move_abnormal_ref_updates(transaction, files_transaction, + err); + if (ret) + goto done; + + /* files backend commit */ + if (get_affected_refnames(files_transaction, + &files_affected_refnames, + err)) { + ret = TRANSACTION_GENERIC_ERROR; + goto done; + } + } + + /* main backend commit */ if (get_affected_refnames(transaction, &affected_refnames, err)) { ret = TRANSACTION_GENERIC_ERROR; goto done; } - ret = the_refs_backend->transaction_commit(transaction, - &affected_refnames, err); + ret = commit_fn(transaction, &affected_refnames, err); + if (ret) + goto done; + + if (files_transaction) { + ret = refs_be_files.transaction_commit(files_transaction, + &files_affected_refnames, + err); + if (ret) { + warning(split_transaction_fail_warning); + goto done; + } + } + done: string_list_clear(&affected_refnames, 0); + string_list_clear(&files_affected_refnames, 0); + if (files_transaction) + ref_transaction_free(files_transaction); return ret; } +int ref_transaction_commit(struct ref_transaction *transaction, + struct strbuf *err) +{ + return do_ref_transaction_commit(transaction, err, + the_refs_backend->transaction_commit); +} + int verify_refname_available(const char *refname, struct string_list *extra, struct string_list *skip, struct strbuf *err) { @@ -1467,6 +1546,9 @@ int peel_ref(const char *refname, unsigned char *sha1) int create_symref(const char *ref_target, const char *refs_heads_master, const char *logmsg) { + if (ref_type(ref_target) != REF_TYPE_NORMAL) + return refs_be_files.create_symref(ref_target, refs_heads_master, + logmsg); return the_refs_backend->create_symref(ref_target, refs_heads_master, logmsg); } @@ -1526,20 +1608,8 @@ int reflog_expire(const char *refname, const unsigned char *sha1, int initial_ref_transaction_commit(struct ref_transaction *transaction, struct strbuf *err) { - struct string_list affected_refnames = STRING_LIST_INIT_NODUP; - int ret; - - if (get_affected_refnames(transaction, - &affected_refnames, err)) { - ret = TRANSACTION_GENERIC_ERROR; - goto done; - } - ret = the_refs_backend->initial_transaction_commit(transaction, - &affected_refnames, - err); -done: - string_list_clear(&affected_refnames, 0); - return ret; + return do_ref_transaction_commit(transaction, err, + the_refs_backend->initial_transaction_commit); } int delete_refs(struct string_list *refnames) -- 2.4.2.767.g62658d5-twtrsrc -- To unsubscribe from this list: send the line "unsubscribe git" in the body of a message to majord...@vger.kernel.org More majordomo info at http://vger.kernel.org/majordomo-info.html