我的第一個(gè)golang command腳本

場(chǎng)景

維護(hù)數(shù)據(jù)庫(kù)的樹(shù)形結(jié)構(gòu)的pid集合

具體代碼

/**
  @author: Administrator
  @since: 2022/9/29
  @desc: 處理category的 腳本暫時(shí)不寫
**/

package main

import (
    "context"
    "database/sql"
    "fmt"
    "strconv"
    "strings"
    "time"

    "github.com/Masterminds/squirrel"
    "github.com/zeromicro/go-zero/core/stores/builder"
    "github.com/zeromicro/go-zero/core/stores/sqlx"
)

var (
    categoryFieldNames = builder.RawFieldNames(&Category{})
    categoryRows       = strings.Join(categoryFieldNames, ",")
)

type (
    Fixer struct {
        ctx        context.Context
        sqlConn    sqlx.SqlConn
        pidListMap map[int]string
        pidList    []int
    }

    Category struct {
        Id         int          `db:"id"`
        Pid        int          `db:"pid"` // 上級(jí)id
        Type       string       `db:"type"`
        Name       string       `db:"name"` // 類型名稱
        Icon       string       `db:"icon"` // icon 路徑
        Sort       int64        `db:"sort"` // 排序
        Ext1       string       `db:"ext1"`
        Ext2       string       `db:"ext2"`
        Ext3       string       `db:"ext3"`
        PidList    string       `db:"pid_list"`
        CreateTime time.Time    `db:"create_time"`
        UpdateTime time.Time    `db:"update_time"`
        DeleteTime sql.NullTime `db:"delete_time"`
    }
)

func newFixer() *Fixer {
    sqlConn := sqlx.NewMysql("iot:STfbeXiRp4Pp2xYZ@tcp(175.128.51.74:1111)/iot?charset=utf8mb4&parseTime=true&loc=Asia%2FShanghai")
    pidlist := make([]int, 0)
    pidListMap := make(map[int]string, 0)
    return &Fixer{sqlConn: sqlConn, ctx: context.Background(), pidListMap: pidListMap, pidList: pidlist}
}

func main() {
    fixer := newFixer()
    fixer.fixPidList()
}

// 重置pidList 保留上一輪
func (f *Fixer) resetPidList(list []Category) {

    f.pidListMap = make(map[int]string, 0)
    f.pidList = make([]int, 0)

    for _, category := range list {
        f.pidListMap[category.Id] = category.PidList
        f.pidList = append(f.pidList, category.Id)
    }
}

func (f *Fixer) fixPidList() error {

    var list []Category

    if len(f.pidListMap) == 0 {
        sqlBuilderList := squirrel.Select(categoryRows).From("category").Where(" pid = 0")
        listQuery, listvalues, _ := sqlBuilderList.OrderBy("sort DESC").ToSql()
        // 查詢列表
        f.sqlConn.QueryRowsCtx(f.ctx, &list, listQuery, listvalues...)
        // 裝載 id  和 k-v
        f.resetPidList(list)
        return f.fixPidList()
    }

    sqlBuilderList := squirrel.Select(categoryRows).From("category").Where(squirrel.Eq{"pid": f.pidList})
    listQuery, listvalues, _ := sqlBuilderList.OrderBy("sort DESC").ToSql()
    // 查詢列表
    f.sqlConn.QueryRowsCtx(f.ctx, &list, listQuery, listvalues...)

    if len(list) <= 0 {
        fmt.Sprintf("全部處理完畢")
        return nil
    }
    // 處理數(shù)據(jù)
    for i, category := range list {

        if category.Pid == 0 {
            continue
        }

        pidList := fmt.Sprintf("%s,%s", f.pidListMap[category.Pid], strconv.Itoa(category.Pid))

        list[i].PidList = pidList

        UpdateQuery, Updatevalues, _ := squirrel.Update("category").Set("pid_list", pidList).Where("id = ?", category.Id).ToSql()
        _, err := f.sqlConn.ExecCtx(f.ctx, UpdateQuery, Updatevalues...)
        if err != nil {
            fmt.Println(err)
            panic(fmt.Sprintf("修改 %s 出現(xiàn)錯(cuò)誤 sql %s ", category.Id, UpdateQuery))
        }
    }

    f.resetPidList(list)
    return f.fixPidList()
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市赖草,隨后出現(xiàn)的幾起案子粘姜,更是在濱河造成了極大的恐慌蛛倦,老刑警劉巖,帶你破解...
    沈念sama閱讀 216,544評(píng)論 6 501
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件溯捆,死亡現(xiàn)場(chǎng)離奇詭異勋拟,居然都是意外死亡,警方通過(guò)查閱死者的電腦和手機(jī)悼枢,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,430評(píng)論 3 392
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)脾拆,“玉大人馒索,你說(shuō)我怎么就攤上這事〖偕ィ” “怎么了双揪?”我有些...
    開(kāi)封第一講書人閱讀 162,764評(píng)論 0 353
  • 文/不壞的土叔 我叫張陵动羽,是天一觀的道長(zhǎng)包帚。 經(jīng)常有香客問(wèn)我,道長(zhǎng)运吓,這世上最難降的妖魔是什么渴邦? 我笑而不...
    開(kāi)封第一講書人閱讀 58,193評(píng)論 1 292
  • 正文 為了忘掉前任,我火速辦了婚禮拘哨,結(jié)果婚禮上谋梭,老公的妹妹穿的比我還像新娘。我一直安慰自己倦青,他們只是感情好瓮床,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,216評(píng)論 6 388
  • 文/花漫 我一把揭開(kāi)白布。 她就那樣靜靜地躺著产镐,像睡著了一般隘庄。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上癣亚,一...
    開(kāi)封第一講書人閱讀 51,182評(píng)論 1 299
  • 那天丑掺,我揣著相機(jī)與錄音,去河邊找鬼述雾。 笑死街州,一個(gè)胖子當(dāng)著我的面吹牛兼丰,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播唆缴,決...
    沈念sama閱讀 40,063評(píng)論 3 418
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼鳍征,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來(lái)了琐谤?” 一聲冷哼從身側(cè)響起蟆技,我...
    開(kāi)封第一講書人閱讀 38,917評(píng)論 0 274
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎斗忌,沒(méi)想到半個(gè)月后质礼,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,329評(píng)論 1 310
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡织阳,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,543評(píng)論 2 332
  • 正文 我和宋清朗相戀三年眶蕉,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片唧躲。...
    茶點(diǎn)故事閱讀 39,722評(píng)論 1 348
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡造挽,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出弄痹,到底是詐尸還是另有隱情饭入,我是刑警寧澤,帶...
    沈念sama閱讀 35,425評(píng)論 5 343
  • 正文 年R本政府宣布肛真,位于F島的核電站谐丢,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏蚓让。R本人自食惡果不足惜乾忱,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,019評(píng)論 3 326
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望历极。 院中可真熱鬧窄瘟,春花似錦、人聲如沸趟卸。這莊子的主人今日做“春日...
    開(kāi)封第一講書人閱讀 31,671評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)锄列。三九已至图云,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間右蕊,已是汗流浹背琼稻。 一陣腳步聲響...
    開(kāi)封第一講書人閱讀 32,825評(píng)論 1 269
  • 我被黑心中介騙來(lái)泰國(guó)打工, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留饶囚,地道東北人帕翻。 一個(gè)月前我還...
    沈念sama閱讀 47,729評(píng)論 2 368
  • 正文 我出身青樓鸠补,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親嘀掸。 傳聞我的和親對(duì)象是個(gè)殘疾皇子紫岩,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,614評(píng)論 2 353

推薦閱讀更多精彩內(nèi)容

  • 1. 書籍《Objective-C編程全解》 [知識(shí)點(diǎn)] 面向?qū)ο缶幊痰母拍?類和繼承 對(duì)象的類型和動(dòng)態(tài)綁定 內(nèi)存...
    vikimemory閱讀 636評(píng)論 0 1
  • 1.軟件測(cè)試級(jí)別? 單元測(cè)試:?jiǎn)卧獪y(cè)試是對(duì)軟件組成單元進(jìn)行測(cè)試睬塌。其目的是檢驗(yàn)軟件基本組成單位的正確性泉蝌。測(cè)試的對(duì)象是...
    聽(tīng)聞白依閱讀 1,410評(píng)論 0 9
  • 程序設(shè)計(jì)中常使用樹(shù)型結(jié)構(gòu)來(lái)表征某些數(shù)據(jù)的關(guān)聯(lián)關(guān)系,如上下級(jí)揩晴、欄目結(jié)構(gòu)勋陪、商品分類、菜單硫兰、回復(fù)等诅愚。 分類的層級(jí)關(guān)系可以...
    JunChow520閱讀 4,098評(píng)論 4 3
  • 六、極客時(shí)間 1劫映、Exception和Error有什么區(qū)別违孝? Error類和Exception類的父類都是thro...
    想54256閱讀 774評(píng)論 0 1
  • 基本概念及操作 常用快捷鍵 常用通配符 所有的手冊(cè)頁(yè)遵循一個(gè)常見(jiàn)的布局,為了通過(guò)簡(jiǎn)單的 ASCII 文本展示而被優(yōu)...
    Jachin111閱讀 216評(píng)論 0 1