github.com/kaydxh/golang@v0.0.131/pkg/logs/log.option.go (about)

     1  /*
     2   *Copyright (c) 2022, kaydxh
     3   *
     4   *Permission is hereby granted, free of charge, to any person obtaining a copy
     5   *of this software and associated documentation files (the "Software"), to deal
     6   *in the Software without restriction, including without limitation the rights
     7   *to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
     8   *copies of the Software, and to permit persons to whom the Software is
     9   *furnished to do so, subject to the following conditions:
    10   *
    11   *The above copyright notice and this permission notice shall be included in all
    12   *copies or substantial portions of the Software.
    13   *
    14   *THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
    15   *IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
    16   *FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
    17   *AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
    18   *LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
    19   *OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
    20   *SOFTWARE.
    21   */
    22  package logs
    23  
    24  import (
    25  	"fmt"
    26  	"path"
    27  	"path/filepath"
    28  	"runtime"
    29  	"time"
    30  )
    31  
    32  func WithPrefixName(prefixName string) RotateOption {
    33  	return RotateOptionFunc(func(c *Rotate) {
    34  		c.prefixName = prefixName
    35  	})
    36  }
    37  
    38  func WithSuffixName(suffixName string) RotateOption {
    39  	return RotateOptionFunc(func(c *Rotate) {
    40  		c.suffixName = suffixName
    41  	})
    42  }
    43  
    44  func WithMaxAge(maxAge time.Duration) RotateOption {
    45  	return RotateOptionFunc(func(c *Rotate) {
    46  		c.maxAge = maxAge
    47  	})
    48  }
    49  
    50  func WithMaxCount(maxCount int64) RotateOption {
    51  	return RotateOptionFunc(func(c *Rotate) {
    52  		c.maxCount = maxCount
    53  	})
    54  }
    55  
    56  func WithRotateSize(rotateSize int64) RotateOption {
    57  	return RotateOptionFunc(func(c *Rotate) {
    58  		c.rotateSize = rotateSize
    59  	})
    60  }
    61  
    62  func WithRotateInterval(rotateInterval time.Duration) RotateOption {
    63  	return RotateOptionFunc(func(c *Rotate) {
    64  		c.rotateInterval = rotateInterval
    65  	})
    66  }
    67  
    68  type CallerPrettyfierFunc func(f *runtime.Frame) (function string, file string)
    69  
    70  func GenShortCallPrettyfier() CallerPrettyfierFunc {
    71  	return func(f *runtime.Frame) (function string, file string) {
    72  		funcname := path.Base(f.Function)
    73  		dir := path.Dir(f.File)
    74  		filename := filepath.Join(path.Base(dir), path.Base(f.File))
    75  		//filename := path.Base(f.File)
    76  		return fmt.Sprintf("%s()", funcname), fmt.Sprintf("%s:%d", filename, f.Line)
    77  	}
    78  }