Как сохранить пользовательские данные, записанные в CSV-файл в Mongodb в любом формате? - PullRequest
0 голосов
/ 28 июня 2018

Предположим, есть CSV-файл, отформатированный как показано ниже: -

первый файл

firstname|lastname|Email| other fields
         |        |     |
         |        |     |
         |        |     |
         |        |     |

Второй файл: -

email|firstname|lastname|other fields
     |         |        |
     |         |        |
     |         |        |
     |         |        |

третий файл

lastname|firstname|email|other fields     
        |         |     |
        |         |     |
        |         |     |
        |         |     |

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

Формат: first_name, last_name,email, otherfield

Код, который я использую: -

  package main

import (
 "encoding/csv"
 "gopkg.in/mgo.v2"
 "io"
 "log"
 "os"
)

type Mongo struct {
  // Id          int    `json:"_id" bson:"_id"`
  FirstName   string `json:"first_name,omitempty" bson:"first_name,omitempty"`
  LastName    string `json:"last_name,omitempty" bson:"last_name,omitempty"`
  Email       string `json:"email,omitempty" bson:"email,omitempty"`
  PhoneNumber string `json:"phone_number,omitempty" bson:"phone_number,omitempty"`
  Gender      string `json:"gender,omitempty" bson:"gender,omitempty"`
  Address     string `json:"address,omitempty" bson:"address,omitempty"`
  Apartment   string `json:"apartment,omitempty" bson:"apartment,omitempty"`
  Description string `json:"description,omitempty" bson:"description,omitempty"`
 }

func main() {

 session, err := mgo.Dial("localhost")
 if err != nil {
    panic(err)
 }

 defer session.Close()
 session.SetMode(mgo.Monotonic, true)

 c := session.DB("Import_Users").C("users")

 file, err := os.Open("customers.csv")

 if err != nil {
    panic(err)
 }
 defer file.Close()

 reader := csv.NewReader(file)

 for {
    record, err := reader.Read()
    if err == io.EOF {
        break
    } else if err != nil {
        panic(err)
    }

    err = c.Insert(&Mongo{record[0], record[1], record[2], record[3], record[4], record[5], record[6], record[7]})

    if err != nil {
        panic(err)
    }
    log.Printf("%#v", record)
 }
}

Код, который я использую, прост для сохранения данных в mongodb, но он не отформатирует данные, которые мне нужны для сохранения

Как мне достичь своей цели?

1 Ответ

0 голосов
/ 28 июня 2018

Я думаю, что вы можете использовать отражение, чтобы отобразить каждую строку CSV над структурой. Нечто подобное пакет кодирования json делает из строки в структуру. Затем эту структуру можно вставить в mongodb.

Что-то вроде этого вы хотите? https://play.golang.org/p/hg3K-u8l9mx

Вам нужно больше защиты на нем, и вы можете сделать много улучшений скорости, но это что-то вроде Unmarshal от json pkg.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...