Improve code readability
This commit is contained in:
+2
-2
@@ -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 {
|
||||
|
||||
@@ -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()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user