liz/PluralKit
liz/PluralKit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make InfluxDB connection options configurable

Browse Source
This commit is contained in:
Ske 2018-09-01 19:16:42 +02:00
parent a130e2215a
commit 99e2fad2b2
3 changed files with 19 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
docker-compose.yml
View File

@ -16,6 +16,9 @@ services:
- "DATABASE_NAME=postgres"
- "DATABASE_HOST=db"
- "DATABASE_PORT=5432"
- "INFLUX_HOST=influx"
- "INFLUX_PORT=8086"
- "INFLUX_DB=pluralkit"
restart: always
api:
build: src/

13
src/pluralkit/bot/__init__.py
View File

@ -126,11 +126,16 @@ class PluralKitBot:
async with self.pool.acquire() as conn:
await db.create_tables(conn)
self.logger.info("Connecting to InfluxDB...")
self.stats = await InfluxStatCollector.connect()
if "INFLUX_HOST" in os.environ:
self.logger.info("Connecting to InfluxDB...")
self.stats = await InfluxStatCollector.connect(
os.environ["INFLUX_HOST"],
os.environ["INFLUX_PORT"],
os.environ["INFLUX_DB"]
)
self.logger.info("Starting periodical stat reporting...")
asyncio.get_event_loop().create_task(self.periodical_stat_timer(self.pool))
self.logger.info("Starting periodical stat reporting...")
asyncio.get_event_loop().create_task(self.periodical_stat_timer(self.pool))
self.logger.info("Connecting to Discord...")
await self.client.start(self.token)

6
src/pluralkit/stats.py
View File

@ -21,9 +21,9 @@ class NullStatCollector(StatCollector):
class InfluxStatCollector(StatCollector):
@staticmethod
async def connect():
client = InfluxDBClient(host="influx", db="pluralkit")
await client.create_database(db="pluralkit")
async def connect(host: str, port: int, db: str):
client = InfluxDBClient(host=host, port=port, db=db)
await client.create_database(db=db)
return InfluxStatCollector(client)