1
0
Fork 0

Improve code readability

This commit is contained in:
Gregory Eremin 2014-09-11 22:53:53 +04:00
parent 3fc1601f5d
commit e0b3743a00
2 changed files with 11 additions and 12 deletions

View File

@ -21,7 +21,7 @@ func New(st *storage.Storage) *Hub {
subscribers: []*Subscription{},
}
go h.cleanupPeriodically()
go h.cleanupEverySecond()
return h
}
@ -57,7 +57,7 @@ func (h *Hub) Sub(s *Subscription) {
h.subscribers = append(h.subscribers, s)
}
func (h *Hub) cleanupPeriodically() {
func (h *Hub) cleanupEverySecond() {
t := time.NewTicker(1 * time.Second)
for {

19
main.go
View File

@ -17,9 +17,8 @@ const (
)
var (
theStorage *storage.Storage
theHub *hub.Hub
config struct {
theHub *hub.Hub
config struct {
storage string
port int
}
@ -30,18 +29,16 @@ func main() {
flag.IntVar(&config.port, "port", 4401, "Server HTTP port")
flag.Parse()
theStorage, err := storage.New(config.storage)
store, err := storage.New(config.storage)
if err != nil {
panic(err)
}
theHub = hub.New(theStorage)
ch := make(chan os.Signal)
signal.Notify(ch, os.Interrupt, os.Kill, syscall.SIGTERM, syscall.SIGINT)
shutdown := make(chan os.Signal)
signal.Notify(shutdown, os.Interrupt, os.Kill, syscall.SIGTERM, syscall.SIGINT)
go func() {
<-ch
theStorage.Close()
<-shutdown
store.Close()
os.Exit(0)
}()
@ -50,5 +47,7 @@ func main() {
fmt.Println("Storage path: %s", config.storage)
fmt.Println("Server is running at http://127.0.0.1:%d", config.port)
theHub = hub.New(store)
startServer()
}