huly-platform/plugins/workbench-resources/src/components/ServerManagerGeneral.svelte

<script lang="ts">
  import core, { RateLimiter } from '@hcengineering/core'
  import login from '@hcengineering/login'
  import { getEmbeddedLabel, getMetadata } from '@hcengineering/platform'
  import presentation, { getClient, isAdminUser } from '@hcengineering/presentation'
  import { Button, IconArrowLeft, IconArrowRight, fetchMetadataLocalStorage } from '@hcengineering/ui'
  import EditBox from '@hcengineering/ui/src/components/EditBox.svelte'

  const _endpoint: string = fetchMetadataLocalStorage(login.metadata.LoginEndpoint) ?? ''
  const token: string = getMetadata(presentation.metadata.Token) ?? ''

  let endpoint = _endpoint.replace(/^ws/g, 'http')
  if (endpoint.endsWith('/')) {
    endpoint = endpoint.substring(0, endpoint.length - 1)
  }
  let warningTimeout = 15

  let commandsToSend = 1000
  let commandsToSendParallel = 1

  let running = false

  let maxTime = 0

  let avgTime = 0

  let rps = 0

  let active = 0

  let opss = 0

  let dataSize = 0

  let responseSize = 0

  function genData (dataSize: number): string {
    let result = ''
    const characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789'
    const charactersLength = characters.length
    for (let i = 0; i < dataSize; i++) {
      result += characters.charAt(Math.floor(Math.random() * charactersLength))
    }
    return result
  }

  async function doBenchmark (): Promise<void> {
    avgTime = 0
    maxTime = 0
    let count = commandsToSend
    let ops = 0
    avgTime = 0
    opss = 0
    const int = setInterval(() => {
      rps = ops
      ops = 0
    }, 1000)
    const rate = new RateLimiter(commandsToSendParallel)
    const client = getClient()

    const doOp = async () => {
      const st = Date.now()
      active++
      await client.createDoc(core.class.BenchmarkDoc, core.space.Configuration, {
        source: genData(dataSize),
        request: {
          documents: 1,
          size: responseSize
        }
      })
      active--
      const ed = Date.now()

      if (ed - st > maxTime) {
        maxTime = ed - st
      }
      if (avgTime !== 0) {
        avgTime += ed - st
      } else {
        avgTime = ed - st
      }
      ops++
      opss++
      count--
    }
    // eslint-disable-next-line no-unmodified-loop-condition
    while (count > 0 && running) {
      if (commandsToSendParallel > 0) {
        await rate.add(async () => {
          await doOp()
        })
      } else {
        await doOp()
      }
    }
    await rate.waitProcessing()
    running = false
    clearInterval(int)
  }
</script>

{#if isAdminUser()}
  <div class="flex flex-col">
    <div class="flex-row-center p-1">
      <div class="p-3">1.</div>
      <Button
        icon={IconArrowRight}
        label={getEmbeddedLabel('Set maintenance warning')}
        on:click={() => {
          const endpoint = getMetadata(login.metadata.AccountsUrl) ?? ''
          if (endpoint !== '') {
            void fetch(endpoint + `/api/v1/manage?token=${token}&operation=maintenance&timeout=${warningTimeout}`, {
              method: 'PUT'
            })
          }
        }}
      />
      <div class="flex-col p-1">
        <div class="flex-row-center p-1">
          <EditBox kind={'underline'} format={'number'} bind:value={warningTimeout} /> min
        </div>
      </div>
      <Button
        icon={IconArrowLeft}
        label={getEmbeddedLabel('Clear warning')}
        on:click={() => {
          const endpoint = getMetadata(login.metadata.AccountsUrl) ?? ''
          if (endpoint !== '') {
            void fetch(endpoint + `/api/v1/manage?token=${token}&operation=maintenance&timeout=-1`, {
              method: 'PUT'
            })
          }
        }}
      />
    </div>
    <div class="flex-col p-1">
      <div class="flex-row-center p-1">
        Command benchmark {avgTime / opss}
        {maxTime} - {active}
      </div>
      <div class="flex-row-center p-1">
        <div class="flex-row-center p-1">
          <EditBox kind={'underline'} format={'number'} bind:value={commandsToSend} /> total
        </div>
        <div class="flex-row-center p-1">
          <EditBox kind={'underline'} format={'number'} bind:value={commandsToSendParallel} /> parallel
        </div>
        <div class="flex-row-center p-1">
          <EditBox kind={'underline'} format={'number'} bind:value={dataSize} /> dsize
        </div>
        <div class="flex-row-center p-1">
          <EditBox kind={'underline'} format={'number'} bind:value={responseSize} /> rsize
        </div>
        <Button
          label={getEmbeddedLabel('Benchmark')}
          on:click={() => {
            running = !running
            void doBenchmark()
          }}
        />
      </div>
    </div>

    <div class="flex-row-center p-1">
      <div class="p-3">2.</div>
      <Button
        icon={IconArrowRight}
        label={getEmbeddedLabel('Reboot server')}
        on:click={() => {
          void fetch(endpoint + `/api/v1/manage?token=${token}&operation=reboot`, {
            method: 'PUT'
          })
        }}
      />
    </div>
  </div>
{/if}

<style lang="scss">
  .greyed {
    color: rgba(black, 0.5);
  }
</style>