first commit
This commit is contained in:
commit
e694f6d7d5
10 changed files with 655 additions and 0 deletions
13
Dockerfile
Normal file
13
Dockerfile
Normal file
|
@ -0,0 +1,13 @@
|
|||
FROM docker.io/golang:1.24.2 AS builder
|
||||
WORKDIR /src
|
||||
COPY . ./
|
||||
RUN go build -o /bin/2b2t-monitor .
|
||||
|
||||
FROM docker.io/debian:latest AS runner
|
||||
WORKDIR /run
|
||||
RUN apt-get update && \
|
||||
apt-get install -y ca-certificates && \
|
||||
update-ca-certificates
|
||||
|
||||
COPY --from=builder /bin/2b2t-monitor /bin/2b2t-monitor
|
||||
CMD ["/bin/2b2t-monitor"]
|
356
LICENSE.md
Normal file
356
LICENSE.md
Normal file
|
@ -0,0 +1,356 @@
|
|||
Mozilla Public License Version 2.0
|
||||
==================================
|
||||
|
||||
### 1. Definitions
|
||||
|
||||
**1.1. “Contributor”**
|
||||
means each individual or legal entity that creates, contributes to
|
||||
the creation of, or owns Covered Software.
|
||||
|
||||
**1.2. “Contributor Version”**
|
||||
means the combination of the Contributions of others (if any) used
|
||||
by a Contributor and that particular Contributor's Contribution.
|
||||
|
||||
**1.3. “Contribution”**
|
||||
means Covered Software of a particular Contributor.
|
||||
|
||||
**1.4. “Covered Software”**
|
||||
means Source Code Form to which the initial Contributor has attached
|
||||
the notice in Exhibit A, the Executable Form of such Source Code
|
||||
Form, and Modifications of such Source Code Form, in each case
|
||||
including portions thereof.
|
||||
|
||||
**1.5. “Incompatible With Secondary Licenses”**
|
||||
means
|
||||
|
||||
* **(a)** that the initial Contributor has attached the notice described
|
||||
in Exhibit B to the Covered Software; or
|
||||
* **(b)** that the Covered Software was made available under the terms of
|
||||
version 1.1 or earlier of the License, but not also under the
|
||||
terms of a Secondary License.
|
||||
|
||||
**1.6. “Executable Form”**
|
||||
means any form of the work other than Source Code Form.
|
||||
|
||||
**1.7. “Larger Work”**
|
||||
means a work that combines Covered Software with other material, in
|
||||
a separate file or files, that is not Covered Software.
|
||||
|
||||
**1.8. “License”**
|
||||
means this document.
|
||||
|
||||
**1.9. “Licensable”**
|
||||
means having the right to grant, to the maximum extent possible,
|
||||
whether at the time of the initial grant or subsequently, any and
|
||||
all of the rights conveyed by this License.
|
||||
|
||||
**1.10. “Modifications”**
|
||||
means any of the following:
|
||||
|
||||
* **(a)** any file in Source Code Form that results from an addition to,
|
||||
deletion from, or modification of the contents of Covered
|
||||
Software; or
|
||||
* **(b)** any new file in Source Code Form that contains any Covered
|
||||
Software.
|
||||
|
||||
**1.11. “Patent Claims” of a Contributor**
|
||||
means any patent claim(s), including without limitation, method,
|
||||
process, and apparatus claims, in any patent Licensable by such
|
||||
Contributor that would be infringed, but for the grant of the
|
||||
License, by the making, using, selling, offering for sale, having
|
||||
made, import, or transfer of either its Contributions or its
|
||||
Contributor Version.
|
||||
|
||||
**1.12. “Secondary License”**
|
||||
means either the GNU General Public License, Version 2.0, the GNU
|
||||
Lesser General Public License, Version 2.1, the GNU Affero General
|
||||
Public License, Version 3.0, or any later versions of those
|
||||
licenses.
|
||||
|
||||
**1.13. “Source Code Form”**
|
||||
means the form of the work preferred for making modifications.
|
||||
|
||||
**1.14. “You” (or “Your”)**
|
||||
means an individual or a legal entity exercising rights under this
|
||||
License. For legal entities, “You” includes any entity that
|
||||
controls, is controlled by, or is under common control with You. For
|
||||
purposes of this definition, “control” means **(a)** the power, direct
|
||||
or indirect, to cause the direction or management of such entity,
|
||||
whether by contract or otherwise, or **(b)** ownership of more than
|
||||
fifty percent (50%) of the outstanding shares or beneficial
|
||||
ownership of such entity.
|
||||
|
||||
|
||||
### 2. License Grants and Conditions
|
||||
|
||||
#### 2.1. Grants
|
||||
|
||||
Each Contributor hereby grants You a world-wide, royalty-free,
|
||||
non-exclusive license:
|
||||
|
||||
* **(a)** under intellectual property rights (other than patent or trademark)
|
||||
Licensable by such Contributor to use, reproduce, make available,
|
||||
modify, display, perform, distribute, and otherwise exploit its
|
||||
Contributions, either on an unmodified basis, with Modifications, or
|
||||
as part of a Larger Work; and
|
||||
* **(b)** under Patent Claims of such Contributor to make, use, sell, offer
|
||||
for sale, have made, import, and otherwise transfer either its
|
||||
Contributions or its Contributor Version.
|
||||
|
||||
#### 2.2. Effective Date
|
||||
|
||||
The licenses granted in Section 2.1 with respect to any Contribution
|
||||
become effective for each Contribution on the date the Contributor first
|
||||
distributes such Contribution.
|
||||
|
||||
#### 2.3. Limitations on Grant Scope
|
||||
|
||||
The licenses granted in this Section 2 are the only rights granted under
|
||||
this License. No additional rights or licenses will be implied from the
|
||||
distribution or licensing of Covered Software under this License.
|
||||
Notwithstanding Section 2.1(b) above, no patent license is granted by a
|
||||
Contributor:
|
||||
|
||||
* **(a)** for any code that a Contributor has removed from Covered Software;
|
||||
or
|
||||
* **(b)** for infringements caused by: **(i)** Your and any other third party's
|
||||
modifications of Covered Software, or **(ii)** the combination of its
|
||||
Contributions with other software (except as part of its Contributor
|
||||
Version); or
|
||||
* **(c)** under Patent Claims infringed by Covered Software in the absence of
|
||||
its Contributions.
|
||||
|
||||
This License does not grant any rights in the trademarks, service marks,
|
||||
or logos of any Contributor (except as may be necessary to comply with
|
||||
the notice requirements in Section 3.4).
|
||||
|
||||
#### 2.4. Subsequent Licenses
|
||||
|
||||
No Contributor makes additional grants as a result of Your choice to
|
||||
distribute the Covered Software under a subsequent version of this
|
||||
License (see Section 10.2) or under the terms of a Secondary License (if
|
||||
permitted under the terms of Section 3.3).
|
||||
|
||||
#### 2.5. Representation
|
||||
|
||||
Each Contributor represents that the Contributor believes its
|
||||
Contributions are its original creation(s) or it has sufficient rights
|
||||
to grant the rights to its Contributions conveyed by this License.
|
||||
|
||||
#### 2.6. Fair Use
|
||||
|
||||
This License is not intended to limit any rights You have under
|
||||
applicable copyright doctrines of fair use, fair dealing, or other
|
||||
equivalents.
|
||||
|
||||
#### 2.7. Conditions
|
||||
|
||||
Sections 3.1, 3.2, 3.3, and 3.4 are conditions of the licenses granted
|
||||
in Section 2.1.
|
||||
|
||||
|
||||
### 3. Responsibilities
|
||||
|
||||
#### 3.1. Distribution of Source Form
|
||||
|
||||
All distribution of Covered Software in Source Code Form, including any
|
||||
Modifications that You create or to which You contribute, must be under
|
||||
the terms of this License. You must inform recipients that the Source
|
||||
Code Form of the Covered Software is governed by the terms of this
|
||||
License, and how they can obtain a copy of this License. You may not
|
||||
attempt to alter or restrict the recipients' rights in the Source Code
|
||||
Form.
|
||||
|
||||
#### 3.2. Distribution of Executable Form
|
||||
|
||||
If You distribute Covered Software in Executable Form then:
|
||||
|
||||
* **(a)** such Covered Software must also be made available in Source Code
|
||||
Form, as described in Section 3.1, and You must inform recipients of
|
||||
the Executable Form how they can obtain a copy of such Source Code
|
||||
Form by reasonable means in a timely manner, at a charge no more
|
||||
than the cost of distribution to the recipient; and
|
||||
|
||||
* **(b)** You may distribute such Executable Form under the terms of this
|
||||
License, or sublicense it under different terms, provided that the
|
||||
license for the Executable Form does not attempt to limit or alter
|
||||
the recipients' rights in the Source Code Form under this License.
|
||||
|
||||
#### 3.3. Distribution of a Larger Work
|
||||
|
||||
You may create and distribute a Larger Work under terms of Your choice,
|
||||
provided that You also comply with the requirements of this License for
|
||||
the Covered Software. If the Larger Work is a combination of Covered
|
||||
Software with a work governed by one or more Secondary Licenses, and the
|
||||
Covered Software is not Incompatible With Secondary Licenses, this
|
||||
License permits You to additionally distribute such Covered Software
|
||||
under the terms of such Secondary License(s), so that the recipient of
|
||||
the Larger Work may, at their option, further distribute the Covered
|
||||
Software under the terms of either this License or such Secondary
|
||||
License(s).
|
||||
|
||||
#### 3.4. Notices
|
||||
|
||||
You may not remove or alter the substance of any license notices
|
||||
(including copyright notices, patent notices, disclaimers of warranty,
|
||||
or limitations of liability) contained within the Source Code Form of
|
||||
the Covered Software, except that You may alter any license notices to
|
||||
the extent required to remedy known factual inaccuracies.
|
||||
|
||||
#### 3.5. Application of Additional Terms
|
||||
|
||||
You may choose to offer, and to charge a fee for, warranty, support,
|
||||
indemnity or liability obligations to one or more recipients of Covered
|
||||
Software. However, You may do so only on Your own behalf, and not on
|
||||
behalf of any Contributor. You must make it absolutely clear that any
|
||||
such warranty, support, indemnity, or liability obligation is offered by
|
||||
You alone, and You hereby agree to indemnify every Contributor for any
|
||||
liability incurred by such Contributor as a result of warranty, support,
|
||||
indemnity or liability terms You offer. You may include additional
|
||||
disclaimers of warranty and limitations of liability specific to any
|
||||
jurisdiction.
|
||||
|
||||
|
||||
### 4. Inability to Comply Due to Statute or Regulation
|
||||
|
||||
If it is impossible for You to comply with any of the terms of this
|
||||
License with respect to some or all of the Covered Software due to
|
||||
statute, judicial order, or regulation then You must: **(a)** comply with
|
||||
the terms of this License to the maximum extent possible; and **(b)**
|
||||
describe the limitations and the code they affect. Such description must
|
||||
be placed in a text file included with all distributions of the Covered
|
||||
Software under this License. Except to the extent prohibited by statute
|
||||
or regulation, such description must be sufficiently detailed for a
|
||||
recipient of ordinary skill to be able to understand it.
|
||||
|
||||
|
||||
### 5. Termination
|
||||
|
||||
**5.1.** The rights granted under this License will terminate automatically
|
||||
if You fail to comply with any of its terms. However, if You become
|
||||
compliant, then the rights granted under this License from a particular
|
||||
Contributor are reinstated **(a)** provisionally, unless and until such
|
||||
Contributor explicitly and finally terminates Your grants, and **(b)** on an
|
||||
ongoing basis, if such Contributor fails to notify You of the
|
||||
non-compliance by some reasonable means prior to 60 days after You have
|
||||
come back into compliance. Moreover, Your grants from a particular
|
||||
Contributor are reinstated on an ongoing basis if such Contributor
|
||||
notifies You of the non-compliance by some reasonable means, this is the
|
||||
first time You have received notice of non-compliance with this License
|
||||
from such Contributor, and You become compliant prior to 30 days after
|
||||
Your receipt of the notice.
|
||||
|
||||
**5.2.** If You initiate litigation against any entity by asserting a patent
|
||||
infringement claim (excluding declaratory judgment actions,
|
||||
counter-claims, and cross-claims) alleging that a Contributor Version
|
||||
directly or indirectly infringes any patent, then the rights granted to
|
||||
You by any and all Contributors for the Covered Software under Section
|
||||
2.1 of this License shall terminate.
|
||||
|
||||
**5.3.** In the event of termination under Sections 5.1 or 5.2 above, all
|
||||
end user license agreements (excluding distributors and resellers) which
|
||||
have been validly granted by You or Your distributors under this License
|
||||
prior to termination shall survive termination.
|
||||
|
||||
|
||||
### 6. Disclaimer of Warranty
|
||||
|
||||
> Covered Software is provided under this License on an “as is”
|
||||
> basis, without warranty of any kind, either expressed, implied, or
|
||||
> statutory, including, without limitation, warranties that the
|
||||
> Covered Software is free of defects, merchantable, fit for a
|
||||
> particular purpose or non-infringing. The entire risk as to the
|
||||
> quality and performance of the Covered Software is with You.
|
||||
> Should any Covered Software prove defective in any respect, You
|
||||
> (not any Contributor) assume the cost of any necessary servicing,
|
||||
> repair, or correction. This disclaimer of warranty constitutes an
|
||||
> essential part of this License. No use of any Covered Software is
|
||||
> authorized under this License except under this disclaimer.
|
||||
|
||||
### 7. Limitation of Liability
|
||||
|
||||
> Under no circumstances and under no legal theory, whether tort
|
||||
> (including negligence), contract, or otherwise, shall any
|
||||
> Contributor, or anyone who distributes Covered Software as
|
||||
> permitted above, be liable to You for any direct, indirect,
|
||||
> special, incidental, or consequential damages of any character
|
||||
> including, without limitation, damages for lost profits, loss of
|
||||
> goodwill, work stoppage, computer failure or malfunction, or any
|
||||
> and all other commercial damages or losses, even if such party
|
||||
> shall have been informed of the possibility of such damages. This
|
||||
> limitation of liability shall not apply to liability for death or
|
||||
> personal injury resulting from such party's negligence to the
|
||||
> extent applicable law prohibits such limitation. Some
|
||||
> jurisdictions do not allow the exclusion or limitation of
|
||||
> incidental or consequential damages, so this exclusion and
|
||||
> limitation may not apply to You.
|
||||
|
||||
|
||||
### 8. Litigation
|
||||
|
||||
Any litigation relating to this License may be brought only in the
|
||||
courts of a jurisdiction where the defendant maintains its principal
|
||||
place of business and such litigation shall be governed by laws of that
|
||||
jurisdiction, without reference to its conflict-of-law provisions.
|
||||
Nothing in this Section shall prevent a party's ability to bring
|
||||
cross-claims or counter-claims.
|
||||
|
||||
|
||||
### 9. Miscellaneous
|
||||
|
||||
This License represents the complete agreement concerning the subject
|
||||
matter hereof. If any provision of this License is held to be
|
||||
unenforceable, such provision shall be reformed only to the extent
|
||||
necessary to make it enforceable. Any law or regulation which provides
|
||||
that the language of a contract shall be construed against the drafter
|
||||
shall not be used to construe this License against a Contributor.
|
||||
|
||||
|
||||
### 10. Versions of the License
|
||||
|
||||
#### 10.1. New Versions
|
||||
|
||||
Mozilla Foundation is the license steward. Except as provided in Section
|
||||
10.3, no one other than the license steward has the right to modify or
|
||||
publish new versions of this License. Each version will be given a
|
||||
distinguishing version number.
|
||||
|
||||
#### 10.2. Effect of New Versions
|
||||
|
||||
You may distribute the Covered Software under the terms of the version
|
||||
of the License under which You originally received the Covered Software,
|
||||
or under the terms of any subsequent version published by the license
|
||||
steward.
|
||||
|
||||
#### 10.3. Modified Versions
|
||||
|
||||
If you create software not governed by this License, and you want to
|
||||
create a new license for such software, you may create and use a
|
||||
modified version of this License if you rename the license and remove
|
||||
any references to the name of the license steward (except to note that
|
||||
such modified license differs from this License).
|
||||
|
||||
#### 10.4. Distributing Source Code Form that is Incompatible With Secondary Licenses
|
||||
|
||||
If You choose to distribute Source Code Form that is Incompatible With
|
||||
Secondary Licenses under the terms of this version of the License, the
|
||||
notice described in Exhibit B of this License must be attached.
|
||||
|
||||
## Exhibit A - Source Code Form License Notice
|
||||
|
||||
This Source Code Form is subject to the terms of the Mozilla Public
|
||||
License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||
|
||||
If it is not possible or desirable to put the notice in a particular
|
||||
file, then You may include the notice in a location (such as a LICENSE
|
||||
file in a relevant directory) where a recipient would be likely to look
|
||||
for such a notice.
|
||||
|
||||
You may add additional accurate notices of copyright ownership.
|
||||
|
||||
## Exhibit B - “Incompatible With Secondary Licenses” Notice
|
||||
|
||||
This Source Code Form is "Incompatible With Secondary Licenses", as
|
||||
defined by the Mozilla Public License, v. 2.0.
|
||||
|
2
README.md
Normal file
2
README.md
Normal file
|
@ -0,0 +1,2 @@
|
|||
# 2b2t
|
||||
A program that collects data about 2b2t's player count and submits it to the webapp.
|
3
go.mod
Normal file
3
go.mod
Normal file
|
@ -0,0 +1,3 @@
|
|||
module git.1e99.eu/2b2t-queue/monitor-player-count
|
||||
|
||||
go 1.24.2
|
75
main.go
Normal file
75
main.go
Normal file
|
@ -0,0 +1,75 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"log"
|
||||
"os"
|
||||
"os/signal"
|
||||
"sync"
|
||||
)
|
||||
|
||||
func main() {
|
||||
logger := log.New(os.Stdout, "", log.Flags())
|
||||
logger.Printf(` ___ ____ ___ _______ __ __ _ _ `)
|
||||
logger.Printf(`|__ \| _ \__ \__ __| | \/ | (_) | `)
|
||||
logger.Printf(` ) | |_) | ) | | | | \ / | ___ _ __ _| |_ ___ _ __ `)
|
||||
logger.Printf(` / /| _ < / / | | | |\/| |/ _ \| '_ \| | __/ _ \| '__|`)
|
||||
logger.Printf(` / /_| |_) / /_ | | | | | | (_) | | | | | || (_) | | `)
|
||||
logger.Printf(`|____|____/____| |_| |_| |_|\___/|_| |_|_|\__\___/|_| `)
|
||||
logger.Printf(` `)
|
||||
|
||||
submitUrl, found := os.LookupEnv("WORKER_SUBMIT_URL")
|
||||
if !found {
|
||||
logger.Printf("No \"WORKER_SUBMIT_URL\", exiting")
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
token, found := os.LookupEnv("WORKER_TOKEN")
|
||||
if !found {
|
||||
logger.Printf("No \"WORKER_TOKEN\", exiting")
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
poller := makePoller(logger)
|
||||
logger.Printf("Submitting data to %s", submitUrl)
|
||||
logger.Printf("Using poller %s", poller)
|
||||
logger.Printf("")
|
||||
|
||||
var wg sync.WaitGroup
|
||||
stop := make(chan bool)
|
||||
queue := make(chan PlayerCount, 1024)
|
||||
|
||||
go SubmitLoop(stop, &wg, queue, logger, submitUrl, token)
|
||||
go PollLoop(stop, &wg, queue, logger, poller)
|
||||
|
||||
notify := make(chan os.Signal, 1)
|
||||
signal.Notify(notify, os.Interrupt)
|
||||
<-notify
|
||||
|
||||
close(stop)
|
||||
wg.Wait()
|
||||
}
|
||||
|
||||
func makePoller(logger *log.Logger) (poller Poller) {
|
||||
pollerType, found := os.LookupEnv("WORKER_POLLER")
|
||||
if !found {
|
||||
logger.Printf("No \"WORKER_POLLER\", defaulting to \"mcapi\"")
|
||||
pollerType = "mcapi"
|
||||
}
|
||||
|
||||
switch pollerType {
|
||||
case "dummy":
|
||||
poller = &dummyPoller{}
|
||||
return
|
||||
|
||||
case "mcapi":
|
||||
poller = &mcApiPoller{
|
||||
Address: "2b2t.org",
|
||||
}
|
||||
return
|
||||
|
||||
default:
|
||||
logger.Printf("Invalid poller type")
|
||||
os.Exit(1)
|
||||
return
|
||||
}
|
||||
}
|
6
poller-direct.go
Normal file
6
poller-direct.go
Normal file
|
@ -0,0 +1,6 @@
|
|||
package main
|
||||
|
||||
// TOOD:
|
||||
// - Reimplement the Minecraft Protocol and reach out to 2b2t that way. I want to make this the default later on
|
||||
// - https://minecraft.wiki/w/Java_Edition_protocol/Packets
|
||||
// - https://minecraft.wiki/w/Java_Edition_protocol/FAQ#What_does_the_normal_status_ping_sequence_look_like?
|
17
poller-dummy.go
Normal file
17
poller-dummy.go
Normal file
|
@ -0,0 +1,17 @@
|
|||
package main
|
||||
|
||||
type dummyPoller struct {
|
||||
}
|
||||
|
||||
func (p *dummyPoller) PollPlayerCount() (PlayerCount, error) {
|
||||
count := PlayerCount{
|
||||
Playing: 0,
|
||||
NormalQueue: 0,
|
||||
PriorityQueue: 0,
|
||||
}
|
||||
return count, nil
|
||||
}
|
||||
|
||||
func (p *dummyPoller) String() string {
|
||||
return "dummy[]"
|
||||
}
|
74
poller-mcapi.go
Normal file
74
poller-mcapi.go
Normal file
|
@ -0,0 +1,74 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
type mcApiPoller struct {
|
||||
Address string
|
||||
}
|
||||
|
||||
func (p *mcApiPoller) PollPlayerCount() (PlayerCount, error) {
|
||||
url, _ := url.Parse("https://mcapi.us/server/status")
|
||||
query := url.Query()
|
||||
query.Set("ip", p.Address)
|
||||
url.RawQuery = query.Encode()
|
||||
|
||||
res, err := http.Get(url.String())
|
||||
if err != nil {
|
||||
return PlayerCount{}, fmt.Errorf("http request: %w", err)
|
||||
}
|
||||
defer res.Body.Close()
|
||||
|
||||
var data struct {
|
||||
Players struct {
|
||||
Max int `json:"max"`
|
||||
Now int `json:"now"`
|
||||
Sample []struct {
|
||||
Name string `json:"name"`
|
||||
Id string `json:"id"`
|
||||
} `json:"sample"`
|
||||
} `json:"players"`
|
||||
}
|
||||
err = json.NewDecoder(res.Body).Decode(&data)
|
||||
if err != nil {
|
||||
return PlayerCount{}, fmt.Errorf("invalid json response: %w", err)
|
||||
}
|
||||
|
||||
var count PlayerCount
|
||||
for _, player := range data.Players.Sample {
|
||||
matched := playerRegex.FindStringSubmatch(player.Name)
|
||||
if matched == nil || len(matched) != 3 {
|
||||
return PlayerCount{}, fmt.Errorf("match player regex against \"%s\"", player.Name)
|
||||
}
|
||||
|
||||
value, err := strconv.Atoi(matched[2])
|
||||
if err != nil {
|
||||
return PlayerCount{}, fmt.Errorf("parse number: %w", err)
|
||||
}
|
||||
|
||||
switch matched[1] {
|
||||
case "§6In-game":
|
||||
count.Playing = value
|
||||
|
||||
case "§6Queue":
|
||||
count.NormalQueue = value
|
||||
|
||||
case "§6Priority queue":
|
||||
count.PriorityQueue = value
|
||||
|
||||
default:
|
||||
return PlayerCount{}, fmt.Errorf("unknown player \"%s\"", matched[1])
|
||||
}
|
||||
}
|
||||
|
||||
return count, nil
|
||||
}
|
||||
|
||||
func (p *mcApiPoller) String() string {
|
||||
return fmt.Sprintf("mcapi[address=%s]", p.Address)
|
||||
}
|
59
poller.go
Normal file
59
poller.go
Normal file
|
@ -0,0 +1,59 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"log"
|
||||
"regexp"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
var (
|
||||
playerRegex = regexp.MustCompile(`^(.+): (\d+)$`)
|
||||
)
|
||||
|
||||
type PlayerCount struct {
|
||||
// The number of minutes elapsed since January 1, 1970 UTC.
|
||||
Time int64 `json:"time"`
|
||||
Playing int `json:"playing"`
|
||||
NormalQueue int `json:"normal_queue"`
|
||||
PriorityQueue int `json:"priority_queue"`
|
||||
}
|
||||
|
||||
type Poller interface {
|
||||
PollPlayerCount() (PlayerCount, error)
|
||||
String() string
|
||||
}
|
||||
|
||||
func PollLoop(stop chan bool, wg *sync.WaitGroup, queue chan PlayerCount, logger *log.Logger, poller Poller) {
|
||||
defer wg.Done()
|
||||
defer logger.Printf("Poll loop exited")
|
||||
wg.Add(1)
|
||||
logger.Printf("Poll loop started")
|
||||
|
||||
var lastPoll int64 = 0
|
||||
for {
|
||||
select {
|
||||
case <-stop:
|
||||
return
|
||||
|
||||
default:
|
||||
now := time.Now().Unix() / 60
|
||||
if lastPoll == now {
|
||||
continue
|
||||
}
|
||||
|
||||
count, err := poller.PollPlayerCount()
|
||||
if err != nil {
|
||||
logger.Printf("Failed polling player count: %s", err)
|
||||
continue
|
||||
}
|
||||
|
||||
// Convert seconds into minutes
|
||||
count.Time = now
|
||||
lastPoll = now
|
||||
logger.Printf("Polled player count: %+v", count)
|
||||
|
||||
queue <- count
|
||||
}
|
||||
}
|
||||
}
|
50
submit.go
Normal file
50
submit.go
Normal file
|
@ -0,0 +1,50 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"log"
|
||||
"net/http"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
func SubmitLoop(stop chan bool, wg *sync.WaitGroup, queue chan PlayerCount, logger *log.Logger, url string, token string) {
|
||||
defer wg.Done()
|
||||
defer logger.Printf("Submit loop exited")
|
||||
wg.Add(1)
|
||||
logger.Printf("Submit loop started")
|
||||
|
||||
for {
|
||||
select {
|
||||
case <-stop:
|
||||
return
|
||||
|
||||
case count := <-queue:
|
||||
logger.Printf("Submitting player count: %+v", count)
|
||||
|
||||
data, err := json.Marshal(&count)
|
||||
if err != nil {
|
||||
logger.Printf("Failed marshalling player count: %s", err)
|
||||
continue
|
||||
}
|
||||
|
||||
req, err := http.NewRequest(http.MethodPut, url, bytes.NewReader(data))
|
||||
if err != nil {
|
||||
logger.Printf("Failed to create submit request: %s", err)
|
||||
continue
|
||||
}
|
||||
|
||||
req.Header.Set("X-Token", token)
|
||||
|
||||
res, err := http.DefaultClient.Do(req)
|
||||
if err != nil {
|
||||
logger.Printf("Failed to execute submit request: %s", err)
|
||||
time.Sleep(10 * time.Second)
|
||||
queue <- count // Put up for retrying
|
||||
continue
|
||||
}
|
||||
res.Body.Close()
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Add table
Reference in a new issue