您的位置 首页 golang

golang使用json格式实现增删查改


需求和思路

在一般的小项目或者一个小软件,例如客户端之类的小程序中,可能会需要数据的持久化.但是使用一般的数据库(Mysql)之类的不合适.使用sqlite3这种嵌入式的是个较好的方法,但是Go语言中sqlite3的库是C语言的,Cgo不支持跨平台编译.正是由于这种需求,才想到使用json格式将数据直接保存在文件中.
具体的思路是怎么样呢? 在Go语言中如果要将数据转化成json格式的话,有两种格式 struct 和 map. 如果同时需要增删查改功能的话,将map作为中间格式是比较合适的.接下来我们就来实现它.

查询操作

这种操作的实现比较简单,直接将文件中的数据读取出来,使用json库反序列化就可以了. 代码如下 :

type Product struct {    Name string `json:"name"`    Num  int    `json:"num"`}func findAll() {    ps := make([]Product, 0)    data, err := ioutil.ReadFile("./index.json")    if err != nil {        log.Fatal(err)    }    // 这里参数要指定为变量的地址    err = json.Unmarshal(data, &ps)    if err != nil {        log.Fatal(err)    }    fmt.Println(ps)}

添加操作

添加的实现实在查询的基础上的,我们需要先查询文件中的数据库,并转化为map格式,再将struct也转化为map格式(这里要使用反射),合并map,json序列化,最后保存在文件中.代码如下:

func create() {    fields := make([]map[string]interface{}, 0)        p1 := &Product{        Name: "Blog",        Num:  2,    }        _, _ = json.Marshal(p1)    // 读取文件中的数据,保存为map格式    data, _ := ioutil.ReadFile("./index.json")    err := json.Unmarshal(data, &fields)    if err != nil {        log.Fatal(err)    }        // 使用反射将struct转化为map    tp := reflect.TypeOf(p1).Elem()    vp := reflect.ValueOf(p1).Elem()    field := make(map[string]interface{}, 0)    for i := 0; i < tp.NumField(); i++ {        field1 := tp.Field(i)        field2 := vp.Field(i)        key := field1.Tag.Get("json")        field[key] = field2.Interface()    }    // 合并map    fields = append(fields, field)        // 写入文件    out, _ := json.Marshal(fields)    _ = ioutil.WriteFile("./index.json", out, 0755)}

条件查询

思路: 将struct转化为map,根据输入的条件查询.查询的结果转化为struct.代码如下:

func FindOne() {    product := &Product{}    p1 := &Product{        Name: "John",        Num:  23,    }    // 使用反射将struct转化为map    tp := reflect.TypeOf(p1).Elem()    vp := reflect.ValueOf(p1).Elem()    field := make(map[string]interface{}, 0)    for i := 0; i < tp.NumField(); i++ {        field1 := tp.Field(i)        field2 := vp.Field(i)        key := field1.Tag.Get("json")        switch field2.Kind() {        case reflect.Int:            field[key] = float64(field2.Interface().(int))        case reflect.Int8:            field[key] = float64(field2.Interface().(int8))        case reflect.Int16:            field[key] = float64(field2.Interface().(int16))        case reflect.Int32:            field[key] = float64(field2.Interface().(int32))        case reflect.Int64:            field[key] = float64(field2.Interface().(int64))        case reflect.Uint:            field[key] = float64(field2.Interface().(uint))        case reflect.Uint8:            field[key] = float64(field2.Interface().(uint8))        case reflect.Uint16:            field[key] = float64(field2.Interface().(uint16))        case reflect.Uint32:            field[key] = float64(field2.Interface().(uint32))        case reflect.Uint64:            field[key] = float64(field2.Interface().(uint64))        case reflect.Float32:            field[key] = float64(field2.Interface().(float32))        case reflect.Float64:            field[key] = field2.Interface()        default:            field[key] = field2.Interface()        }    }    _, _ = json.Marshal(p1)    // 读取文件中的数据,保存为map格式    // 数据转化为map时,数值类型的统一变成float64    data, _ := ioutil.ReadFile("./index.json")    fields := make([]map[string]interface{}, 0)    err := json.Unmarshal(data, &fields)    if err != nil {        log.Fatal(err)    }    // 查询的条件    columns := []string{"name", "num"}    length := len(columns)    for _, item := range fields {        for i := 0; i < length; i++ {            // 这里的比较需要改进            if item[columns[i]] != field[columns[i]] {                break            }            if i == length-1 {                field = item                goto OVER            }        }    }OVER:    fmt.Println(field)    out, _ := json.Marshal(field)    _ = json.Unmarshal(out, &product)    fmt.Println(product)}

修改操作

修改操作在查询操作的基础上实现, 修改操作需要有一个id值,能确定元素的唯一性.代码如下:

func Update() {    p1 := &Product{        Id:   "2bbec87025968879c3c9682abe3bf730",        Name: "John_e",        Num:  100,    }    // 使用反射将struct转化为map    tp := reflect.TypeOf(p1).Elem()    vp := reflect.ValueOf(p1).Elem()    field := make(map[string]interface{}, 0)    for i := 0; i < tp.NumField(); i++ {        field1 := tp.Field(i)        field2 := vp.Field(i)        key := field1.Tag.Get("json")        switch field2.Kind() {        case reflect.Int:            field[key] = float64(field2.Interface().(int))        case reflect.Int8:            field[key] = float64(field2.Interface().(int8))        case reflect.Int16:            field[key] = float64(field2.Interface().(int16))        case reflect.Int32:            field[key] = float64(field2.Interface().(int32))        case reflect.Int64:            field[key] = float64(field2.Interface().(int64))        case reflect.Uint:            field[key] = float64(field2.Interface().(uint))        case reflect.Uint8:            field[key] = float64(field2.Interface().(uint8))        case reflect.Uint16:            field[key] = float64(field2.Interface().(uint16))        case reflect.Uint32:            field[key] = float64(field2.Interface().(uint32))        case reflect.Uint64:            field[key] = float64(field2.Interface().(uint64))        case reflect.Float32:            field[key] = float64(field2.Interface().(float32))        case reflect.Float64:            field[key] = field2.Interface()        default:            field[key] = field2.Interface()        }    }    _, _ = json.Marshal(p1)    // 读取文件中的数据,保存为map格式    // 数据转化为map时,数值类型的统一变成float64    data, _ := ioutil.ReadFile("./index.json")    fields := make([]map[string]interface{}, 0)    err := json.Unmarshal(data, &fields)    if err != nil {        log.Fatal(err)    }    // 修改的条件    columns := []string{"name", "num"}    for _, v := range fields {        if v["_id"] == field["_id"] {            for _, col := range columns {                v[col] = field[col]            }            field = v        }    }    out, _ := json.MarshalIndent(fields, "", "  ")    _ = ioutil.WriteFile("./index.json", out, 0755)}

删除操作

最后就是删除操作了,这个比较思路简单,输入唯一的id值,删除对应的字段,再保存到文件就可以了.代码如下:

func Delete() {    p1 := &Product{        Id:   "db43fa2d4f69cddce7494941cb36032b",        Name: "John_e",        Num:  100,    }    _, _ = json.Marshal(p1)    // 读取文件中的数据,保存为map格式    // 数据转化为map时,数值类型的统一变成float64    data, _ := ioutil.ReadFile("./index.json")    fields := make([]map[string]interface{}, 0)    err := json.Unmarshal(data, &fields)    if err != nil {        log.Fatal(err)    }    length := len(fields)    for index, field := range fields {        if field["_id"] == p1.Id {            if index == length - 1 {                fields = fields[0:index]            } else {                fields = append(fields[0:index], fields[index+1:]...)            }        }    }    out, _ := json.MarshalIndent(fields, "", "  ")    _ = ioutil.WriteFile("./index.json", out, 0755)}

完整版

最后在附上完整版代码:

package storeimport (    "bytes"    "crypto/md5"    "encoding/json"    "errors"    "fmt"    "io/ioutil"    "os"    "path/filepath"    "reflect"    "strings"    "time")type Store struct {    Dir string}func NewStore(dir string) (*Store, error) {    // .开头的为相对路径,补全为全路径    if strings.HasPrefix(dir, ".") {        pwd, _ := os.Getwd()        dir = filepath.Join(pwd, dir)    }    store := &Store{Dir: dir}    st, err := os.Stat(dir)    if err != nil {        err = os.Mkdir(dir, 0755)        if err != nil {            return nil, err        }    } else if st != nil && !st.IsDir() {        return nil, errors.New("file already exists")    }    return store, nil}// 创建与结构体对应的json文件func (s *Store) Sync(values ...interface{}) error {    for _, v := range values {        tb := parseTn(v)        if tb == "" {            return errors.New("does not find store")        }        _path := filepath.Join(s.Dir, tb)        _, err := os.Stat(_path)        if err != nil {            _ = ioutil.WriteFile(_path, []byte("[]"), 0755)        }    }    return nil}// 删除所有func (s *Store) Destroy() error {    return os.RemoveAll(s.Dir)}func (s *Store) FindAll(v interface{}) error {    _path, err := s.before(v)    if err != nil {        return err    }    out, err := s.readAll(_path)    if err != nil {        return err    }    err = json.Unmarshal(out, &v)    return err}func (s *Store) FindOne(v interface{}, columns ...string) (interface{}, error) {    _path, err := s.before(v)    if err != nil {        return nil, err    }    data, err := s.readAll(_path)    if err != nil {        return nil, err    }    fields := make([]map[string]interface{}, 0)    err = json.Unmarshal(data, &fields)    if err != nil {        return nil, err    }    m := structToMap(v)    length := len(columns)    for _, item := range fields {        for i := 0; i < length; i++ {            // TODO 这里的比较需要改进            if item[columns[i]] != m[columns[i]] {                break            }            if i == length-1 {                m = item                goto OVER            }        }    }OVER:    err = mapToStruct(m, &v)    if err != nil {        return nil, err    }    return v, nil}func (s *Store) Create(v interface{}) error {    _path, err := s.before(v)    if err != nil {        return err    }    data, err := s.readAll(_path)    if err != nil {        return err    }    fields := make([]map[string]interface{}, 0)    err = json.Unmarshal(data, &fields)    if err != nil {        return err    }    m := structToMap(v)    m["_id"] = randId()    fields = append(fields, m)    err = s.writeAll(_path, fields)    if err != nil {        return err    }    err = mapToStruct(m, v)    if err != nil {        return err    }    return nil}func (s *Store) Update(v interface{}, columns ...string) error {    _path, err := s.before(v)    if err != nil {        return err    }    data, err := s.readAll(_path)    if err != nil {        return err    }    fields := make([]map[string]interface{}, 0)    err = json.Unmarshal(data, &fields)    if err != nil {        return err    }    m := structToMap(v)    for _, v := range fields {        if v["_id"] == m["_id"] {            for _, col := range columns {                v[col] = m[col]            }            m = v        }    }    err = s.writeAll(_path, fields)    if err != nil {        return err    }    return nil}func (s *Store) Delete(v interface{}) error {    _path, err := s.before(v)    if err != nil {        return err    }    data, err := s.readAll(_path)    if err != nil {        return err    }    fields := make([]map[string]interface{}, 0)    err = json.Unmarshal(data, &fields)    if err != nil {        return err    }    m := structToMap(v)    length := len(fields)    for index, field := range fields {        if field["_id"] == m["_id"] {            if index == length-1 {                fields = fields[0:index]            } else {                fields = append(fields[0:index], fields[index+1:]...)            }        }    }    err = s.writeAll(_path, fields)    if err != nil {        return err    }    return nil}func (s *Store) Clean(v interface{}) error {    _path, err := s.before(v)    if err != nil {        return err    }    return os.Remove(_path)}func (s *Store) readAll(file string) ([]byte, error) {    out, err := ioutil.ReadFile(file)    if err != nil {        return nil, err    }    return out, nil}func (s *Store) writeAll(file string, v interface{}) error {    out, err := json.MarshalIndent(v, "", "  ")    if err != nil {        return err    }    err = ioutil.WriteFile(file, out, 0755)    if err != nil {        return err    }    return nil}func (s *Store) before(v interface{}) (string, error) {    tb := parseTn(v)    if tb == "" {        return "", errors.New("invalid table name")    }    _path := filepath.Join(s.Dir, tb)    _, err := os.Stat(_path)    if err != nil {        return "", err    }    return _path, nil}func structToMap(v interface{}) map[string]interface{} {    tp := reflect.TypeOf(v).Elem()    vp := reflect.ValueOf(v).Elem()    field := make(map[string]interface{}, 0)    for i := 0; i < tp.NumField(); i++ {        field1 := tp.Field(i)        field2 := vp.Field(i)        key := field1.Tag.Get("json")        field[key] = field2.Interface()        switch field2.Kind() {        case reflect.Int:            field[key] = float64(field2.Interface().(int))        case reflect.Int8:            field[key] = float64(field2.Interface().(int8))        case reflect.Int16:            field[key] = float64(field2.Interface().(int16))        case reflect.Int32:            field[key] = float64(field2.Interface().(int32))        case reflect.Int64:            field[key] = float64(field2.Interface().(int64))        case reflect.Uint:            field[key] = float64(field2.Interface().(uint))        case reflect.Uint8:            field[key] = float64(field2.Interface().(uint8))        case reflect.Uint16:            field[key] = float64(field2.Interface().(uint16))        case reflect.Uint32:            field[key] = float64(field2.Interface().(uint32))        case reflect.Uint64:            field[key] = float64(field2.Interface().(uint64))        case reflect.Float32:            field[key] = float64(field2.Interface().(float32))        case reflect.Float64:            field[key] = field2.Interface()        default:            field[key] = field2.Interface()        }    }    return field}func mapToStruct(m map[string]interface{}, v interface{}) error {    out, err := json.Marshal(m)    if err != nil {        return err    }    return json.Unmarshal(out, &v)}func toSnake(s string) string {    out := bytes.Buffer{}    bName := []byte(s)    point := 0    for index, b := range bName {        // 非大写,不需要转化        if b < 65 || b > 90 || index-point < 2 {            out.WriteByte(b)            continue        }        // 首字符大写,直接转化为小写        if index == 0 {            out.WriteByte(b + 32)            point = index        }        // 连续三个大写,触发转化        if index-point >= 2 {            out.WriteByte(95)            out.WriteByte(b + 32)            point = index        }    }    return out.String()}func parseTn(v interface{}) string {    var name string    tp := reflect.TypeOf(v).Elem()    switch tp.Kind() {    case reflect.Ptr:        sp := strings.Split(tp.String(), ".")        name = sp[len(sp)-1]    case reflect.Slice:        sp := strings.Split(tp.String(), ".")        name = sp[len(sp)-1]    case reflect.Struct:        name = tp.Name()    }    name = toSnake(name)    return name + ".json"}func randId() string {    return fmt.Sprintf("%x", md5.Sum([]byte(time.Now().String())))}

文章来源:智云一二三科技

文章标题:golang使用json格式实现增删查改

文章地址:https://www.zhihuclub.com/484.shtml

关于作者: 智云科技

热门文章

发表评论

您的电子邮箱地址不会被公开。

网站地图