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

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

Reflect дает мне Reflect.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 ответ

Решение

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

Я просто импортировал типы. Массив из 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))
}
Другие вопросы по тегам