Как выполнить итерацию фрагмента в структуре в Golang или можно использовать для l oop в структуре? - PullRequest
Новая
       7

Как выполнить итерацию фрагмента в структуре в Golang или можно использовать для l oop в структуре?

1 голос
/ 16 июня 2020

У меня две структуры. Один предназначен для лотереи, а другой - для вознаграждения. Я использую базу данных Mysql. Я хочу читать данные из базы данных и писать в формате JSON. Я могу это сделать, но я хочу иметь вложенную структуру, в которой я хочу повторять структуру вознаграждения в структуре лотереи. Могу ли я это сделать?

Вот моя структура лотереи и вознаграждения 

type Lottery struct{
    Id int `json:"lottery_id"`
    Lottery string `json:"lottery_name"`
    Description string `json:"lottery_description"`
    Reward []Rew `json:"rewards"`
}

type Rew struct{
    Id int `json:"reward_id"`
    RewardName string `json:"reward_name"`
    Description string `json:"reward_description"`
    Asset int `json:"reward_asset"`
    AssetName string `json:"reward_asset_name"`
}

Вот мой код 

app.Get("/lottery/{id:int}", func (ctx iris.Context){
        id1 := ctx.Params().GetIntDefault("id",0)

        stmtOut, err := db.Prepare("select lottery_name, lottery_description from lottery_table where id = ?")
        if err !=nil{
            panic(err.Error())
        }
        defer stmtOut.Close()

        var lottery_name, lottery_description string

        err1 := stmtOut.QueryRow(id1).Scan(&lottery_name,&lottery_description)
        if err != nil{
            panic(err1.Error())
        }

        stmtOut1, err := db.Query("select id, reward_name, reward_description, reward_asset, reward_asset_name from rewards_table where lottery_id = ?",id1)
        if err != nil{
            panic(err.Error())
        }
        defer stmtOut1.Close()

        for stmtOut1.Next() {

            var id, reward_asset int
            var reward_name, reward_description, reward_asset_name string

            err2 := stmtOut1.Scan(&id, &reward_name, &reward_description, &reward_asset, &reward_asset_name)
            if err2 != nil {
                panic(err.Error())
            }

                rew := Lottery{
                Id:          id1,
                Lottery:     lottery_name,
                Description: lottery_description,
                Reward : []Rew{Rew{
                    Id:          id,
                    RewardName:  reward_name,
                    Description: reward_description,
                    Asset:       reward_asset,
                    AssetName:   reward_asset_name,
                },
                },
            }

            ctx.JSON(&rew)
        }


    })

Когда я использую вышеуказанную функцию, я получаю JSON формат, подобный этому 

{
    "lottery_id": 7,
    "lottery_name": "lottery2",
    "lottery_description": "lottery for 7",
    "rewards": [
        {
            "reward_id": 9,
            "reward_name": "Reward3",
            "reward_description": "Reward for lottery 7",
            "reward_asset": 20,
            "reward_asset_name": "AC"
        }
    ]
}{
    "lottery_id": 7,
    "lottery_name": "lottery2",
    "lottery_description": "lottery for 7",
    "rewards": [
        {
            "reward_id": 10,
            "reward_name": "Reward5",
            "reward_description": "Reward for lottery 7",
            "reward_asset": 15,
            "reward_asset_name": "AC"
        }
    ]
}

Но я хочу получить JSON как 

{
"lottery_id":7,
"lottery_name":"lottery2"
"lottery_description":"lottery for 7"
"rewards":[
    {
        "reward_id":9,
        "reward_name": "Reward3",
        "reward_description": "Reward for lottery 7",
        "reward_asset": 20,
        "reward_asset_name": "ABC"
    },
    {   "reward_id":10,
        "reward_name": "Reward5",
        "reward_description": "Reward for lottery 7",
        "reward_asset": 15,
        "reward_asset_name": "ABC"
    },
    {   "reward_id":11,
        "reward_name": "Reward7",
        "reward_description": "Reward for lottery 7",
        "reward_asset": 10,
        "reward_asset_name": "ABC"
    }
    ]
}

Как я могу выполнить итерацию в структуре? Возможно ли иметь l oop в структуре? или есть ли другой способ получить JSON, как указано выше?

Пожалуйста, помогите мне.

1 Ответ

3 голосов
/ 16 июня 2020

Примерно так: 

app.Get("/lottery/{id:int}", func(ctx iris.Context) {
    id1 := ctx.Params().GetIntDefault("id", 0)

    stmtOut, err := db.Prepare("select lottery_name, lottery_description from lottery_table where id = ?")
    if err != nil {
        panic(err.Error())
    }
    defer stmtOut.Close()

    lot := Lottery{Id: id1}
    err1 := stmtOut.QueryRow(id1).Scan(&lot.Lottery, &lot.Description)
    if err != nil {
        panic(err1.Error())
    }

    stmtOut1, err := db.Query("select id, reward_name, reward_description, reward_asset, reward_asset_name from rewards_table where lottery_id = ?", id1)
    if err != nil {
        panic(err.Error())
    }
    defer stmtOut1.Close()

    for stmtOut1.Next() {
        rew := Rew{}
        err2 := stmtOut1.Scan(&rew.Id, &rew.RewardName, &rew.Description, &rew.Asset, &rew.AssetName)
        if err2 != nil {
            panic(err.Error())
        }
        lot.Reward = append(lot.Reward, rew)

    }
    ctx.JSON(lot)
})
...