Compare commits

..
4 Commits
Author SHA1 Message Date
Dustin Pianalto 6262ebfeb4 Fix bug with query
CI / build (push) Has been cancelled
2021-02-23 00:38:20 -09:00
Dustin Pianalto 005dfc815f Move migrations to separate func
CI / build (push) Has been cancelled
2021-02-22 23:59:13 -09:00
Dustin Pianalto 986d16b346 Add ftp credentials to domain types
CI / build (push) Has been cancelled
2021-02-22 23:24:19 -09:00
Dustin Pianalto 83c41e2788 Add ftp credentials to db schema
CI / build (push) Has been cancelled
2021-02-22 23:10:57 -09:00
6 changed files with 34 additions and 9 deletions
+1
View File
@@ -28,6 +28,7 @@ func main() {
}
database.ConnectDatabase(os.Getenv("DATABASE_URL"))
database.RunMigrations()
services.InitializeServices()
owners := []string{
+10 -6
View File
@@ -23,10 +23,6 @@ var (
)
func ConnectDatabase(dbConnString string) {
s := bindata.Resource(migrations.AssetNames(),
func(name string) ([]byte, error) {
return migrations.Asset(name)
})
var err error
db, err = sql.Open("postgres", dbConnString)
if err != nil {
@@ -36,6 +32,15 @@ func ConnectDatabase(dbConnString string) {
db.SetMaxOpenConns(75)
db.SetMaxIdleConns(5)
db.SetConnMaxLifetime(300)
initServices()
log.Println("Services Initialized")
}
func RunMigrations() {
s := bindata.Resource(migrations.AssetNames(),
func(name string) ([]byte, error) {
return migrations.Asset(name)
})
d, err := bindata.WithInstance(s)
if err != nil {
log.Fatal(fmt.Errorf("cannot load migrations: %w", err))
@@ -57,8 +62,7 @@ func ConnectDatabase(dbConnString string) {
}
}
log.Println("Migrations Run")
initServices()
log.Println("Services Initialized")
}
func initServices() {
@@ -0,0 +1,6 @@
BEGIN;
ALTER TABLE servers
DROP COLUMN ftp_port,
DROP COLUMN ftp_username,
DROP COLUMN ftp_password;
COMMIT;
@@ -0,0 +1,6 @@
BEGIN;
ALTER TABLE servers
ADD COLUMN ftp_port int4,
ADD COLUMN ftp_username varchar(200),
ADD COLUMN ftp_password varchar(200);
COMMIT;
+8 -3
View File
@@ -19,11 +19,12 @@ func (s serverService) ServerByID(id int) (geeksbot.Server, error) {
var iMsgID sql.NullString
var sMsgID sql.NullString
queryString := `SELECT id, name, ip_address, port, password, alerts_channel_id,
guild_id, info_channel_id, info_message_id, settings_message_id
guild_id, info_channel_id, info_message_id, settings_message_id,
ftp_port, ftp_username, ftp_password
FROM servers WHERE id = $1`
row := s.db.QueryRow(queryString, id)
err := row.Scan(&server.ID, &server.Name, &server.IPAddr, &server.Port, &server.Password,
&aChanID, &guildID, &iChanID, &iMsgID, &sMsgID)
&aChanID, &guildID, &iChanID, &iMsgID, &sMsgID, &server.FTPPort, &server.FTPUser, &server.FTPPass)
if err != nil {
return geeksbot.Server{}, err
}
@@ -115,7 +116,8 @@ func (s serverService) DeleteServer(server geeksbot.Server) error {
func (s serverService) UpdateServer(server geeksbot.Server) (geeksbot.Server, error) {
queryString := `UPDATE servers SET name = $2, ip_address = $3, port = $4, password = $5,
alerts_channel_id = $6, info_channel_id = $7, info_message_id = $8,
settings_message_id = $9 WHERE id = $1`
settings_message_id = $9, ftp_port = $10, ftp_username = $11,
ftp_password = $12 WHERE id = $1`
_, err := s.db.Exec(queryString,
server.Name,
server.IPAddr,
@@ -125,6 +127,9 @@ func (s serverService) UpdateServer(server geeksbot.Server) (geeksbot.Server, er
server.InfoChannel.ID,
server.InfoMessage.ID,
server.SettingsMessage.ID,
server.FTPPort,
server.FTPUser,
server.FTPPass,
)
return server, err
}
+3
View File
@@ -11,6 +11,9 @@ type Server struct {
InfoChannel *Channel
InfoMessage *Message
SettingsMessage *Message
FTPPort int
FTPUser string
FTPPass string
}
type ServerService interface {