feat: add create or replace stage support

[lichuang]

I hereby agree to the terms of the CLA available at: https://docs.databend.com/dev/policies/cla/

Summary

feat: add create or replace stage support

• Fixes feat: add create or replace stage support #14547

Tests

• Unit Test
• Logic Test
• Benchmark Test
• No Test - Explain why

Type of change

• Bug Fix (non-breaking change which fixes an issue)
• New Feature (non-breaking change which adds functionality)
• Breaking Change (fix or feature that could cause existing functionality not to work as expected)
• Documentation Update
• Refactoring
• Performance Improvement
• Other (please describe):

---

This change is 

[drmingdrmer]

Reviewed 2 of 13 files at r1, all commit messages.
Reviewable status: 2 of 13 files reviewed, 1 unresolved discussion (waiting on @lichuang)

---

src/query/management/src/stage/stage_mgr.rs line 107 at r1 (raw file):

                &key,
                serialize_struct(&info, ErrorCode::IllegalUserStageFormat, || "")?,
            )];

Why is it written as a transaction here? Wouldn't a simple upsert be enough? For CreateIfNotExists, it would be MatchSeq::Exact(0); and for CreateOrReplace, it would be MatchSeq::GE(0).

[drmingdrmer]

Reviewable status: 2 of 13 files reviewed, 1 unresolved discussion (waiting on @lichuang)

---

src/query/management/src/stage/stage_mgr.rs line 106 at r2 (raw file):

        self.kv_api
            .upsert_kv(UpsertKVReq::new(&key, seq, val, None))
            .await?;

The result of the upsert must be checked to see if it succeeded.
And the get() before upsert is unnecessary.

Suggestion:

let seq = match create_option {
    CreateOption::CreateIfNotExists(_) => MatchSeq::Exact(0),
    CreateOption::CreateOrReplace => MatchSeq::GE(0),
};

let res = self.kv_api
    .upsert_kv(UpsertKVReq::new(&key, seq, val, None))
    .await?;

match create_option {
    CreateOption::CreateIfNotExists(false) => {
        if res.prev.is_some() {
            return Err(ErrorCode::StageAlreadyExists(format!(
                "Stage '{}' already exists.",
                info.stage_name
            )));
        }
    },
    _ => {}
}

[drmingdrmer]

Reviewed 7 of 13 files at r1, 1 of 1 files at r3, all commit messages.
Reviewable status: 10 of 13 files reviewed, all discussions resolved

[lichuang]

@soyeric128