main.go

package main

import (
	"fmt"
	_ "github.com/dimiro1/banner/autoload"
	"github.com/gocql/gocql"
	"github.com/kataras/iris"
	"gopkg.in/yaml.v2"
	"io/ioutil"
	"log"
)

type Config struct {
	Scyllaclusters []string
	Serverport     int
}

var session *gocql.Session
var config Config

func main() {
	config = Config{}
	configFile, err := ioutil.ReadFile("application.yml")
	if err != nil {
		panic(err)
	}

	err = yaml.Unmarshal(configFile, &config)
	if err != nil {
		panic(err)
	}

	cluster := gocql.NewCluster(config.Scyllaclusters...)
	cluster.Keyspace = "orders"
	cluster.ProtoVersion = 3
	cluster.Consistency = gocql.Two

	session, _ = cluster.CreateSession()
	defer session.Close()

	setupWebServer(session)
}

func setupWebServer(session *gocql.Session) {
	iris.Get("/hello", hello)
	iris.Get("/scylla", func(ctx *iris.Context) {
		scylla(ctx, session)
	})

	iris.API("/orders", OrderAPI{})
	iris.API("/orders/:id", OrderAPI{})
	iris.API("/orders/:id/items", OrderItemAPI{})
	iris.API("/orders/:id/transactions", TransactionAPI{})

	iris.Listen(fmt.Sprintf(":%v", config.Serverport))
}

func scylla(ctx *iris.Context, session *gocql.Session) {
	if err := session.Query("INSERT INTO \"order\" (id, number) VALUES (?, ?)",
		gocql.TimeUUID(), "10").Exec(); err != nil {
		log.Fatal(err)
	}

	ctx.Write("OK")
}

func hello(ctx *iris.Context) {
	ctx.Write("World!\n")
}