seaweed/sea.go

/***
Пакет seaweed предназначен для облегчения работы с БД sea
и weed.
Текущая версия: v1.0.7
***/

package seaweed

import (
	"errors"
	"fmt"
	"os"

	_ "github.com/go-sql-driver/mysql"
	"github.com/jmoiron/sqlx"

	// "github.com/kataras/pio"
	"github.com/kataras/golog"
)

const VERSION = "v1.0.7"

// ============= SEA ============

type Sea struct {
	db    *sqlx.DB
	debug bool
}

// Окрывает БД sea, соединяется с weed.
// Если options = true, то будут выводиться отладочные сообщения
func Connect(options ...bool) (*Sea, error) {
	debug := false
	if len(options) > 0 && options[0] {
		debug = true
	}

	user := os.Getenv("SEA_USER")
	if user == "" {
		return nil, errors.New("не установлена переменная SEA_USER")
	}

	password := os.Getenv("SEA_PASSWORD")
	if password == "" {
		return nil, errors.New("не установлена переменная SEA_PASSWORD")
	}

	db, err := openDB(user, password)
	if err != nil {
		return nil, err
	}

	sea := Sea{db: db, debug: debug}

	if debug {
		golog.Infof("БД \"sea\" успешно открыта. %s", VERSION)
	}

	return &sea, nil
}

func openDB(user, password string) (*sqlx.DB, error) {
	conn := fmt.Sprintf("%s:%s@(xigmanas:3306)/sea", user, password)
	db, err := sqlx.Open("mysql", conn)
	if err != nil {
		return nil, err
	}
	return db, nil
}

// ============= ALBUM ============

// Album содержит информацию об активном альбоме
type Album struct {
	Seria string
	Cnt   int
	Hash  string
}

// Albums возвращает список активных альбомов,
// без учета комиксов.
func (s *Sea) Albums() ([]Album, error) {
	cmd := "call getAlbums()"
	var albums []Album
	err := s.db.Select(&albums, cmd)
	if err != nil {
		return nil, err
	}
	if s.debug {
		golog.Infof("Получен список альбомов: %d", len(albums))
	}

	return albums, nil
}