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.
49 lines
1007 B
Go
49 lines
1007 B
Go
/*
|
|
* SPDX-FileCopyrightText: © 2017-2026 Istari Digital, Inc.
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
package x
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"runtime"
|
|
|
|
"github.com/pkg/profile"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// Stopper is an interface tasked with stopping the profiling process.
|
|
type Stopper interface {
|
|
Stop()
|
|
}
|
|
|
|
// StartProfile starts a new mode for profiling.
|
|
func StartProfile(conf *viper.Viper) Stopper {
|
|
profileMode := conf.GetString("profile_mode")
|
|
switch profileMode {
|
|
case "cpu":
|
|
return profile.Start(profile.CPUProfile)
|
|
case "mem":
|
|
return profile.Start(profile.MemProfile)
|
|
case "mutex":
|
|
return profile.Start(profile.MutexProfile)
|
|
case "block":
|
|
blockRate := conf.GetInt("block_rate")
|
|
runtime.SetBlockProfileRate(blockRate)
|
|
return profile.Start(profile.BlockProfile)
|
|
case "":
|
|
// do nothing
|
|
return noOpStopper{}
|
|
default:
|
|
fmt.Printf("Invalid profile mode: %q\n", profileMode)
|
|
os.Exit(1)
|
|
return noOpStopper{}
|
|
}
|
|
}
|
|
|
|
type noOpStopper struct{}
|
|
|
|
func (noOpStopper) Stop() {}
|