arnaucube
arnaucube/goRecommenderPublic Self-hosted repository
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}