1. 從一次「桶里套桶」的聚合查詢說起如果你正在用 Elasticsearch 做數(shù)據(jù)統(tǒng)計(jì)大概率寫過restHighLevelClient的聚合查詢。terms分組、dateHistogram按月分桶、再往桶里塞一個(gè)sum子聚合這套組合拳在業(yè)務(wù)里非常常見。但真正讓人頭疼的不是 DSL 怎么寫而是 Java 端把SearchResponse拆開的那幾行代碼先拿父聚合再遍歷Histogram.Bucket然后bucket.getAggregations().getAsMap()最后強(qiáng)轉(zhuǎn)ParsedSum取值。層級(jí)一多別名一多ClassCastException和NullPointerException就輪番上陣。這篇就圍繞一個(gè)具體需求展開按create_time按月分桶統(tǒng)計(jì)每個(gè)月所有入庫記錄的success_count累加值。索引里每條記錄長(zhǎng)這樣uuid、file_name、type、create_time、success_count。目標(biāo)是輸出「月份 當(dāng)月成功條數(shù)總和」的列表。我會(huì)先把原始寫法講清楚再演示怎么用走 TaoToken 的 Codex 來陪你改這段代碼——讓它幫你核對(duì)父子聚合的別名、calendar_interval和拆桶取值是否對(duì)得上你回到本地 IDE 編譯運(yùn)行驗(yàn)證。適合誰看正在用restHighLevelClient寫聚合、被嵌套桶取值繞暈的后端同學(xué)想借助 AI 編碼助手提升改代碼效率、又不想把 Key 和 Base URL 配錯(cuò)的開發(fā)者。2. 前置注冊(cè) TaoToken 并創(chuàng)建 Key在讓 Codex 幫你改代碼之前先把模型接入通道配好。這一步不復(fù)雜但有兩個(gè)細(xì)節(jié)特別容易填錯(cuò)我見過不少人卡在這里。先打開官網(wǎng)注冊(cè)賬號(hào)https://taotoken.net/?utm_sourcetaotoken_aicg_blog_endutm_mediumcsdnutm_campaignrewriteutm_content注冊(cè)完成后進(jìn)入控制臺(tái)創(chuàng)建一把 API Key。Key 只在創(chuàng)建時(shí)完整展示一次復(fù)制下來存好后面配置要用。https://taotoken.net/console?utm_sourcetaotoken_aicg_blog_endutm_contentconsoleutm_campaignrewrite創(chuàng)建 Key 的入口在這里https://taotoken.net/api-keys?utm_sourcetaotoken_aicg_blog_endutm_contentapi-keysutm_campaignrewrite注意Base URL 只填到/api不要帶/v1也不要把官網(wǎng)那個(gè)帶 UTM 參數(shù)的地址填進(jìn)去。這是最常見的兩個(gè)配置錯(cuò)誤。3. 可復(fù)制配置Codex 的 Base URL 與 config.tomlCodex 的配置有兩種方式改config.toml或者用環(huán)境變量。我建議先用config.toml直觀、好排查。配置文件一般放在~/.codex/config.tomlWindows 是%USERPROFILE%\.codex\config.toml。核心是把model_provider的base_url指向 TaoToken 的 API 地址model_provider taotoken [model_providers.taotoken] name TaoToken base_url https://taotoken.net/api env_key TAOTOKEN_API_KEY然后在環(huán)境變量里放剛創(chuàng)建的那把 Keyexport TAOTOKEN_API_KEYsk-你創(chuàng)建的那把KeyWindows PowerShell 用$env:TAOTOKEN_API_KEYsk-你創(chuàng)建的那把Key如果你更習(xí)慣純環(huán)境變量方式也可以直接設(shè)置export OPENAI_BASE_URLhttps://taotoken.net/api export OPENAI_API_KEYsk-你創(chuàng)建的那把Key配好之后啟動(dòng) Codex它會(huì)讀取config.toml里的 provider 配置。這里的關(guān)鍵點(diǎn)再?gòu)?qiáng)調(diào)一次base_url是https://taotoken.net/api結(jié)尾沒有/v1。填錯(cuò)的話請(qǐng)求會(huì) 404而且報(bào)錯(cuò)信息往往不直觀容易誤以為是 Key 的問題。4. 把原始聚合代碼交給 Codex 檢查配置通了接下來是正題。先把原始需求對(duì)應(yīng)的完整代碼貼給 Codex讓它對(duì)照檢查。原始代碼大致是這樣public ListSimpleTotalCountVO dataUpdateTrend() { SearchRequest searchRequest new SearchRequest(esIndexData.getReadDataLog()); SearchSourceBuilder searchSourceBuilder new SearchSourceBuilder(); // 父聚合按月分桶 AggregationBuilder histogram AggregationBuilders.dateHistogram(DataManageConstant.TO_TIME) .field(DataManageConstant.CREATE_TIME) .format(DataManageConstant.DATE_TIME_PATTERN2) .calendarInterval(DateHistogramInterval.MONTH); // 子聚合對(duì) success_count 求和 AggregationBuilder sum AggregationBuilders.sum(DataManageConstant.MONTH_SUCCESS_SUM) .field(DataManageConstant.SUCCESS_COUNT); histogram.subAggregation(sum); searchSourceBuilder.aggregation(histogram).size(0); searchRequest.source(searchSourceBuilder); ListSimpleTotalCountVO simpleTotalCountVOS new ArrayList(); try { SearchResponse search restHighLevelClient.search(searchRequest, RequestOptions.DEFAULT); Aggregation aggregation search.getAggregations().get(DataManageConstant.TO_TIME); List? extends Histogram.Bucket buckets ((Histogram) aggregation).getBuckets(); for (Histogram.Bucket bucket : buckets) { MapString, Aggregation asMap bucket.getAggregations().getAsMap(); ParsedSum parsedSum (ParsedSum) asMap.get(DataManageConstant.MONTH_SUCCESS_SUM); double value parsedSum.getValue(); int i new Double(value).intValue(); simpleTotalCountVOS.add(SimpleTotalCountVO.builder() .name(bucket.getKeyAsString()) .value(i) .build()); } } catch (IOException e) { throw new RuntimeException(e); } return simpleTotalCountVOS; }把這段貼給 Codex 時(shí)可以這樣提問讓它聚焦在幾個(gè)易錯(cuò)點(diǎn)上這段 restHighLevelClient 的聚合查詢父聚合是 dateHistogram 按月分桶 子聚合是 sum 對(duì) success_count 求和。請(qǐng)幫我檢查 1. 父子聚合的別名是否一致取值時(shí) get 的 key 有沒有寫錯(cuò) 2. calendar_interval 用 MONTH 是否正確format 會(huì)不會(huì)影響 keyAsString 3. 拆桶時(shí) getAggregations().getAsMap() 強(qiáng)轉(zhuǎn) ParsedSum 的寫法有沒有隱患 4. size(0) 的位置對(duì)不對(duì)。Codex 通常會(huì)指出幾個(gè)點(diǎn)。第一DataManageConstant.TO_TIME和DataManageConstant.MONTH_SUCCESS_SUM這兩個(gè)別名必須和 DSL 里完全一致大小寫、下劃線都不能差否則get()返回 null強(qiáng)轉(zhuǎn)直接 NPE。第二calendarInterval(DateHistogramInterval.MONTH)在 7.x 之后推薦用calendarInterval老版本可能用interval要確認(rèn) ES 版本。第三bucket.getKeyAsString()返回的是格式化后的時(shí)間字符串如果format設(shè)成yyyy-MM-dd那 key 就是每月第一天符合預(yù)期。5. 驗(yàn)證請(qǐng)求跑通并確認(rèn)結(jié)果改完代碼后回到本地 IDE 編譯運(yùn)行。為了確認(rèn)聚合邏輯本身沒問題可以先用 Kibana 或 curl 直接跑一遍 DSL對(duì)照結(jié)果GET your_index/_search { size: 0, aggs: { toTime: { date_histogram: { field: create_time, calendar_interval: month, format: yyyy-MM-dd }, aggs: { monthSuccessSum: { sum: { field: success_count } } } } } }返回結(jié)構(gòu)里每個(gè) bucket 的key_as_string是月份monthSuccessSum.value是當(dāng)月累加值。Java 端跑出來的SimpleTotalCountVO列表應(yīng)該和這個(gè)一一對(duì)應(yīng)。如果success_count在索引里是字符串類型原始數(shù)據(jù)里它確實(shí)是8990這種帶引號(hào)的那sum聚合會(huì)報(bào)錯(cuò)或者返回 0。這時(shí)候要么在 mapping 里把它改成數(shù)值類型要么在寫入時(shí)轉(zhuǎn)換。Codex 在檢查時(shí)一般也會(huì)提醒這一點(diǎn)值得留意。提示驗(yàn)證階段如果只想快速確認(rèn)模型接入是否正?？梢韵扔媚Ｐ蛯?duì)話跑一個(gè)簡(jiǎn)單請(qǐng)求確認(rèn)通道沒問題再回到代碼調(diào)試。6. 本篇常見錯(cuò)排查報(bào)錯(cuò)一java.lang.NullPointerException在parsedSum.getValue()。八成是別名對(duì)不上。檢查AggregationBuilders.sum(monthSuccessSum)里的名字和asMap.get(monthSuccessSum)是否完全一致。常量類里如果拼寫有出入編譯不報(bào)錯(cuò)但運(yùn)行必炸。報(bào)錯(cuò)二ClassCastException: ParsedLong cannot be cast to ParsedSum。如果你用的是value_count或cardinality子聚合返回類型不是ParsedSum。本篇用的是sum所以是ParsedSum別混。報(bào)錯(cuò)三請(qǐng)求 404 或連接被拒。先查 Base URL。https://taotoken.net/api結(jié)尾不帶/v1也不要把官網(wǎng)帶 UTM 的地址填進(jìn)去。Key 用創(chuàng)建的那把環(huán)境變量名要和config.toml里的env_key對(duì)上。報(bào)錯(cuò)四calendar_interval不識(shí)別。ES 7.2 之前沒有calendar_interval要用interval。確認(rèn)你的 ES 版本或者讓 Codex 幫你按版本改寫。報(bào)錯(cuò)五sum結(jié)果為 0。success_count字段類型是 text/keyword 時(shí)無法求和。用GET your_index/_mapping確認(rèn)字段類型必要時(shí)重建索引或加fielddata不推薦最好改 mapping。7. 繼續(xù)用 Codex 陪你改代碼聚合查詢這類代碼難點(diǎn)往往不在 DSL 本身而在 Java 端拆響應(yīng)時(shí)的層級(jí)和類型轉(zhuǎn)換。把原始代碼和報(bào)錯(cuò)信息一起丟給 Codex讓它逐行核對(duì)別名、類型和取值路徑比自己在 IDE 里打斷點(diǎn)快得多。配好通道之后日常改代碼、查報(bào)錯(cuò)、寫單元測(cè)試都可以讓它參與。如果你后面要做長(zhǎng)期的編碼任務(wù)或者 Agent 類工作流可以了解下 Coding Planhttps://taotoken.net/coding-plan?utm_sourcetaotoken_aicg_blog_endutm_contentcoding-planutm_campaignrewrite接入文檔在這里配置細(xì)節(jié)和參數(shù)說明都能查到https://taotoken.net/doc?utm_sourcetaotoken_aicg_blog_endutm_contentdocutm_campaignrewrite我自己的習(xí)慣是先把 DSL 在 Kibana 里跑通確認(rèn)聚合結(jié)構(gòu)沒問題再把 Java 代碼貼給 Codex 檢查拆桶邏輯。這樣兩邊對(duì)照出錯(cuò)時(shí)能快速定位是查詢寫錯(cuò)了還是解析寫錯(cuò)了。