package main

import (
	"context"
	"fmt"
	"time"

	"github.com/gocql/gocql"
)

type MyRetryPolicy struct {
}

func (*MyRetryPolicy) Attempt(q gocql.RetryableQuery) bool {
	if q.Attempts() > 2 {
		return false
	}
	return true
}

func (*MyRetryPolicy) GetRetryType(error) gocql.RetryType {
	return gocql.Ignore
}

type LoggingObserver struct{}

func (*LoggingObserver) ObserveQuery(ctx context.Context, q gocql.ObservedQuery) {
	fmt.Printf("observer attempt: %d err: %s\n", q.Attempt, q.Err)
}

func main() {
	cluster := gocql.NewCluster("node-1")
	cluster.Authenticator = gocql.PasswordAuthenticator{
		Username: "scylla",
		Password: "***",
	}

	s, err := gocql.NewSession(*cluster)
	if err != nil {
		fmt.Printf("session err: %s\n", err)
	}
	defer s.Close()

	q := s.Query("INSERT INTO  my.events(event_id, time, args) VALUES (?,?,?)", 1, gocql.UUIDFromTime(time.Now()), "test")

	q.RetryPolicy(&MyRetryPolicy{})
	q.Observer(&LoggingObserver{})

	fmt.Printf("QUERY IsIdempotent: %v\n", q.IsIdempotent())

	err = q.Exec()
	if err != nil {
		fmt.Printf("exec err: %s\n", err)
	}

}

➜  test go run main.go
QUERY IsIdempotent: false
observer attempt: 0 err: Keyspace my does not exist
exec err: Keyspace my does not exist