Как динамически создавать фильтры запросов - PullRequest
Новая
       45

Как динамически создавать фильтры запросов

1 голос
/ 19 января 2020

Я использую Go для настройки собственного API. Я как бы застрял прямо сейчас из-за того, как я написал код для динамического создания / применения фильтра запросов. Это работает, но мне интересно, есть ли лучший способ сделать сценарий ниже.

Например, у меня есть страница поиска с флажками (1 для электронной почты и 1 для имени), чтобы сузить поиск. 

// If I checked the email, the query would be like this
query findOne() {
  user(func: type(user)) @filter(eq(email, "john.doe@email.com")) {
    name
    email
    age
    home_address
  }
}

// If name checkedbox is also checked, it would be like this
query findOne() {
  user(func: type(user)) @filter(eq(email, "john") OR eq(name, "john")) {
    name
    email
    age
    home_address
  }
}

Это то, что я получил до сих пор, и я думаю, что есть лучший способ сделать это: 

func (s *Service) GetUser(email, name string) (*Users, error) {
    c := db.NewClient()
    defer db.Close()
    var u Users
    var filter string

    if email != "" && mobileNumber != "" {
      filter = fmt.Sprintf(`eq(email, "%s") OR eq(mobileNumber, "%s")`, email, mobileNumber)
    } else if email != "" && mobileNumber == "" {
      filter = fmt.Sprintf(`eq(email, "%s")`, email)
    } else if email == "" && mobileNumber != "" {
      filter = fmt.Sprintf(`eq(mobileNumber, "%s")`, mobileNumber)
    }

    q := fmt.Sprintf(`query findOne() {
        users(func: type("user")) @filter(%s) {
            name
            email
            home_address
            contact_number
        }
    }`, filter)
    ctx := context.Background()
    res, err := c.NewTxn().Query(ctx, q)
    if err != nil {
        return nil, err
    }
    if err = json.Unmarshal(res.Json, &u); err != nil {
        return nil, err
    }
    return &u, nil
}

Есть ли лучший способ сделать это вместо создания длинного условия

Ответы [ 2 ]

2 голосов
/ 20 января 2020

Вот его вариант отражения. По сути, он перечисляет поля, получает значения и имена полей для построения массива строк на их основе. Пожалуйста, обратите внимание, что я не очень опытный, это может также потребовать некоторых улучшений. 

import (
    "fmt"
    "reflect"
    "strings"
)

type User struct {
    Id       int    
    FullName string 
    Phone    string 
    Mail     string 
}

func main() {
    u := &User{Id: 10, FullName: "John", Mail: "john@mail"}
    u2 := struct {
        id   int
        name string
    }{10, "john"};
   // inline struct
    q := getQuery(&u2, "OR")
    fmt.Println(q)

   // typed struct
    q = getQuery(u, "AND")
    fmt.Println(q)
}

func getQuery(target interface{}, join string) string {
    var filters []string
    val := reflect.ValueOf(target).Elem()

    for i := 0; i < val.NumField(); i++ {
        value := val.Field(i)
        s :=fmt.Sprintf("%v",value);
        // this little trick is to check if it is an empty value
        // so don't generate empty condition expressions
        if s == "" {
             continue
        }
        fieldType := val.Type().Field(i)
        filters = append(filters, fmt.Sprintf(" eq(%s, %v) ", fieldType.Name, value))
    }

    return strings.Join(filters, join)
}

Здесь находится детская площадка

2 голосов
/ 19 января 2020

Я бы посоветовал провести рефакторинг логики фильтра c, как указано ниже: 

package main

import (
    "fmt"
    "strings"
)

func getQuery(key, val string, filters *[]string) {
    if val != "" {
        *filters = append(*filters, fmt.Sprintf(`eq("%s", "%s")`, key, val))
    }
}
func main() {
    var filters []string
    email := "demo@demo.com"
    mobileNumber := "123456789"

    getQuery("email", email, &filters)
    getQuery("mobileNumber", mobileNumber, &filters)
    filter := strings.Join(filters, " OR ")
    fmt.Println(filter)
}
...