formatato con gofmt

This commit is contained in:
Michele Fadda 2015-07-10 14:35:35 +02:00
parent 2c5e8b35a4
commit eb67b14625

View file

@ -2,43 +2,43 @@
package main package main
import ( import (
"github.com/fzzy/radix/redis"
"flag" "flag"
"fmt" "fmt"
"github.com/fzzy/radix/redis"
"gopkg.in/mgo.v2"
"os" "os"
"gopkg.in/mgo.v2" // "gopkg.in/mgo.v2/bson"
// "gopkg.in/mgo.v2/bson"
"time"
"strings"
"strconv"
"log" "log"
"strconv"
"strings"
"time"
) )
type Options struct { type Options struct {
MongoUri string MongoUri string
RedisUri string RedisUri string
rdb *redis.Client rdb *redis.Client
mdb *mgo.Session mdb *mgo.Session
ll *mgo.Collection ll *mgo.Collection
RedisTTL time.Duration RedisTTL time.Duration
LogFile string LogFile string
} }
type MongoLogin struct { type MongoLogin struct {
User string `json:"user"` User string `json:"user"`
Protocol string `json:"protocol"` Protocol string `json:"protocol"`
Ip string `json:"ip"` Ip string `json:"ip"`
Date time.Time `json:"date"` Date time.Time `json:"date"`
} }
var ( var (
opts = Options{ opts = Options{
// MongoUri: "mongodb://127.0.0.1:27018", // MongoUri: "mongodb://127.0.0.1:27018",
MongoUri: "mongodb://10.39.81.85:27018", MongoUri: "mongodb://10.39.81.85:27018",
RedisUri: "127.0.0.1:6379", RedisUri: "127.0.0.1:6379",
RedisTTL: time.Hour * 11688, // 16 mesi RedisTTL: time.Hour * 11688, // 16 mesi
LogFile: "./log/llmongo.log", LogFile: "./log/llmongo.log",
// RedisUri: "redis-ll.mail.tiscali.sys:6379", // RedisUri: "redis-ll.mail.tiscali.sys:6379",
} }
) )
@ -60,7 +60,7 @@ func connectRedis() {
if err != nil { if err != nil {
log.Println("Redis connect Error: ", err.Error()) log.Println("Redis connect Error: ", err.Error())
os.Exit(-1) os.Exit(-1)
} }
} }
func connectMongo() { func connectMongo() {
@ -75,16 +75,16 @@ func connectMongo() {
func main() { func main() {
flag.Usage = usage flag.Usage = usage
flag.Parse() flag.Parse()
fs, err := os.OpenFile(opts.LogFile, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0666) fs, err := os.OpenFile(opts.LogFile, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0666)
if err != nil { if err != nil {
fmt.Println("Log file error: ", err.Error()) fmt.Println("Log file error: ", err.Error())
os.Exit(-4) os.Exit(-4)
} }
log.SetOutput(fs) log.SetOutput(fs)
// log.SetPrefix("[llmongo] ") // log.SetPrefix("[llmongo] ")
start := time.Now() start := time.Now()
fmt.Printf("Start: %+v\n", opts) fmt.Printf("Start: %+v\n", opts)
@ -98,9 +98,9 @@ func main() {
// Estrae la lista degli utenti che hanno fatto login negli ultimi X min. // Estrae la lista degli utenti che hanno fatto login negli ultimi X min.
llindex := opts.rdb.Cmd("smembers", "llindex") llindex := opts.rdb.Cmd("smembers", "llindex")
lista, err := llindex.List() lista, err := llindex.List()
for _, user := range lista { for _, user := range lista {
// cicla fino a che esistono righe di login // cicla fino a che esistono righe di login
var date int64 var date int64
var lastval, val string var lastval, val string
@ -114,7 +114,7 @@ func main() {
// reinserisce l'ultimo login e imposta il ttl su Redis // reinserisce l'ultimo login e imposta il ttl su Redis
retval := opts.rdb.Cmd("lpush", user, lastval) retval := opts.rdb.Cmd("lpush", user, lastval)
ttl := opts.rdb.Cmd("expire", user, opts.RedisTTL.Seconds()) ttl := opts.rdb.Cmd("expire", user, opts.RedisTTL.Seconds())
log.Println("lpush retval: ", retval, ttl, user, lastval) log.Println("lpush retval: ", retval, ttl, user, lastval)
} }
break break
} }
@ -123,39 +123,39 @@ func main() {
log.Println("Login empty: ", user) log.Println("Login empty: ", user)
retval := opts.rdb.Cmd("lrem", user, "-1", val) retval := opts.rdb.Cmd("lrem", user, "-1", val)
log.Println("lrem retval: ", user, val, retval) log.Println("lrem retval: ", user, val, retval)
continue continue
} }
sval := strings.Split(val, ":") sval := strings.Split(val, ":")
// se il formato della riga di login non e' corretto // se il formato della riga di login non e' corretto
if sval[1] == "" { if sval[1] == "" {
log.Println("Login format error: ", val, user) log.Println("Login format error: ", val, user)
retval := opts.rdb.Cmd("lrem", user, "-1", val) retval := opts.rdb.Cmd("lrem", user, "-1", val)
log.Println("lrem retval: ", user, val, retval) log.Println("lrem retval: ", user, val, retval)
continue continue
} }
date, err = strconv.ParseInt(sval[1], 10, 64) date, err = strconv.ParseInt(sval[1], 10, 64)
if err != nil { if err != nil {
log.Printf("Date Error: %+v - %s\n", err, user) log.Printf("Date Error: %+v - %s\n", err, user)
continue continue
} }
ml := MongoLogin{ ml := MongoLogin{
User: user, User: user,
Protocol: sval[0], Protocol: sval[0],
Ip: sval[2], Ip: sval[2],
Date: time.Unix(date, 0), Date: time.Unix(date, 0),
} }
// inserisce il login su Mongodb // inserisce il login su Mongodb
err = opts.ll.Insert(ml) err = opts.ll.Insert(ml)
if err != nil { if err != nil {
log.Printf("Insert error: %+v\n", err) log.Printf("Insert error: %+v\n", err)
// se l'errore e' "duplicate key error" // se l'errore e' "duplicate key error"
if strings.Contains(err.Error(), "E11000") { if strings.Contains(err.Error(), "E11000") {
retval := opts.rdb.Cmd("lrem", user, "-1", val) retval := opts.rdb.Cmd("lrem", user, "-1", val)
log.Println("lrem duplicate retval: ", retval, val) log.Println("lrem duplicate retval: ", retval, val)
lastval = "" lastval = ""
} }
continue continue
} }
// cancella da Redis la riga di login inserita // cancella da Redis la riga di login inserita
retval := opts.rdb.Cmd("lrem", user, "-1", val) retval := opts.rdb.Cmd("lrem", user, "-1", val)
log.Println("lrem retval: ", retval, user, val) log.Println("lrem retval: ", retval, user, val)
@ -165,11 +165,11 @@ func main() {
llen, _ := opts.rdb.Cmd("llen", user).Int64() llen, _ := opts.rdb.Cmd("llen", user).Int64()
if llen == 0 { if llen == 0 {
// elimina l'utente dalla lista di quelli che hanno fatto login // elimina l'utente dalla lista di quelli che hanno fatto login
retval := opts.rdb.Cmd("srem", "llindex", user) retval := opts.rdb.Cmd("srem", "llindex", user)
log.Println("srem retval: ", retval, user) log.Println("srem retval: ", retval, user)
} }
} }
fmt.Printf("Stop %v\n", time.Since(start)) fmt.Printf("Stop %v\n", time.Since(start))
log.Printf("Stop %v\n", time.Since(start)) log.Printf("Stop %v\n", time.Since(start))
} }