manager.go

package olapsql

import (
	"fmt"

	"github.com/awatercolorpen/olap-sql/api/types"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

type Manager struct {
	clients    Clients
	dictionary *Dictionary
}

func (m *Manager) GetClients() (Clients, error) {
	if m.clients == nil {
		return nil, fmt.Errorf("it is no initialization")
	}
	return m.clients, nil
}

func (m *Manager) GetDictionary() (*Dictionary, error) {
	if m.dictionary == nil {
		return nil, fmt.Errorf("it is no initialization")
	}
	return m.dictionary, nil
}

func (m *Manager) SetLogger(log logger.Interface) {
	c, err := m.GetClients()
	if err == nil {
		c.SetLogger(log)
	}
}

func (m *Manager) RunSync(query *types.Query) (*types.Result, error) {
	db, err := m.BuildTransaction(query)
	if err != nil {
		return nil, err
	}
	rows, err := RunSync(db)
	if err != nil {
		return nil, err
	}
	return BuildResultSync(query, rows)
}

func (m *Manager) RunChan(query *types.Query) (*types.Result, error) {
	db, err := m.BuildTransaction(query)
	if err != nil {
		return nil, err
	}
	rows, err := RunChan(db)
	if err != nil {
		return nil, err
	}
	return BuildResultChan(query, rows)
}

func (m *Manager) BuildTransaction(query *types.Query) (*gorm.DB, error) {
	clients, clause, err := m.build(query)
	if err != nil {
		return nil, err
	}
	return clients.BuildDB(clause)
}

func (m *Manager) BuildSQL(query *types.Query) (string, error) {
	clients, clause, err := m.build(query)
	if err != nil {
		return "", err
	}
	return clients.BuildSQL(clause)
}

func (m *Manager) build(query *types.Query) (Clients, types.Clause, error) {
	query.TranslateTimeIntervalToFilter()
	dict, err := m.GetDictionary()
	if err != nil {
		return nil, nil, err
	}
	clause, err := dict.Translate(query)
	if err != nil {
		return nil, nil, err
	}
	clients, err := m.GetClients()
	if err != nil {
		return nil, nil, err
	}
	return clients, clause, nil
}

func NewManager(configuration *Configuration) (*Manager, error) {
	m := &Manager{}
	if configuration.ClientsOption != nil {
		clients, err := NewClients(configuration.ClientsOption)
		if err != nil {
			return nil, err
		}
		m.clients = clients
	}
	if configuration.DictionaryOption != nil {
		dict, err := NewDictionary(configuration.DictionaryOption)
		if err != nil {
			return nil, err
		}
		m.dictionary = dict
	}
	return m, nil
}