mirror of
https://github.com/hanzoai/graph.git
synced 2026-10-30 18:35:37 +00:00
zeekay
f5aaeb712e
Derived from Dgraph (github.com/dgraph-io/dgraph) at commit 64804bd2, Apache-2.0. See NOTICE. WHAT THIS IS. The graph engine as a Go library: posting lists, the DQL parser, the type and tokenizer system, the index and schema layers, and the wire types. 39,170 lines of the upstream's 166,337. THE STORE is github.com/luxfi/zapdb, a Badger fork that kept the package name, the subpackages (options, pb, y) and the API — Open, OpenManaged, DB, Options — so the 74 import sites are aliased rather than rewritten. THE CODEC is ZAP. zapdb serializes with ZAP, not protobuf: its pb package builds the ZAP types by default and the protobuf ones only under `-tags grpc`, a path that does not compile (table/builder.go calls pb.Marshal, which that tag removes). One call site in posting needed converting; every other type here is Hanzo's own protobuf and is untouched. NO TRANSPORT. An embedded engine dials nothing, so nothing here imports google.golang.org/grpc. Removed: 20 transport functions in x, the generated service stubs for Raft, Zero and Worker (1,657 lines), and the JWT helper — identity belongs to IAM. The core called none of them. NOT KEPT: worker, conn, raftwal, the cluster commands, the GraphQL layer and the test harnesses. `query`, the DQL execution engine, went with them because it depends on worker; parsing DQL is available here, executing it is not. The history of the upstream project lives at github.com/dgraph-io/dgraph. This repository begins at the derivation because three quarters of that history describes code this fork does not carry.
71 lines
2.0 KiB
Go
71 lines
2.0 KiB
Go
/*
|
|
* SPDX-FileCopyrightText: © 2017-2026 Istari Digital, Inc.
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
package x
|
|
|
|
import (
|
|
"sync/atomic"
|
|
|
|
"github.com/golang/glog"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
var (
|
|
// the drainingMode variable should be accessed through the atomic.Store and atomic.Load
|
|
// functions. The value 0 means the draining-mode is disabled, and the value 1 means the
|
|
// mode is enabled
|
|
drainingMode uint32
|
|
extSnapshotStreamingState uint32
|
|
|
|
healthCheck uint32
|
|
errHealth = errors.New("Please retry again, server is not ready to accept requests")
|
|
errDrainingMode = errors.New("the server is in draining mode " +
|
|
"and client requests will only be allowed after exiting the mode " +
|
|
" by sending a GraphQL draining(enable: false) mutation to /admin")
|
|
)
|
|
|
|
// UpdateHealthStatus updates the server's health status so it can start accepting requests.
|
|
func UpdateHealthStatus(ok bool) {
|
|
setStatus(&healthCheck, ok)
|
|
}
|
|
|
|
// UpdateDrainingMode updates the server's draining mode
|
|
func UpdateDrainingMode(enable bool) {
|
|
setStatus(&drainingMode, enable)
|
|
}
|
|
|
|
// ExtSnapshotStreamingState updates the server's import mode
|
|
func ExtSnapshotStreamingState(enable bool) {
|
|
glog.Info("[import] Updating import mode to ", enable)
|
|
setStatus(&extSnapshotStreamingState, enable)
|
|
setStatus(&drainingMode, enable)
|
|
}
|
|
|
|
// IsExtSnapshotStreamingStateTrue returns whether the server is in import mode or not
|
|
func IsExtSnapshotStreamingStateTrue() bool {
|
|
return atomic.LoadUint32(&extSnapshotStreamingState) == 1
|
|
}
|
|
|
|
// HealthCheck returns whether the server is ready to accept requests or not
|
|
// Load balancer would add the node to the endpoint once health check starts
|
|
// returning true
|
|
func HealthCheck() error {
|
|
if atomic.LoadUint32(&healthCheck) == 0 {
|
|
return errHealth
|
|
}
|
|
if atomic.LoadUint32(&drainingMode) == 1 {
|
|
return errDrainingMode
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func setStatus(v *uint32, ok bool) {
|
|
if ok {
|
|
atomic.StoreUint32(v, 1)
|
|
} else {
|
|
atomic.StoreUint32(v, 0)
|
|
}
|
|
}
|