Create public interface by exporting internal types.

Author: clems4ever

cmd/importer-csv/main.go

@@ -1,15 +1,83 @@
 package main
 
 import (
+	"encoding/csv"
 	"fmt"
+	"io"
 	"log"
+	"os"
 
-	"github.com/clems4ever/go-graphkb/importer"
-	"github.com/clems4ever/go-graphkb/internal/sources"
+	"github.com/clems4ever/go-graphkb/graphkb"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
 )
 
+type CSVSource struct {
+	dataPath string
+}
+
+func NewCSVSource() *CSVSource {
+	csvSource := new(CSVSource)
+	csvSource.dataPath = viper.GetString("path")
+
+	if csvSource.dataPath == "" {
+		log.Fatal(fmt.Errorf("Unable to detect CSV file path in configuration. Check patch configuration is provided"))
+	}
+
+	return csvSource
+}
+
+func (cs *CSVSource) Start(importer *graphkb.GraphImporter) error {
+	file, err := os.Open(cs.dataPath)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+
+	r := csv.NewReader(file)
+
+	previousGraph, err := importer.ReadCurrentGraph()
+	if err != nil {
+		return fmt.Errorf("Unable to read previous graph: %v", err)
+	}
+
+	tx := importer.CreateTransaction(previousGraph)
+
+	header := true
+
+	for {
+		record, err := r.Read()
+		if err == io.EOF {
+			break
+		}
+		if err != nil {
+			return fmt.Errorf("Unable to read data in CSV file: %v", err)
+		}
+
+		// Skip header line
+		if header {
+			header = false
+			continue
+		}
+
+		relationType := graphkb.RelationType{
+			FromType: graphkb.AssetType(record[0]),
+			ToType:   graphkb.AssetType(record[3]),
+			Type:     graphkb.RelationKeyType(record[2]),
+		}
+
+		tx.Relate(record[1], relationType, record[4])
+	}
+
+	_, err = tx.Commit()
+	fmt.Println("CSV data has been sent successfully")
+	return err
+}
+
+func (cs *CSVSource) Stop() error {
+	return nil
+}
+
 // ConfigPath string
 var ConfigPath string
 
@@ -30,7 +98,7 @@ func main() {
 	rootCmd := &cobra.Command{
 		Use: "source-csv [opts]",
 		Run: func(cmd *cobra.Command, args []string) {
-			if err := importer.Start(sources.NewCSVSource(), nil); err != nil {
+			if err := graphkb.Start(NewCSVSource(), nil); err != nil {
 				log.Fatal(err)
 			}
 		},

graphkb/graph.go

@@ -0,0 +1,5 @@
+package graphkb
+
+import "github.com/clems4ever/go-graphkb/internal/knowledge"
+
+type Graph = knowledge.Graph

graphkb/graph_api.go

@@ -0,0 +1,5 @@
+package graphkb
+
+import "github.com/clems4ever/go-graphkb/internal/knowledge"
+
+type GraphAPI = knowledge.GraphAPI

graphkb/graph_importer.go

@@ -0,0 +1,5 @@
+package graphkb
+
+import "github.com/clems4ever/go-graphkb/internal/knowledge"
+
+type GraphImporter = knowledge.GraphImporter

importer/importer.go renamed to graphkb/importer.go

@@ -1,4 +1,4 @@
-package importer
+package graphkb
 
 import (
 	"fmt"

graphkb/schema.go

@@ -0,0 +1,7 @@
+package graphkb
+
+import "github.com/clems4ever/go-graphkb/internal/schema"
+
+type AssetType = schema.AssetType
+type RelationKeyType = schema.RelationKeyType
+type RelationType = schema.RelationType

graphkb/tasks.go

@@ -0,0 +1,5 @@
+package graphkb
+
+import "github.com/clems4ever/go-graphkb/internal/utils"
+
+type RecurrentTask = utils.RecurrentTask