接入阿里sls, 增加关闭日志通道
This commit is contained in:
		
							parent
							
								
									b3a1b78006
								
							
						
					
					
						commit
						1d9b78eb37
					
				
							
								
								
									
										19
									
								
								log.go
									
									
									
									
									
								
							
							
						
						
									
										19
									
								
								log.go
									
									
									
									
									
								
							@ -86,23 +86,36 @@ func GetAli(label string, channelLens ...int64) (l *Logger) {
 | 
				
			|||||||
	return
 | 
						return
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func Close(lable string) {
 | 
					func Close(lables ...string) {
 | 
				
			||||||
 | 
						wg := &sync.WaitGroup{}
 | 
				
			||||||
 | 
						for _, lable := range lables {
 | 
				
			||||||
 | 
							wg.Add(1)
 | 
				
			||||||
 | 
							go func() {
 | 
				
			||||||
			if v, ok := loggers.Load(lable); ok {
 | 
								if v, ok := loggers.Load(lable); ok {
 | 
				
			||||||
				if tmp, ok := v.(*Logger); ok {
 | 
									if tmp, ok := v.(*Logger); ok {
 | 
				
			||||||
 | 
										loggers.Delete(lable)
 | 
				
			||||||
					tmp.BeeLogger.Close()
 | 
										tmp.BeeLogger.Close()
 | 
				
			||||||
				}
 | 
									}
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
	loggers.Delete(lable)
 | 
								wg.Done()
 | 
				
			||||||
 | 
							}()
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						wg.Wait()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func CloseAll() {
 | 
					func CloseAll() {
 | 
				
			||||||
 | 
						wg := &sync.WaitGroup{}
 | 
				
			||||||
	loggers.Range(func(key, value interface{}) bool {
 | 
						loggers.Range(func(key, value interface{}) bool {
 | 
				
			||||||
 | 
							wg.Add(1)
 | 
				
			||||||
		go func() {
 | 
							go func() {
 | 
				
			||||||
			if tmp, ok := value.(*Logger); ok {
 | 
								if tmp, ok := value.(*Logger); ok {
 | 
				
			||||||
				tmp.BeeLogger.Close()
 | 
									tmp.BeeLogger.Close()
 | 
				
			||||||
 | 
									loggers.Delete(key)
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
 | 
								wg.Done()
 | 
				
			||||||
		}()
 | 
							}()
 | 
				
			||||||
 | 
					 | 
				
			||||||
		return true
 | 
							return true
 | 
				
			||||||
	})
 | 
						})
 | 
				
			||||||
 | 
						wg.Wait()
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
				
			|||||||
							
								
								
									
										17
									
								
								log_test.go
									
									
									
									
									
								
							
							
						
						
									
										17
									
								
								log_test.go
									
									
									
									
									
								
							@ -34,17 +34,18 @@ func TestGetLogger(t *testing.T) {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
func TestGetAliLs(t *testing.T) {
 | 
					func TestGetAliLs(t *testing.T) {
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	l := GetAli("test_alils")
 | 
						lable := "test_alils"
 | 
				
			||||||
 | 
						l := GetAli(lable).Async(128)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	for i := 0; i < 1; i++ {
 | 
						for i := 0; i < 2; i++ {
 | 
				
			||||||
		l.Info("endport")
 | 
							l.Info("Info")
 | 
				
			||||||
		l.Info("endport")
 | 
							l.Debug("Debug")
 | 
				
			||||||
		l.Info("endport")
 | 
							l.Warn("Warn")
 | 
				
			||||||
		l.Info("endport")
 | 
							l.Warning("Warning")
 | 
				
			||||||
		l.Error("Error")
 | 
							l.Error("Error")
 | 
				
			||||||
		l.Error("Error\n\n Error")
 | 
							l.Error("Error\n\n 测试换行")
 | 
				
			||||||
		l.Warn("Warn")
 | 
							l.Warn("Warn")
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	Close("test_alils")
 | 
						Close(lable, lable, lable)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
				
			|||||||
		Loading…
	
		Reference in New Issue
	
	Block a user