formatato con gofmt
This commit is contained in:
parent
2c5e8b35a4
commit
eb67b14625
1 changed files with 42 additions and 42 deletions
84
llmongo.go
84
llmongo.go
|
|
@ -2,43 +2,43 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"github.com/fzzy/radix/redis"
|
||||
"flag"
|
||||
"fmt"
|
||||
"github.com/fzzy/radix/redis"
|
||||
"gopkg.in/mgo.v2"
|
||||
"os"
|
||||
"gopkg.in/mgo.v2"
|
||||
// "gopkg.in/mgo.v2/bson"
|
||||
"time"
|
||||
"strings"
|
||||
"strconv"
|
||||
// "gopkg.in/mgo.v2/bson"
|
||||
"log"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Options struct {
|
||||
MongoUri string
|
||||
RedisUri string
|
||||
rdb *redis.Client
|
||||
mdb *mgo.Session
|
||||
ll *mgo.Collection
|
||||
RedisTTL time.Duration
|
||||
LogFile string
|
||||
MongoUri string
|
||||
RedisUri string
|
||||
rdb *redis.Client
|
||||
mdb *mgo.Session
|
||||
ll *mgo.Collection
|
||||
RedisTTL time.Duration
|
||||
LogFile string
|
||||
}
|
||||
|
||||
type MongoLogin struct {
|
||||
User string `json:"user"`
|
||||
Protocol string `json:"protocol"`
|
||||
Ip string `json:"ip"`
|
||||
Date time.Time `json:"date"`
|
||||
User string `json:"user"`
|
||||
Protocol string `json:"protocol"`
|
||||
Ip string `json:"ip"`
|
||||
Date time.Time `json:"date"`
|
||||
}
|
||||
|
||||
var (
|
||||
opts = Options{
|
||||
// MongoUri: "mongodb://127.0.0.1:27018",
|
||||
// MongoUri: "mongodb://127.0.0.1:27018",
|
||||
MongoUri: "mongodb://10.39.81.85:27018",
|
||||
RedisUri: "127.0.0.1:6379",
|
||||
RedisTTL: time.Hour * 11688, // 16 mesi
|
||||
LogFile: "./log/llmongo.log",
|
||||
// RedisUri: "redis-ll.mail.tiscali.sys:6379",
|
||||
RedisTTL: time.Hour * 11688, // 16 mesi
|
||||
LogFile: "./log/llmongo.log",
|
||||
// RedisUri: "redis-ll.mail.tiscali.sys:6379",
|
||||
}
|
||||
)
|
||||
|
||||
|
|
@ -60,7 +60,7 @@ func connectRedis() {
|
|||
if err != nil {
|
||||
log.Println("Redis connect Error: ", err.Error())
|
||||
os.Exit(-1)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func connectMongo() {
|
||||
|
|
@ -75,16 +75,16 @@ func connectMongo() {
|
|||
|
||||
func main() {
|
||||
flag.Usage = usage
|
||||
flag.Parse()
|
||||
flag.Parse()
|
||||
|
||||
fs, err := os.OpenFile(opts.LogFile, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0666)
|
||||
if err != nil {
|
||||
fmt.Println("Log file error: ", err.Error())
|
||||
os.Exit(-4)
|
||||
}
|
||||
|
||||
|
||||
log.SetOutput(fs)
|
||||
// log.SetPrefix("[llmongo] ")
|
||||
// log.SetPrefix("[llmongo] ")
|
||||
|
||||
start := time.Now()
|
||||
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.
|
||||
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
|
||||
var date int64
|
||||
var lastval, val string
|
||||
|
|
@ -114,7 +114,7 @@ func main() {
|
|||
// reinserisce l'ultimo login e imposta il ttl su Redis
|
||||
retval := opts.rdb.Cmd("lpush", user, lastval)
|
||||
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
|
||||
}
|
||||
|
|
@ -123,39 +123,39 @@ func main() {
|
|||
log.Println("Login empty: ", user)
|
||||
retval := opts.rdb.Cmd("lrem", user, "-1", val)
|
||||
log.Println("lrem retval: ", user, val, retval)
|
||||
continue
|
||||
continue
|
||||
}
|
||||
sval := strings.Split(val, ":")
|
||||
// se il formato della riga di login non e' corretto
|
||||
if sval[1] == "" {
|
||||
log.Println("Login format error: ", val, user)
|
||||
retval := opts.rdb.Cmd("lrem", user, "-1", val)
|
||||
log.Println("lrem retval: ", user, val, retval)
|
||||
log.Println("lrem retval: ", user, val, retval)
|
||||
continue
|
||||
}
|
||||
date, err = strconv.ParseInt(sval[1], 10, 64)
|
||||
date, err = strconv.ParseInt(sval[1], 10, 64)
|
||||
if err != nil {
|
||||
log.Printf("Date Error: %+v - %s\n", err, user)
|
||||
continue
|
||||
}
|
||||
ml := MongoLogin{
|
||||
User: user,
|
||||
User: user,
|
||||
Protocol: sval[0],
|
||||
Ip: sval[2],
|
||||
Date: time.Unix(date, 0),
|
||||
Ip: sval[2],
|
||||
Date: time.Unix(date, 0),
|
||||
}
|
||||
// inserisce il login su Mongodb
|
||||
err = opts.ll.Insert(ml)
|
||||
if err != nil {
|
||||
log.Printf("Insert error: %+v\n", err)
|
||||
// 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)
|
||||
log.Println("lrem duplicate retval: ", retval, val)
|
||||
lastval = ""
|
||||
log.Println("lrem duplicate retval: ", retval, val)
|
||||
lastval = ""
|
||||
}
|
||||
continue
|
||||
}
|
||||
}
|
||||
// cancella da Redis la riga di login inserita
|
||||
retval := opts.rdb.Cmd("lrem", user, "-1", val)
|
||||
log.Println("lrem retval: ", retval, user, val)
|
||||
|
|
@ -165,11 +165,11 @@ func main() {
|
|||
llen, _ := opts.rdb.Cmd("llen", user).Int64()
|
||||
if llen == 0 {
|
||||
// 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)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
fmt.Printf("Stop %v\n", time.Since(start))
|
||||
log.Printf("Stop %v\n", time.Since(start))
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue