foodoles/bdd/bdd.go

package bdd

import (
	"bytes"
	"fmt"
	"log"
	"time"

	"github.com/boltdb/bolt"
)

// OpenDB a key-value store.
func OpenDB() (*bolt.DB, error) {
	db, err := bolt.Open("foods.db", 0600, nil)
	if err != nil {
		return nil, fmt.Errorf("could not open db, %v", err)
	}
	err = db.Update(func(tx *bolt.Tx) error {
		root, err := tx.CreateBucketIfNotExists([]byte("DB"))
		if err != nil {
			return fmt.Errorf("could not create root bucket: %v", err)
		}
		_, err = root.CreateBucketIfNotExists([]byte("VOTES"))
		if err != nil {
			return fmt.Errorf("could not create votes bucket: %v", err)
		}
		return nil
	})
	if err != nil {
		return nil, fmt.Errorf("could not set up buckets, %v", err)
	}
	return db, nil

}

// CloseDB closes the key-value store file.
func CloseDB(db *bolt.DB) error {
	return db.Close()
}

// InitDB init the db
func InitDB() error {
	db, err := OpenDB()
	if err != nil {
		log.Printf("\nOpenDB error: %v", err)
		return err
	}
	defer func() {
		if err1 := CloseDB(db); err1 != nil {
			err = err1
		}
	}()
	return nil
}

// AddVote ajoute un vote à la bdd
func AddVote(db *bolt.DB, vote string, date time.Time) error {

	err := db.Update(func(tx *bolt.Tx) error {
		err := tx.Bucket([]byte("DB")).Bucket([]byte("VOTES")).Put([]byte(date.Format(time.RFC3339)), []byte(vote))
		if err != nil {
			return fmt.Errorf("could not insert vote: %v", err)
		}
		return nil
	})
	fmt.Println("Added vote")
	return err
}

// GetAllVotes liste tous les votes
func GetAllVotes(db *bolt.DB) ([]string, error) {

	res := []string{}
	err := db.View(func(tx *bolt.Tx) error {
		b := tx.Bucket([]byte("DB")).Bucket([]byte("VOTES"))
		b.ForEach(func(k, v []byte) error {
			res = append(res, string(v))
			return nil
		})
		return nil
	})
	return res, err
}

// GetVotesOfTheDay liste tous les votes du jour
func GetVotesOfTheDay(db *bolt.DB) ([]string, error) {
	res := []string{}

	err := db.View(func(tx *bolt.Tx) error {
		b := tx.Bucket([]byte("DB")).Bucket([]byte("VOTES"))
		b.ForEach(func(k, v []byte) error {
			hop, _ := time.Parse(time.RFC3339, string(k))
			if DateEqual(hop, time.Now()) {
				res = append(res, string(v))
				return nil
			}
			return nil
		})
		return nil
	})
	//fmt.Print(res)
	return res, err
}

// GetVotesOfTheRange liste tous les votes du jour
func GetVotesOfTheRange(db *bolt.DB, min time.Time, max time.Time) ([]string, error) {
	res := []string{}
	err := db.View(func(tx *bolt.Tx) error {
		c := tx.Bucket([]byte("DB")).Bucket([]byte("VOTES")).Cursor()
		mino := []byte(min.Format(time.RFC3339))
		maxo := []byte(max.Format(time.RFC3339))
		for k, v := c.Seek(mino); k != nil && bytes.Compare(k, maxo) <= 0; k, v = c.Next() {
			res = append(res, string(v))
		}
		return nil
	})
	return res, err
}

// DateEqual check if dates are equal
func DateEqual(date1, date2 time.Time) bool {
	y1, m1, d1 := date1.Date()
	y2, m2, d2 := date2.Date()

	return y1 == y2 && m1 == m2 && d1 == d2
}