goRecommender/readItemSamples.go
readItemSamples.goRaw
1package main
2
3import (
4 "fmt"
5 "io/ioutil"
6 "strings"
7
8 mgo "gopkg.in/mgo.v2"
9)
10
11func readDataset(path string, lineSeparation string, valueSeparation string) [][]string {
12 var dataset [][]string
13 b, err := ioutil.ReadFile(path)
14 check(err)
15 str := string(b)
16 str = strings.Replace(str, "\r", "", -1)
17 lines := strings.Split(str, lineSeparation)
18 for _, v1 := range lines {
19 params := strings.Split(v1, valueSeparation)
20 var datasetLine []string
21 for _, v2 := range params {
22 datasetLine = append(datasetLine, v2)
23 }
24 dataset = append(dataset, datasetLine)
25 }
26 return dataset
27}
28func getItemsFromDataset(dataset [][]string) []ItemModel {
29 var items []ItemModel
30 for _, v := range dataset {
31 var newItem ItemModel
32 newItem.ID = v[0]
33 items = append(items, newItem)
34 }
35 return items
36}
37
38func datasetToMongodbIfNotExist(c *mgo.Collection, items []ItemModel) {
39 fmt.Println(items)
40 for _, item := range items {
41 saveItem(c, item)
42 }
43}