main.go

// Copyright (C) 2022 The go-redis Authors All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

/*
 go-redisd is an example of implementing a compatible Redis server using go-mysql.
	NAME
	 go-redisd

	SYNOPSIS
	 go-redisd [OPTIONS]

	OPTIONS
	-v      : Enable verbose output.
	-p      : Enable profiling.

	RETURN VALUE
	  Return EXIT_SUCCESS or EXIT_FAILURE
*/

package main

import (
	"flag"
	"log"
	"net/http"
	"os"
	"os/signal"
	"syscall"

	clog "github.com/cybergarage/go-logger/log"
	"github.com/cybergarage/go-redis/examples/go-redisd/server"
)

const (
	programName = " go-redisd"
)

func main() {
	isDebugEnabled := flag.Bool("debug", false, "enable debugging log output")
	isProfileEnabled := flag.Bool("profile", false, "enable profiling server")
	flag.Parse()

	logLevel := clog.LevelTrace
	if *isDebugEnabled {
		logLevel = clog.LevelDebug
	}
	clog.SetSharedLogger(clog.NewStdoutLogger(logLevel))

	if *isProfileEnabled {
		go func() {
			// nolint: gosec
			log.Println(http.ListenAndServe("localhost:6060", nil))
		}()
	}

	server := server.NewServer()
	if err := server.Start(); err != nil {
		clog.Errorf("%s couldn't be started (%s)", programName, err.Error())
		os.Exit(1)
	}

	sigCh := make(chan os.Signal, 1)

	signal.Notify(sigCh,
		os.Interrupt,
		syscall.SIGHUP,
		syscall.SIGINT,
		syscall.SIGTERM)

	exitCh := make(chan int)

	go func() {
		for {
			s := <-sigCh
			switch s {
			case syscall.SIGHUP:
				clog.Infof("caught SIGHUP, restarting...")
				if err := server.Restart(); err != nil {
					clog.Errorf("%s couldn't be restarted (%s)", programName, err.Error())
					os.Exit(1)
				}
			case syscall.SIGINT, syscall.SIGTERM:
				clog.Infof("caught %s, stopping...", s.String())
				if err := server.Stop(); err != nil {
					clog.Errorf("%s couldn't be stopped (%s)", programName, err.Error())
					os.Exit(1)
				}
				exitCh <- 0
			}
		}
	}()

	code := <-exitCh

	os.Exit(code)
}