Skip to content

Commit a8eeb9d

Browse files
authored
Merge pull request #7 from rtmelsov/develop
feat: добавлены данные из env
2 parents df88ed6 + d908b3b commit a8eeb9d

File tree

3 files changed

+29
-12
lines changed

3 files changed

+29
-12
lines changed

.env.dev

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,18 @@
11
AK_ENV=prod
2-
AK_GRPC_ADDR=0.0.0.0:3200
2+
GRPC_ADDR=127.0.0.1:8080
3+
AK_GRPC_ADDR=127.0.0.1:8080
34

4-
# строка подключения к Postgres (пример)
5-
AK_POSTGRES_DSN=postgres://admin:[email protected]:5432/adv?sslmode=disable
5+
# Postgres DSN (одинаковое значение под разными именами)
6+
DB_DSN=postgres://admin:[email protected]:5432/adv?sslmode=disable
7+
DSN=postgres://admin:[email protected]:5432/adv?sslmode=disable
8+
DATABASE_URL=postgres://admin:[email protected]:5432/adv?sslmode=disable
9+
AK_POSTGRES_DSN=postgres://admin:[email protected]:5432/adv?sslmode=disable
10+
AK_DSN=postgres://admin:[email protected]:5432/adv?sslmode=disable
611

7-
# куда складывать файлы пользователей
12+
# хранилище файлов
13+
FILES_DIR=/opt/adv-keeper/data
814
AK_STORAGE_DIR=/opt/adv-keeper/data
915

10-
# опционально
16+
# логирование
1117
AK_LOG_LEVEL=info
12-
18+
LOG_LEVEL=info

cmd/server/main.go

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,10 @@ package main
22

33
import (
44
"database/sql"
5-
"fmt"
65
"google.golang.org/grpc/reflection"
76
"log"
87
"net"
8+
"os"
99

1010
commonv1 "github.com/rtmelsov/adv-keeper/gen/go/proto/common/v1"
1111
filev1 "github.com/rtmelsov/adv-keeper/gen/go/proto/file/v1"
@@ -18,12 +18,17 @@ import (
1818
)
1919

2020
func main() {
21+
dsn := os.Getenv("DB_DSN")
22+
if dsn == "" {
23+
log.Fatal("DB_DSN is required")
24+
}
25+
addr := os.Getenv("GRPC_ADDR")
26+
if addr == "" {
27+
addr = "127.0.0.1:8080"
28+
}
2129
// получаем урл бд
22-
dsn := "postgres://postgres@localhost:5432/dbname?sslmode=disable"
23-
24-
fmt.Println("dsn", dsn)
2530

26-
lis, err := net.Listen("tcp", "127.0.0.1:8080")
31+
lis, err := net.Listen("tcp", addr)
2732
if err != nil {
2833
log.Fatal(err)
2934
}

internal/file/service.go

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,13 @@ type FileServer struct {
2525
uploadDir string
2626
}
2727

28-
func New(q *db.Queries) *FileServer { return &FileServer{Q: q, uploadDir: "var/"} }
28+
func New(q *db.Queries) *FileServer {
29+
filesDir := os.Getenv("FILES_DIR")
30+
if filesDir == "" {
31+
filesDir = "./data"
32+
}
33+
return &FileServer{Q: q, uploadDir: filesDir}
34+
}
2935

3036
func (s *FileServer) Upload(stream filev1.FileService_UploadServer) error {
3137
var (

0 commit comments

Comments
 (0)