我的第一个golang command脚本

场景

维护数据库的树形结构的pid集合

具体代码

/**
  @author: Administrator
  @since: 2022/9/29
  @desc: 处理category的 脚本暂时不写
**/

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"` // 上级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
    }
    // 处理数据
    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 出现错误 sql %s ", category.Id, UpdateQuery))
        }
    }

    f.resetPidList(list)
    return f.fixPidList()
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 217,509评论 6 504
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,806评论 3 394
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 163,875评论 0 354
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,441评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,488评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,365评论 1 302
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,190评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,062评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,500评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,706评论 3 335
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,834评论 1 347
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,559评论 5 345
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,167评论 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,779评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,912评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,958评论 2 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,779评论 2 354

推荐阅读更多精彩内容

  • 1. 书籍《Objective-C编程全解》 [知识点] 面向对象编程的概念 类和继承 对象的类型和动态绑定 内存...
    vikimemory阅读 639评论 0 1
  • 1.软件测试级别? 单元测试:单元测试是对软件组成单元进行测试。其目的是检验软件基本组成单位的正确性。测试的对象是...
    听闻白依阅读 1,413评论 0 9
  • 程序设计中常使用树型结构来表征某些数据的关联关系,如上下级、栏目结构、商品分类、菜单、回复等。 分类的层级关系可以...
    JunChow520阅读 4,098评论 4 3
  • 六、极客时间 1、Exception和Error有什么区别? Error类和Exception类的父类都是thro...
    想54256阅读 774评论 0 1
  • 基本概念及操作 常用快捷键 常用通配符 所有的手册页遵循一个常见的布局,为了通过简单的 ASCII 文本展示而被优...
    Jachin111阅读 216评论 0 1