main.go

// Copyright (c) 2013-2016 The btcsuite developers
// Use of this source code is governed by an ISC
// license that can be found in the LICENSE file.

package main

import (
	"context"
	"errors"
	"fmt"
	"net/http"
	_ "net/http/pprof"
	"os"
	"runtime"
	"runtime/debug"

	"github.com/copernet/copernicus/conf"
	"github.com/copernet/copernicus/model"
	"github.com/copernet/copernicus/net/limits"
	"github.com/copernet/copernicus/net/server"
	"github.com/copernet/copernicus/rpc"
	"net"
)

const (
	// blockDbNamePrefix is the prefix for the block database name.  The
	// database type is appended to this value to form the full block
	// database name.
	blockDbNamePrefix = "blocks"
)

// bchMain is the real main function for copernicus.  It is necessary to work around
// the fact that deferred functions do not run when os.Exit() is called.
func bchMain(ctx context.Context, args []string) error {
	// Load configuration and parse command line.  This function also
	// initializes logging and configures it accordingly.
	appInitMain(args)
	go func() {
		listenAddr := net.JoinHostPort(conf.Cfg.PProf.IP, conf.Cfg.PProf.Port)
		fmt.Printf("Profile server listening on %s\n", listenAddr)
		profileRedirect := http.RedirectHandler("/debug/pprof", http.StatusSeeOther)
		http.Handle("/", profileRedirect)
		err := fmt.Errorf("%v", http.ListenAndServe(listenAddr, nil))
		fmt.Println(err.Error())
	}()
	interrupt := interruptListener()

	s, err := server.NewServer(model.ActiveNetParams, interrupt)
	if err != nil {
		return err
	}
	var rpcServer *rpc.Server
	if !conf.Cfg.P2PNet.DisableRPC {
		rpcServer, err = rpc.InitRPCServer()
		if err != nil {
			return errors.New("failed to init rpc")
		}
		// Start the rebroadcastHandler, which ensures user tx received by
		// the RPC server are rebroadcast until being included in a block.
		//go s.rebroadcastHandler()
		rpcServer.Start()
	}

	server.SetMsgHandle(context.TODO(), s.MsgChan, s)
	if interruptRequested(interrupt) {
		return nil
	}
	s.Start()
	defer func() {
		s.Stop()
		// Shutdown the RPC server if it's not disabled.
		if !conf.Cfg.P2PNet.DisableRPC {
			rpcServer.Stop()
		}
	}()
	go func() {
		<-rpcServer.RequestedProcessShutdown()
		shutdownRequestChannel <- struct{}{}
	}()
	<-interrupt
	return nil
}

func main() {
	// Use all processor cores.
	runtime.GOMAXPROCS(runtime.NumCPU())

	// Block and transaction processing can cause bursty allocations.  This
	// limits the garbage collector from excessively overallocating during
	// bursts.  This value was arrived at with the help of profiling live
	// usage.
	debug.SetGCPercent(10)

	// Up some limits.
	if err := limits.SetLimits(); err != nil {
		fmt.Fprintf(os.Stderr, "failed to set limits: %v\n", err)
		os.Exit(1)
	}

	args := os.Args
	// Work around defer not working after os.Exit()
	if err := bchMain(context.Background(), args); err != nil {
		os.Exit(1)
	}
}