#!/bin/bash
# install loki
# by zack.zhu@aloudata.com
# 202112301633

FILE_NAME="loki-linux-amd64"
FILE_URL="http://10.5.20.30/download/tools/loki-linux-amd64"
LOKI_HOME="/data/loki"
LOKI_CONF="$LOKI_HOME/loki-config.yaml"
# IP_ADDR=$(ip a|grep inet|grep ' 10.[5|10].'|awk '{print $2}'|awk -F '/' '{print $1}')

mkdir -p $LOKI_HOME >/dev/null 2>&1
curl -s -o $LOKI_HOME/$FILE_NAME $FILE_URL
if [ $? -ne 0 ]; then
        echo "download file is error."
        exit 1
fi
chmod +x $LOKI_HOME/$FILE_NAME

cat > $LOKI_CONF <<EOF
auth_enabled: false

server:
  http_listen_port: 3100

ingester:
  lifecycler:
    address: 127.0.0.1
    ring:
      kvstore:
        store: inmemory
      replication_factor: 1
    final_sleep: 0s
  chunk_idle_period: 5m
  chunk_retain_period: 30s

schema_config:
  configs:
  - from: 2020-05-15
    store: boltdb
    object_store: filesystem
    schema: v11
    index:
      prefix: index_
      period: 720h

storage_config:
  boltdb:
    directory: $LOKI_HOME/index

  filesystem:
    directory: $LOKI_HOME/chunks

limits_config:
  enforce_metric_name: false
  reject_old_samples: true
  reject_old_samples_max_age: 168h
EOF

if [ $? -ne 0 ]; then
        echo 'config error.'
        exit 1
fi

cat > /usr/lib/systemd/system/loki.service <<EOF
[Unit]
Description=loki
After=network.target
[Service]
Type=simple
User=root
ExecStart=$LOKI_HOME/$FILE_NAME -config.file=$LOKI_CONF -config.expand-env=true
Restart=on-failure
[Install]
WantedBy=multi-user.target
EOF

if [ $? -ne 0 ]; then
        echo "systemd error."
        exit 1
fi

systemctl daemon-reload
systemctl enable loki.service
systemctl start loki.service