Skip to content

Commit ecab777

Browse files
committed
Add addr flag
1 parent 76888cb commit ecab777

File tree

3 files changed

+28
-4
lines changed

3 files changed

+28
-4
lines changed

cmd/skill/flags.go

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
package main
2+
3+
import (
4+
"flag"
5+
)
6+
7+
var flagRunAddr string
8+
9+
func parseFlags() {
10+
flag.StringVar(&flagRunAddr, "a", ":8080", "address and port to run server")
11+
flag.Parse()
12+
}

cmd/skill/main.go

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,9 @@
11
// Yandex Alice Skill
22
package main
33

4-
import "github.com/srg-bnd/alice-skill/internal/server"
4+
import (
5+
"github.com/srg-bnd/alice-skill/internal/server"
6+
)
57

68
type App struct {
79
server *server.Server
@@ -14,7 +16,9 @@ func NewApp() *App {
1416
}
1517

1618
func main() {
17-
if err := NewApp().server.Run(); err != nil {
19+
parseFlags()
20+
21+
if err := NewApp().server.Run(flagRunAddr); err != nil {
1822
panic(err)
1923
}
2024
}

internal/server/server.go

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,14 @@ func NewServer() *Server {
1717
}
1818

1919
// Init & run server
20-
func (s *Server) Run() error {
21-
return http.ListenAndServe(defaultHost, http.HandlerFunc(handlers.Webhook))
20+
func (s *Server) Run(addr string) error {
21+
return http.ListenAndServe(GetAddr(addr), http.HandlerFunc(handlers.Webhook))
22+
}
23+
24+
func GetAddr(addr string) string {
25+
if len(addr) != 0 {
26+
return addr
27+
} else {
28+
return defaultHost
29+
}
2230
}

0 commit comments

Comments
 (0)