Beispiel #1
0
// NewService loads configs and starts listeners
func NewService(config *Config) Service {
	// setup echo
	echo := echo.New()
	echopprof.Wrapper(echo)

	logger := log.New(os.Stderr, "EVENT CONSUMER:", log.Ldate|log.Ltime|log.Lshortfile)

	DBAdaptor := database.NewAdaptor(strings.Split(config.CassandraInterfaces, ","), config.CassandraUser, config.CassandraPassword)

	CacheAdaptor := cache.NewAdaptor(config.CacheURL, config.CacheRedisPassword)

	service := Service{
		Logger: logger,
		config: config,
		Router: Routes(
			// add each dependent service as a dependency to the router
			dependencies{
				config:       config,
				DBAdaptor:    DBAdaptor,
				CacheAdaptor: CacheAdaptor,
				Router:       echo,
			},
		),
		DBAdaptor: DBAdaptor,
	}

	return service
}
Beispiel #2
0
func NewService(config *Config) Service {
	logger := log.New(os.Stderr, "EVENT CONSUMER:", log.Ldate|log.Ltime|log.Lshortfile)

	DBAdaptor := database.NewAdaptor(strings.Split(config.CassandraInterfaces, ","), config.CassandraUser, config.CassandraPassword)

	dispatcher := NewDispatcher(config.MaxWorker, config.MaxJobQueue, logger, DBAdaptor)
	return Service{
		Logger:     logger,
		config:     config,
		Dispatcher: dispatcher,
		DBAdaptor:  DBAdaptor,
	}
}
func setUp(cmd *cobra.Command, args []string) {
	fmt.Println("[+] Running Cassandra set up ")

	var config consumerapi.Config

	if err := envconfig.Process("CONSUMER", &config); err != nil {
		log.Fatal("Error occurred during startup:", err.Error())
	}

	cluster := gocql.NewCluster(strings.Split(config.CassandraInterfaces, ",")...)
	cluster.ProtoVersion = 3

	session, err := cluster.CreateSession()
	if err != nil {
		log.Print(err.Error())

		return
	}

	q := session.Query("CREATE KEYSPACE IF NOT EXISTS events WITH REPLICATION = { 'class' : 'SimpleStrategy', 'replication_factor' : 1};", "")
	if err := q.Exec(); err != nil {
		log.Print(err.Error())
	}

	DBAdaptor := database.NewAdaptor(strings.Split(config.CassandraInterfaces, ","), config.CassandraUser, config.CassandraPassword)

	for _, indexField := range database.IndexFields {
		eventsTable := DBAdaptor.GetEventMultimapTable(indexField)

		if err := eventsTable.Create(); err != nil {
			log.Print(err.Error())
		}
	}

	eventTable := DBAdaptor.GetTimeSeriesEventTable()
	if err := eventTable.Create(); err != nil {
		log.Print(err.Error())
	}

	eventMapTable := DBAdaptor.GetEventMapTable()
	if err := eventMapTable.Create(); err != nil {
		log.Print(err.Error())
	}
}