基于go的反射实现查询生成器

SmallForest · · 700 次点击 · · 开始浏览

这是一个创建于的文章，其中的信息可能已经有所发展或是发生改变。

基于go的反射实现查询生成器

参考go系列教材反射章节，源代码没有实现输出字段，我根据对反射的理解实现了对字段进行输出。beego的orm就是基于反射实现的。

/*
# @Time : 2019-01-05 16:57
# @Author : smallForest
# @SoftWare : GoLand
*/
package main

import (
    "fmt"
    "reflect"
)

type order3 struct {
    ordId      int
    customerId int
}
type employee3 struct {
    name    string
    id      int
    address string
    salary  int
    country string
}

func createQuery3(q interface{}) {
    if reflect.ValueOf(q).Kind() == reflect.Struct {
        t := reflect.TypeOf(q).Name()

        query := fmt.Sprintf("insert into %s(", t)
        Names := reflect.TypeOf(q)
        for j := 0; j < Names.NumField(); j++ {
            if j==0{
                query = fmt.Sprintf("%s%s",query,Names.Field(j).Name)
            }else{
                query = fmt.Sprintf("%s,%s",query,Names.Field(j).Name)
            }
        }
        query = fmt.Sprintf("%s) values(",query)

        v := reflect.ValueOf(q)
        for i := 0; i < v.NumField(); i++ {
            switch v.Field(i).Kind() {
            case reflect.Int:
                if i == 0 {
                    query = fmt.Sprintf("%s%d", query, v.Field(i).Int())
                } else {
                    query = fmt.Sprintf("%s,%d", query, v.Field(i).Int())
                }
            case reflect.String:
                if i == 0 {
                    query = fmt.Sprintf("%s\"%s\"", query, v.Field(i).String())

                } else {
                    query = fmt.Sprintf("%s,\"%s\"", query, v.Field(i).String())
                }
            default:
                fmt.Println("Unsupported type")
                return

            }
        }
        query = fmt.Sprintf("%s)", query)
        fmt.Println(query)
        return
    }
    fmt.Println("unsupported type")
}

func main() {
    o := order3{456, 33}
    createQuery3(o)
    e3 := employee3{
        "Naveen",
        555,
        "Coimbatore",
        9000,
        "India",
    }
    createQuery3(e3)
    i := 90
    createQuery3(i)
}

有疑问加站长微信联系（非本文作者）