package executor

import "github.com/everything-on-mysql/cassandra-on-mysql/pkg/cql/parser"

func statementKeyspace(stmt parser.Statement) string {
	switch s := stmt.(type) {
	case *parser.InsertStatement:
		if s.Table != nil {
			return s.Table.Schema
		}
	case *parser.UpdateStatement:
		if s.Table != nil {
			return s.Table.Schema
		}
	case *parser.DeleteStatement:
		if s.From != nil {
			return s.From.Schema
		}
	case *parser.SelectStatement:
		if s.From != nil {
			return s.From.Schema
		}
	}
	return ""
}