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
|
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))
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Reference in a new issue