Как преобразовать refle.Pointer () в строку [] при использовании pg.Array? - PullRequest
Новая
       10

Как преобразовать refle.Pointer () в строку [] при использовании pg.Array?

0 голосов
/ 02 мая 2019

Я использую go-pg для написания собственной системы кэширования запросов, которая принимает аргументы запроса, которые передаются в функцию Query, и генерирует ключ хеша, который используется для Redis. Я использую Go для отражения, чтобы проверить типы аргументов, которые работают, пока я не использую pg.Array в качестве переданного аргумента.

Reflect дает мне refle.Ptr, но как мне извлечь структуру / массив указателя, когда вызывается блок регистра переключателя? 

func GenerateQueryCacheKey(args ...interface{}) string {
    var argumentString = ""

    for _, arg := range args {

        v := reflect.ValueOf(arg)
        switch v.Kind() {
        case reflect.Array, reflect.Slice:
            ret := make([]interface{}, v.Len())

            for i := 0; i < v.Len(); i++ {
                ret[i] = v.Index(i).Interface()
            }

            GenerateQueryCacheKey(ret...)
        case reflect.Bool:
            argumentString += strconv.FormatBool(v.Bool())
        case reflect.String:
            argumentString += v.String()
        case reflect.Int:
            argumentString += string(v.Int())
        case reflect.Uint:
            argumentString += string(v.Uint())
        case reflect.Float32:
            argumentString += strconv.FormatFloat(v.Float(), 'E', -1, 32)
        case reflect.Invalid:
            log.Printf("Invalid type handle! " + fmt.Sprintf("%T", arg))
            argumentString += "nil"
        case reflect.Ptr:
            p := v.Elem()

            ret := make([]interface{}, p.Len())

            for i := 0; i < p.Len(); i++ {
                ret[i] = p.Index(i).Interface()
            }

            GenerateQueryCacheKey(ret...)
        default:
            log.Printf("Unhandled reflect type supplied! " + fmt.Sprintf("%T %T", arg, v))
            argumentString += "nil"
        }
    }

    h := md5.New()
    io.WriteString(h, argumentString)

    return fmt.Sprintf("%x", h.Sum(nil))
}

стр. Определение массива: https://sourcegraph.com/github.com/lib/pq/-/blob/array.go#L29:6

РЕДАКТИРОВАТЬ: размещенная ссылка имеет неправильное определение pg.Array. Я случайно забрал не ту библиотеку с sourcegraph.com.

1 Ответ

0 голосов
/ 03 мая 2019

Решено путем отвлечения от отражения и использования более прямого подхода.

Я просто импортировал тип. Массив из pg, создал фрагмент интерфейса {} и вручную добавил строки в интерфейс {}нарезать и использовать оператор распространения, чтобы создать рекурсивную функцию в настоящее время.Не уверен, что это лучший метод, но сейчас он работает для меня. 

func GenerateQueryCacheKey(args ...interface{}) string {
    var argumentString = ""

    for _, arg := range args {

        switch v := arg.(type) {
        case bool:
            argumentString += strconv.FormatBool(v)
        case string:
            argumentString += v
        case int:
            argumentString += strconv.Itoa(v)
        case uint64:
            argumentString += string(v)
        case float64:
            argumentString += strconv.FormatFloat(v, 'E', -1, 32)
        case *types.Array:
            stringArrayArgs := make([]interface{}, len(v.Value().([]string)))
            for i, vs := range v.Value().([]string) {
                stringArrayArgs[i] = vs
            }

            argumentString += GenerateQueryCacheKey(stringArrayArgs...)
        case nil:
        default:
            log.Printf("%T was requested and not handled!\n", v)
            argumentString += "nil"
        }
    }

    h := md5.New()
    io.WriteString(h, argumentString)

    return fmt.Sprintf("%x", h.Sum(nil))
}
...