birbnetes/input-service
birbnetes/input-service
4
0
Fork 0
Code Pull Requests Releases Activity

made InfluxDB optional
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Pünkösd Marcell 2021-07-26 12:51:52 +02:00
parent 3c10a351ba
commit f15517af62
3 changed files with 19 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/app.py
View File

@ -40,12 +40,14 @@ api = Api(app)
health = HealthCheck() health = HealthCheck()
db.init_app(app) db.init_app(app)
ma.init_app(app) ma.init_app(app)
influx_db.init_app(app) if Config.ENABLE_INFLUXDB:
influx_db.init_app(app)
@app.before_first_request @app.before_first_request
def init_db(): def init_db():
influx_db.database.create(Config.INFLUXDB_DATABASE) if Config.ENABLE_INFLUXDB:
influx_db.database.create(Config.INFLUXDB_DATABASE)
db.create_all() db.create_all()

1
src/config.py
View File

@ -40,6 +40,7 @@ class Config:
STORAGE_HOSTNAME = os.getenv("INPUT_STORAGE_HOSTNAME", "localhost:8042") STORAGE_HOSTNAME = os.getenv("INPUT_STORAGE_HOSTNAME", "localhost:8042")
ENABLE_INFLUXDB = os.environ.get("INPUT_ENABLE_INFLUX", "true").lower() in ["true", "yes", "1"]
INFLUXDB_HOST = os.getenv("INFLUX_HOST", "input-influx") INFLUXDB_HOST = os.getenv("INFLUX_HOST", "input-influx")
INFLUXDB_PORT = os.getenv("INFLUX_PORT", "8086") INFLUXDB_PORT = os.getenv("INFLUX_PORT", "8086")
INFLUXDB_USER = os.getenv("INFLUX_USERNAME", "input-service") INFLUXDB_USER = os.getenv("INFLUX_USERNAME", "input-service")

27
src/resources.py
View File

@ -118,20 +118,21 @@ class SampleResource(Resource):
current_app.logger.exception(e) current_app.logger.exception(e)
return abort(569, "AMPQ Publish error") return abort(569, "AMPQ Publish error")
influx_db.write_points( if current_app.config['ENABLE_INFLUXDB']:
[ influx_db.write_points(
{ [
'time': datetime.now(tz=tzlocal.get_localzone()), {
'measurement': 'cloudinput', 'time': datetime.now(tz=tzlocal.get_localzone()),
'tags': { 'measurement': 'cloudinput',
'device': desc['device_id'] 'tags': {
}, 'device': desc['device_id']
'fields': { },
'bruh': 1.0 'fields': {
'bruh': 1.0
}
} }
} ]
] )
)
db.session.commit() db.session.commit()
return {"tag": generated_tag}, 200 return {"tag": generated_tag}, 200