/
main.go
55 lines (47 loc) · 1.16 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
package main
import (
"database/sql"
"flag"
"github.com/KHYehor/Za_stepuhu/Lab2/server/db"
"log"
"net/http"
"os"
"os/signal"
)
var httpPortNumber = flag.Int("p", 8080, "HTTP port number")
func NewDbConnection() (*sql.DB, error) {
conn := &db.Connection{
DbName: "tablets",
User: "ykh2k",
Password: "1111",
Host: "localhost",
DisableSSL: true,
}
return conn.Open()
}
func main() {
// Parse command line arguments. Port number may be defined with "-p" flag.
flag.Parse()
// Create the server.
if server, err := ComposeApiServer(HttpPortNumber(*httpPortNumber)); err == nil {
// Start it.
go func() {
log.Println("Starting chat server...")
err := server.Start()
if err == http.ErrServerClosed {
log.Printf("HTTP server stopped")
} else {
log.Fatalf("Cannot start HTTP server: %s", err)
}
}()
// Wait for Ctrl-C signal.
sigChannel := make(chan os.Signal, 1)
signal.Notify(sigChannel, os.Interrupt)
<-sigChannel
if err := server.Stop(); err != nil && err != http.ErrServerClosed {
log.Printf("Error stopping the server: %s", err)
}
} else {
log.Fatalf("Cannot initialize chat server: %s", err)
}
}