set ldap auth

bdd/bdd.go

@@ -3,6 +3,7 @@ package bdd
 import (
 	"bytes"
 	"fmt"
+	"log"
 	"time"
 
 	"github.com/boltdb/bolt"
@@ -37,6 +38,16 @@ func CloseDB(db *bolt.DB) error {
 	return db.Close()
 }
 
+// InitDB init the db
+func InitDB() {
+	db, err := OpenDB()
+	if err != nil {
+		log.Printf("\nOpenDB error: %v", err)
+		return
+	}
+	CloseDB(db)
+}
+
 // AddVote ajoute un vote à la bdd
 func AddVote(db *bolt.DB, vote string, date time.Time) error {
 
@@ -72,9 +83,28 @@ func GetVotesOfTheDay(db *bolt.DB, date time.Time) ([]string, error) {
 	err := db.View(func(tx *bolt.Tx) error {
 		c := tx.Bucket([]byte("DB")).Bucket([]byte("VOTES")).Cursor()
 		min := []byte(time.Now().AddDate(0, 0, -1).Format(time.RFC3339))
-		max := []byte(time.Now().AddDate(0, 0, 0).Format(time.RFC3339))
+		max := []byte(date.Format(time.RFC3339))
 		for k, v := c.Seek(min); k != nil && bytes.Compare(k, max) <= 0; k, v = c.Next() {
 			res = append(res, string(v))
+			fmt.Println(string(k))
+			fmt.Println(string(v))
+		}
+		return nil
+	})
+	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))
+			fmt.Println(string(k))
+			fmt.Println(string(v))
 		}
 		return nil
 	})